Decouple Blimp transport output from BlimpConnections using MessagePort.

blimp/net/ssl_client_transport_unittest.cc

 // Copyright 2016 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 "base/bind.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "blimp/net/blimp_connection.h"
+#include "blimp/net/message_port.h"
 #include "blimp/net/ssl_client_transport.h"
 #include "net/base/address_list.h"
 #include "net/base/ip_address.h"
 #include "net/base/net_errors.h"
 #include "net/socket/socket_test_util.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace blimp {
 namespace {
@@ skipping 55 matching lines @@
 
 TEST_F(SSLClientTransportTest, ConnectSyncOK) {
   net::IPEndPoint endpoint(kIPV4Address, kPort);
   ConfigureTransport(endpoint);
   for (int i = 0; i < 5; ++i) {
     EXPECT_CALL(*this, ConnectComplete(net::OK));
     SetupTCPSyncSocketConnect(endpoint, net::OK);
     SetupSSLSyncSocketConnect(net::OK);
     transport_->Connect(base::Bind(&SSLClientTransportTest::ConnectComplete,
                                    base::Unretained(this)));
-    EXPECT_NE(nullptr, transport_->TakeConnection().get());
+    EXPECT_NE(nullptr, transport_->TakeMessagePort().get());
     base::RunLoop().RunUntilIdle();
   }
 }
 
 TEST_F(SSLClientTransportTest, ConnectAsyncOK) {
   net::IPEndPoint endpoint(kIPV4Address, kPort);
   ConfigureTransport(endpoint);
   for (int i = 0; i < 5; ++i) {
     EXPECT_CALL(*this, ConnectComplete(net::OK));
     SetupTCPAsyncSocketConnect(endpoint, net::OK);
     SetupSSLAsyncSocketConnect(net::OK);
     transport_->Connect(base::Bind(&SSLClientTransportTest::ConnectComplete,
                                    base::Unretained(this)));
     base::RunLoop().RunUntilIdle();
-    EXPECT_NE(nullptr, transport_->TakeConnection().get());
+    EXPECT_NE(nullptr, transport_->TakeMessagePort().get());
   }
 }
 
 TEST_F(SSLClientTransportTest, ConnectSyncTCPError) {
   net::IPEndPoint endpoint(kIPV4Address, kPort);
   ConfigureTransport(endpoint);
   EXPECT_CALL(*this, ConnectComplete(net::ERR_FAILED));
   SetupTCPSyncSocketConnect(endpoint, net::ERR_FAILED);
   transport_->Connect(base::Bind(&SSLClientTransportTest::ConnectComplete,
                                  base::Unretained(this)));
@@ skipping 38 matching lines @@
   transport_->Connect(base::Bind(&SSLClientTransportTest::ConnectComplete,
                                  base::Unretained(this)));
   base::RunLoop().RunUntilIdle();
 
   // Subsequent TCP+SSL connections succeed.
   EXPECT_CALL(*this, ConnectComplete(net::OK));
   SetupTCPSyncSocketConnect(endpoint, net::OK);
   SetupSSLSyncSocketConnect(net::OK);
   transport_->Connect(base::Bind(&SSLClientTransportTest::ConnectComplete,
                                  base::Unretained(this)));
-  EXPECT_NE(nullptr, transport_->TakeConnection().get());
+  EXPECT_NE(nullptr, transport_->TakeMessagePort().get());
   base::RunLoop().RunUntilIdle();
 }
 
 }  // namespace blimp