Add P2PDatagramSocket and P2PStreamSocket interfaces.

remoting/protocol/channel_multiplexer_unittest.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/channel_multiplexer.h"
 
 #include "base/bind.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "net/base/net_errors.h"
@@ skipping 27 matching lines @@
                                          base::MessageLoop::QuitClosure());
 }
 
 class MockSocketCallback {
  public:
   MOCK_METHOD1(OnDone, void(int result));
 };
 
 class MockConnectCallback {
  public:
-  MOCK_METHOD1(OnConnectedPtr, void(net::StreamSocket* socket));
-  void OnConnected(scoped_ptr<net::StreamSocket> socket) {
+  MOCK_METHOD1(OnConnectedPtr, void(P2PStreamSocket* socket));
+  void OnConnected(scoped_ptr<P2PStreamSocket> socket) {
     OnConnectedPtr(socket.release());
   }
 };
 
 }  // namespace
 
 class ChannelMultiplexerTest : public testing::Test {
  public:
   void DeleteAll() {
     host_socket1_.reset();
@@ skipping 27 matching lines @@
     FakeStreamSocket* client_socket =
         client_session_.fake_channel_factory().GetFakeChannel(
             ChannelMultiplexer::kMuxChannelName);
     host_socket->PairWith(client_socket);
 
     // Make writes asynchronous in one direction.
     host_socket->set_async_write(true);
   }
 
   void CreateChannel(const std::string& name,
-                     scoped_ptr<net::StreamSocket>* host_socket,
-                     scoped_ptr<net::StreamSocket>* client_socket) {
+                     scoped_ptr<P2PStreamSocket>* host_socket,
+                     scoped_ptr<P2PStreamSocket>* client_socket) {
     int counter = 2;
     host_mux_->CreateChannel(name, base::Bind(
         &ChannelMultiplexerTest::OnChannelConnected, base::Unretained(this),
         host_socket, &counter));
     client_mux_->CreateChannel(name, base::Bind(
         &ChannelMultiplexerTest::OnChannelConnected, base::Unretained(this),
         client_socket, &counter));
 
     message_loop_.Run();
 
     EXPECT_TRUE(host_socket->get());
     EXPECT_TRUE(client_socket->get());
   }
 
   void OnChannelConnected(
-      scoped_ptr<net::StreamSocket>* storage,
+      scoped_ptr<P2PStreamSocket>* storage,
       int* counter,
-      scoped_ptr<net::StreamSocket> socket) {
+      scoped_ptr<P2PStreamSocket> socket) {
     *storage = socket.Pass();
     --(*counter);
     EXPECT_GE(*counter, 0);
     if (*counter == 0)
       QuitCurrentThread();
   }
 
   scoped_refptr<net::IOBufferWithSize> CreateTestBuffer(int size) {
     scoped_refptr<net::IOBufferWithSize> result =
         new net::IOBufferWithSize(size);
     for (int i = 0; i< size; ++i) {
       result->data()[i] = rand() % 256;
     }
     return result;
   }
 
   base::MessageLoop message_loop_;
 
   FakeSession host_session_;
   FakeSession client_session_;
 
   scoped_ptr<ChannelMultiplexer> host_mux_;
   scoped_ptr<ChannelMultiplexer> client_mux_;
 
-  scoped_ptr<net::StreamSocket> host_socket1_;
-  scoped_ptr<net::StreamSocket> client_socket1_;
-  scoped_ptr<net::StreamSocket> host_socket2_;
-  scoped_ptr<net::StreamSocket> client_socket2_;
+  scoped_ptr<P2PStreamSocket> host_socket1_;
+  scoped_ptr<P2PStreamSocket> client_socket1_;
+  scoped_ptr<P2PStreamSocket> host_socket2_;
+  scoped_ptr<P2PStreamSocket> client_socket2_;
 };
 
 
 TEST_F(ChannelMultiplexerTest, OneChannel) {
-  scoped_ptr<net::StreamSocket> host_socket;
-  scoped_ptr<net::StreamSocket> client_socket;
+  scoped_ptr<P2PStreamSocket> host_socket;
+  scoped_ptr<P2PStreamSocket> client_socket;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel(kTestChannelName, &host_socket, &client_socket));
 
   ConnectSockets();
 
   StreamConnectionTester tester(host_socket.get(), client_socket.get(),
                                 kMessageSize, kMessages);
   tester.Start();
   message_loop_.Run();
   tester.CheckResults();
 }
 
 TEST_F(ChannelMultiplexerTest, TwoChannels) {
-  scoped_ptr<net::StreamSocket> host_socket1_;
-  scoped_ptr<net::StreamSocket> client_socket1_;
+  scoped_ptr<P2PStreamSocket> host_socket1_;
+  scoped_ptr<P2PStreamSocket> client_socket1_;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel(kTestChannelName, &host_socket1_, &client_socket1_));
 
-  scoped_ptr<net::StreamSocket> host_socket2_;
-  scoped_ptr<net::StreamSocket> client_socket2_;
+  scoped_ptr<P2PStreamSocket> host_socket2_;
+  scoped_ptr<P2PStreamSocket> client_socket2_;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel(kTestChannelName2, &host_socket2_, &client_socket2_));
 
   ConnectSockets();
 
   StreamConnectionTester tester1(host_socket1_.get(), client_socket1_.get(),
                                 kMessageSize, kMessages);
   StreamConnectionTester tester2(host_socket2_.get(), client_socket2_.get(),
                                  kMessageSize, kMessages);
   tester1.Start();
   tester2.Start();
   while (!tester1.done() || !tester2.done()) {
     message_loop_.Run();
   }
   tester1.CheckResults();
   tester2.CheckResults();
 }
 
 // Four channels, two in each direction
 TEST_F(ChannelMultiplexerTest, FourChannels) {
-  scoped_ptr<net::StreamSocket> host_socket1_;
-  scoped_ptr<net::StreamSocket> client_socket1_;
+  scoped_ptr<P2PStreamSocket> host_socket1_;
+  scoped_ptr<P2PStreamSocket> client_socket1_;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel(kTestChannelName, &host_socket1_, &client_socket1_));
 
-  scoped_ptr<net::StreamSocket> host_socket2_;
-  scoped_ptr<net::StreamSocket> client_socket2_;
+  scoped_ptr<P2PStreamSocket> host_socket2_;
+  scoped_ptr<P2PStreamSocket> client_socket2_;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel(kTestChannelName2, &host_socket2_, &client_socket2_));
 
-  scoped_ptr<net::StreamSocket> host_socket3;
-  scoped_ptr<net::StreamSocket> client_socket3;
+  scoped_ptr<P2PStreamSocket> host_socket3;
+  scoped_ptr<P2PStreamSocket> client_socket3;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel("test3", &host_socket3, &client_socket3));
 
-  scoped_ptr<net::StreamSocket> host_socket4;
-  scoped_ptr<net::StreamSocket> client_socket4;
+  scoped_ptr<P2PStreamSocket> host_socket4;
+  scoped_ptr<P2PStreamSocket> client_socket4;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel("ch4", &host_socket4, &client_socket4));
 
   ConnectSockets();
 
   StreamConnectionTester tester1(host_socket1_.get(), client_socket1_.get(),
                                 kMessageSize, kMessages);
   StreamConnectionTester tester2(host_socket2_.get(), client_socket2_.get(),
                                  kMessageSize, kMessages);
   StreamConnectionTester tester3(client_socket3.get(), host_socket3.get(),
                                  kMessageSize, kMessages);
   StreamConnectionTester tester4(client_socket4.get(), host_socket4.get(),
                                  kMessageSize, kMessages);
   tester1.Start();
   tester2.Start();
   tester3.Start();
   tester4.Start();
   while (!tester1.done() || !tester2.done() ||
          !tester3.done() || !tester4.done()) {
     message_loop_.Run();
   }
   tester1.CheckResults();
   tester2.CheckResults();
   tester3.CheckResults();
   tester4.CheckResults();
 }
 
 TEST_F(ChannelMultiplexerTest, WriteFailSync) {
-  scoped_ptr<net::StreamSocket> host_socket1_;
-  scoped_ptr<net::StreamSocket> client_socket1_;
+  scoped_ptr<P2PStreamSocket> host_socket1_;
+  scoped_ptr<P2PStreamSocket> client_socket1_;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel(kTestChannelName, &host_socket1_, &client_socket1_));
 
-  scoped_ptr<net::StreamSocket> host_socket2_;
-  scoped_ptr<net::StreamSocket> client_socket2_;
+  scoped_ptr<P2PStreamSocket> host_socket2_;
+  scoped_ptr<P2PStreamSocket> client_socket2_;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel(kTestChannelName2, &host_socket2_, &client_socket2_));
 
   ConnectSockets();
 
   host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)->
       set_next_write_error(net::ERR_FAILED);
   host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)->
       set_async_write(false);
 
@@ skipping 115 matching lines @@
       .WillOnce(InvokeWithoutArgs(
           this, &ChannelMultiplexerTest::DeleteAfterSessionFail));
   EXPECT_CALL(cb2, OnConnectedPtr(_))
       .Times(0);
 
   base::RunLoop().RunUntilIdle();
 }
 
 }  // namespace protocol
 }  // namespace remoting