Move the core state machine of SSLClientSocketNSS into a thread-safe Core

remoting/protocol/authenticator_test_base.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "remoting/protocol/authenticator_test_base.h"
 
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/path_service.h"
+#include "base/test/test_timeouts.h"
+#include "base/timer.h"
 #include "crypto/rsa_private_key.h"
 #include "net/base/cert_test_util.h"
 #include "remoting/protocol/authenticator.h"
 #include "remoting/protocol/channel_authenticator.h"
 #include "remoting/protocol/fake_session.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/libjingle/source/talk/xmllite/xmlelement.h"
 
 using testing::_;
 using testing::SaveArg;
 
 namespace remoting {
 namespace protocol {
 
-AuthenticatorTestBase::MockChannelDoneCallback::MockChannelDoneCallback() {
-}
-AuthenticatorTestBase::MockChannelDoneCallback::~MockChannelDoneCallback() {
+namespace {
+
+ACTION_P(QuitThreadOnCounter, counter) {
+  --(*counter);
+  EXPECT_GE(*counter, 0);
+  if (*counter == 0)
+    MessageLoop::current()->Quit();
 }
 
-AuthenticatorTestBase::AuthenticatorTestBase() {
-}
-AuthenticatorTestBase::~AuthenticatorTestBase() {
-}
+}  // namespace
+
+AuthenticatorTestBase::MockChannelDoneCallback::MockChannelDoneCallback() {}
+
+AuthenticatorTestBase::MockChannelDoneCallback::~MockChannelDoneCallback() {}
+
+AuthenticatorTestBase::AuthenticatorTestBase() {}
+
+AuthenticatorTestBase::~AuthenticatorTestBase() {}
 
 void AuthenticatorTestBase::SetUp() {
   FilePath certs_dir(net::GetTestCertsDirectory());
 
   FilePath cert_path = certs_dir.AppendASCII("unittest.selfsigned.der");
   ASSERT_TRUE(file_util::ReadFileToString(cert_path, &host_cert_));
 
   FilePath key_path = certs_dir.AppendASCII("unittest.key.bin");
   std::string key_string;
   ASSERT_TRUE(file_util::ReadFileToString(key_path, &key_string));
@@ skipping 49 matching lines @@
                  base::Unretained(this)));
 
   host_auth_->SecureAndAuthenticate(
       host_fake_socket_.PassAs<net::StreamSocket>(),
       base::Bind(&AuthenticatorTestBase::OnHostConnected,
                  base::Unretained(this)));
 
   net::StreamSocket* client_socket = NULL;
   net::StreamSocket* host_socket = NULL;
 
+  // Expect two callbacks to be called - the client callback and the host
+  // callback.
+  int callback_counter = 2;
+
   EXPECT_CALL(client_callback_, OnDone(net::OK, _))
-      .WillOnce(SaveArg<1>(&client_socket));
+      .WillOnce(DoAll(SaveArg<1>(&client_socket),
+                      QuitThreadOnCounter(&callback_counter)));
   if (expected_fail) {
-    EXPECT_CALL(host_callback_, OnDone(net::ERR_FAILED, NULL));
+    EXPECT_CALL(host_callback_, OnDone(net::ERR_FAILED, NULL))
+         .WillOnce(QuitThreadOnCounter(&callback_counter));
   } else {
     EXPECT_CALL(host_callback_, OnDone(net::OK, _))
-        .WillOnce(SaveArg<1>(&host_socket));
+        .WillOnce(DoAll(SaveArg<1>(&host_socket),
+                        QuitThreadOnCounter(&callback_counter)));
   }
 
-  message_loop_.RunAllPending();
+  // Ensure that .Run() does not run unbounded if the callbacks are never
+  // called.
+  base::Timer shutdown_timer(false, false);
+  shutdown_timer.Start(FROM_HERE, TestTimeouts::action_timeout(),
+                       MessageLoop::QuitClosure());
+  message_loop_.Run();
+  shutdown_timer.Stop();
 
   testing::Mock::VerifyAndClearExpectations(&client_callback_);
   testing::Mock::VerifyAndClearExpectations(&host_callback_);
 
   client_socket_.reset(client_socket);
   host_socket_.reset(host_socket);
+
+  if (!expected_fail) {
+    ASSERT_TRUE(client_socket_.get() != NULL);
+    ASSERT_TRUE(host_socket_.get() != NULL);
+  }
 }
 
 void AuthenticatorTestBase::OnHostConnected(
     net::Error error,
     scoped_ptr<net::StreamSocket> socket) {
   host_callback_.OnDone(error, socket.get());
   host_socket_ = socket.Pass();
 }
 
 void AuthenticatorTestBase::OnClientConnected(
     net::Error error,
     scoped_ptr<net::StreamSocket> socket) {
   client_callback_.OnDone(error, socket.get());
   client_socket_ = socket.Pass();
 }
 
 }  // namespace protocol
 }  // namespace remoting