Add P2PDatagramSocket and P2PStreamSocket interfaces.

remoting/protocol/pseudotcp_adapter_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 "remoting/protocol/pseudotcp_adapter.h"
 
 #include <vector>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/compiler_specific.h"
 #include "jingle/glue/thread_wrapper.h"
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
 #include "net/base/test_completion_callback.h"
+#include "remoting/protocol/p2p_datagram_socket.h"
+#include "remoting/protocol/p2p_stream_socket.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace remoting {
 namespace protocol {
 
 namespace {
 
 const int kMessageSize = 1024;
 const int kMessages = 100;
@@ skipping 32 matching lines @@
     return false;
   }
 
  private:
   double volume_;
   double rate_;
   double level_;
   base::TimeTicks last_update_;
 };
 
-class FakeSocket : public net::Socket {
+class FakeSocket : public P2PDatagramSocket {
  public:
   FakeSocket()
       : rate_limiter_(NULL),
         latency_ms_(0) {
   }
   ~FakeSocket() override {}
 
   void AppendInputPacket(const std::vector<char>& data) {
     if (rate_limiter_ && rate_limiter_->DropNextPacket())
       return;  // Lose the packet.
@@ skipping 13 matching lines @@
   void Connect(FakeSocket* peer_socket) {
     peer_socket_ = peer_socket;
   }
 
   void set_rate_limiter(RateLimiter* rate_limiter) {
     rate_limiter_ = rate_limiter;
   };
 
   void set_latency(int latency_ms) { latency_ms_ = latency_ms; };
 
-  // net::Socket interface.
-  int Read(net::IOBuffer* buf,
-           int buf_len,
+  // P2PDatagramSocket interface.
+  int Recv(const scoped_refptr<net::IOBuffer>& buf, int buf_len,
            const net::CompletionCallback& callback) override {
     CHECK(read_callback_.is_null());
     CHECK(buf);
 
     if (incoming_packets_.size() > 0) {
       scoped_refptr<net::IOBuffer> buffer(buf);
       int size = std::min(
           static_cast<int>(incoming_packets_.front().size()), buf_len);
       memcpy(buffer->data(), &*incoming_packets_.front().begin(), size);
       incoming_packets_.pop_front();
       return size;
     } else {
       read_callback_ = callback;
       read_buffer_ = buf;
       read_buffer_size_ = buf_len;
       return net::ERR_IO_PENDING;
     }
   }
 
-  int Write(net::IOBuffer* buf,
-            int buf_len,
-            const net::CompletionCallback& callback) override {
+  int Send(const scoped_refptr<net::IOBuffer>& buf, int buf_len,
+           const net::CompletionCallback& callback) override {
     DCHECK(buf);
     if (peer_socket_) {
       base::MessageLoop::current()->PostDelayedTask(
           FROM_HERE,
           base::Bind(&FakeSocket::AppendInputPacket,
                      base::Unretained(peer_socket_),
                      std::vector<char>(buf->data(), buf->data() + buf_len)),
           base::TimeDelta::FromMilliseconds(latency_ms_));
     }
 
     return buf_len;
   }
 
-  int SetReceiveBufferSize(int32 size) override {
-    NOTIMPLEMENTED();
-    return net::ERR_NOT_IMPLEMENTED;
-  }
-  int SetSendBufferSize(int32 size) override {
-    NOTIMPLEMENTED();
-    return net::ERR_NOT_IMPLEMENTED;
-  }
-
  private:
   scoped_refptr<net::IOBuffer> read_buffer_;
   int read_buffer_size_;
   net::CompletionCallback read_callback_;
 
   std::deque<std::vector<char> > incoming_packets_;
 
   FakeSocket* peer_socket_;
   RateLimiter* rate_limiter_;
   int latency_ms_;
 };
 
 class TCPChannelTester : public base::RefCountedThreadSafe<TCPChannelTester> {
  public:
   TCPChannelTester(base::MessageLoop* message_loop,
-                   net::Socket* client_socket,
-                   net::Socket* host_socket)
+                   P2PStreamSocket* client_socket,
+                   P2PStreamSocket* host_socket)
       : message_loop_(message_loop),
         host_socket_(host_socket),
         client_socket_(client_socket),
         done_(false),
         write_errors_(0),
         read_errors_(0) {}
 
   void Start() {
     message_loop_->PostTask(
         FROM_HERE, base::Bind(&TCPChannelTester::DoStart, this));
@@ skipping 97 matching lines @@
       input_buffer_->SetCapacity(input_buffer_->capacity() + result);
       if (input_buffer_->capacity() == kTestDataSize + kMessageSize)
         Done();
     }
   }
 
  private:
   friend class base::RefCountedThreadSafe<TCPChannelTester>;
 
   base::MessageLoop* message_loop_;
-  net::Socket* host_socket_;
-  net::Socket* client_socket_;
+  P2PStreamSocket* host_socket_;
+  P2PStreamSocket* client_socket_;
   bool done_;
 
   scoped_refptr<net::DrainableIOBuffer> output_buffer_;
   scoped_refptr<net::GrowableIOBuffer> input_buffer_;
 
   int write_errors_;
   int read_errors_;
 };
 
 class PseudoTcpAdapterTest : public testing::Test {
@@ skipping 135 matching lines @@
 
   tester->Start();
   message_loop_.Run();
   tester->CheckResults();
 }
 
 }  // namespace
 
 }  // namespace protocol
 }  // namespace remoting