Blimp: add support for SSL connections.

blimp/net/tcp_transport_unittest.cc

 // Copyright 2015 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 <string>
 
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "blimp/common/create_blimp_message.h"
 #include "blimp/common/proto/blimp_message.pb.h"
@@ skipping 17 matching lines @@
 
 // Integration test for TCPEngineTransport and TCPClientTransport.
 class TCPTransportTest : public testing::Test {
  protected:
   TCPTransportTest() {
     net::IPEndPoint local_address;
     ParseAddress("127.0.0.1", 0, &local_address);
     engine_.reset(new TCPEngineTransport(local_address, nullptr));
   }
 
-  net::AddressList GetLocalAddressList() const {
+  net::IPEndPoint GetLocalEndpoint() const {
     net::IPEndPoint local_address;
-    engine_->GetLocalAddressForTesting(&local_address);
-    return net::AddressList(local_address);
+    CHECK_EQ(net::OK, engine_->GetLocalAddressForTesting(&local_address));
+    return local_address;
   }
 
   void ParseAddress(const std::string& ip_str,
                     uint16_t port,
                     net::IPEndPoint* address) {
     net::IPAddressNumber ip_number;
     bool rv = net::ParseIPLiteralToNumber(ip_str, &ip_number);
     if (!rv)
       return;
     *address = net::IPEndPoint(ip_number, port);
   }
 
   base::MessageLoopForIO message_loop_;
   scoped_ptr<TCPEngineTransport> engine_;
 };
 
 TEST_F(TCPTransportTest, Connect) {
   net::TestCompletionCallback accept_callback;
   engine_->Connect(accept_callback.callback());
 
   net::TestCompletionCallback connect_callback;
-  TCPClientTransport client(GetLocalAddressList(), nullptr);
+  TCPClientTransport client(GetLocalEndpoint(), nullptr);
   client.Connect(connect_callback.callback());
 
   EXPECT_EQ(net::OK, connect_callback.WaitForResult());
   EXPECT_EQ(net::OK, accept_callback.WaitForResult());
   EXPECT_TRUE(engine_->TakeConnection() != nullptr);
 }
 
 TEST_F(TCPTransportTest, TwoClientConnections) {
   net::TestCompletionCallback accept_callback1;
   engine_->Connect(accept_callback1.callback());
 
   net::TestCompletionCallback connect_callback1;
-  TCPClientTransport client1(GetLocalAddressList(), nullptr);
+  TCPClientTransport client1(GetLocalEndpoint(), nullptr);
   client1.Connect(connect_callback1.callback());
 
   net::TestCompletionCallback connect_callback2;
-  TCPClientTransport client2(GetLocalAddressList(), nullptr);
+  TCPClientTransport client2(GetLocalEndpoint(), nullptr);
   client2.Connect(connect_callback2.callback());
 
   EXPECT_EQ(net::OK, connect_callback1.WaitForResult());
   EXPECT_EQ(net::OK, accept_callback1.WaitForResult());
   EXPECT_TRUE(engine_->TakeConnection() != nullptr);
 
   net::TestCompletionCallback accept_callback2;
   engine_->Connect(accept_callback2.callback());
   EXPECT_EQ(net::OK, connect_callback2.WaitForResult());
   EXPECT_EQ(net::OK, accept_callback2.WaitForResult());
   EXPECT_TRUE(engine_->TakeConnection() != nullptr);
 }
 
 TEST_F(TCPTransportTest, ExchangeMessages) {
   // Start the Engine transport and connect a client to it.
   net::TestCompletionCallback accept_callback;
   engine_->Connect(accept_callback.callback());
   net::TestCompletionCallback client_connect_callback;
-  TCPClientTransport client(GetLocalAddressList(), nullptr /* NetLog */);
+  TCPClientTransport client(GetLocalEndpoint(), nullptr /* NetLog */);
   client.Connect(client_connect_callback.callback());
   EXPECT_EQ(net::OK, client_connect_callback.WaitForResult());
   EXPECT_EQ(net::OK, accept_callback.WaitForResult());
 
   // Expect the engine to get two messages from the client, and the client to
   // get one from the engine.
   MockBlimpMessageProcessor engine_incoming_processor;
   MockBlimpMessageProcessor client_incoming_processor;
   net::CompletionCallback engine_process_message_cb;
   scoped_ptr<BlimpMessage> client_message1 =
@@ skipping 35 matching lines @@
   // Client sends the second message.
   net::TestCompletionCallback client_send_callback2;
   client_connnection->GetOutgoingMessageProcessor()->ProcessMessage(
       std::move(client_message2), client_send_callback2.callback());
   EXPECT_EQ(net::OK, client_send_callback2.WaitForResult());
 }
 
 }  // namespace
 
 }  // namespace blimp