Chromium Code Reviews| Index: net/socket/ssl_client_socket_openssl_unittest.cc |
| diff --git a/net/socket/ssl_client_socket_openssl_unittest.cc b/net/socket/ssl_client_socket_openssl_unittest.cc |
| index 00cf9f35cb4b612841207513f112780a405880df..1f8c7df8a564d575b3993b3f7712cdbc8ce0ed57 100644 |
| --- a/net/socket/ssl_client_socket_openssl_unittest.cc |
| +++ b/net/socket/ssl_client_socket_openssl_unittest.cc |
| @@ -98,7 +98,8 @@ class SSLClientSocketOpenSSLClientAuthTest : public PlatformTest { |
| SSLClientSocketOpenSSLClientAuthTest() |
| : socket_factory_(net::ClientSocketFactory::GetDefaultFactory()), |
| cert_verifier_(new net::MockCertVerifier), |
| - transport_security_state_(new net::TransportSecurityState) { |
| + transport_security_state_(new net::TransportSecurityState), |
| + ran_completion_callback_(false) { |
| cert_verifier_->set_default_result(net::OK); |
| context_.cert_verifier = cert_verifier_.get(); |
| context_.transport_security_state = transport_security_state_.get(); |
| @@ -109,6 +110,8 @@ class SSLClientSocketOpenSSLClientAuthTest : public PlatformTest { |
| key_store_->Flush(); |
| } |
| + void RecordCompletedHandshake() { ran_completion_callback_ = true; } |
| + |
| protected: |
| scoped_ptr<SSLClientSocket> CreateSSLClientSocket( |
| scoped_ptr<StreamSocket> transport_socket, |
| @@ -173,6 +176,10 @@ class SSLClientSocketOpenSSLClientAuthTest : public PlatformTest { |
| test_server_->host_port_pair(), |
| ssl_config); |
| + sock_->SetHandshakeCompletionCallback(base::Bind( |
| + &SSLClientSocketOpenSSLClientAuthTest::RecordCompletedHandshake, |
| + base::Unretained(this))); |
| + |
| if (sock_->IsConnected()) { |
| LOG(ERROR) << "SSL Socket prematurely connected"; |
| return false; |
| @@ -202,6 +209,7 @@ class SSLClientSocketOpenSSLClientAuthTest : public PlatformTest { |
| CapturingNetLog log_; |
| scoped_ptr<StreamSocket> transport_; |
| scoped_ptr<SSLClientSocket> sock_; |
| + bool ran_completion_callback_; |
|
wtc
2014/07/30 21:56:57
Because of the "TestCompletionCallback callback_"
mshelley
2014/07/31 00:51:21
Done.
|
| }; |
| // Connect to a server requesting client authentication, do not send |
| @@ -220,6 +228,7 @@ TEST_F(SSLClientSocketOpenSSLClientAuthTest, NoCert) { |
| EXPECT_EQ(ERR_SSL_CLIENT_AUTH_CERT_NEEDED, rv); |
| EXPECT_FALSE(sock_->IsConnected()); |
| + EXPECT_TRUE(ran_completion_callback_); |
| } |
| // Connect to a server requesting client authentication, and send it |