Implement chrome.socket.bind/listen/accept for TCP server socket.

chrome/browser/extensions/api/socket/tcp_socket_unittest.cc

Index: chrome/browser/extensions/api/socket/tcp_socket_unittest.cc
diff --git a/chrome/browser/extensions/api/socket/tcp_socket_unittest.cc b/chrome/browser/extensions/api/socket/tcp_socket_unittest.cc
index 54fd8a9fa7f5a11275f88551e546c2ee2698f68b..5eec8c7b1d4d87a3a0b6bbad1980f2ee403b98cd 100644
--- a/chrome/browser/extensions/api/socket/tcp_socket_unittest.cc
+++ b/chrome/browser/extensions/api/socket/tcp_socket_unittest.cc
@@ -12,6 +12,7 @@
 #include "net/base/net_errors.h"
 #include "net/base/rand_callback.h"
 #include "net/socket/tcp_client_socket.h"
+#include "net/socket/tcp_server_socket.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 using testing::_;
@@ -41,9 +42,22 @@ class MockTCPSocket : public net::TCPClientSocket {
   DISALLOW_COPY_AND_ASSIGN(MockTCPSocket);
 };
 
+class MockTCPServerSocket : public net::TCPServerSocket {
+ public:
+  explicit MockTCPServerSocket()
+      : net::TCPServerSocket(NULL, net::NetLog::Source()) {
+  }
+  MOCK_METHOD2(Listen, int(const net::IPEndPoint& address, int backlog));
+  MOCK_METHOD2(Accept, int(scoped_ptr<net::StreamSocket>* socket,
+                            const net::CompletionCallback& callback));
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MockTCPServerSocket);
+};
+
 class MockApiResourceEventNotifier : public ApiResourceEventNotifier {
  public:
-  MockApiResourceEventNotifier(const std::string& owner_extension_id)
+  explicit MockApiResourceEventNotifier(const std::string& owner_extension_id)
       : ApiResourceEventNotifier(NULL, NULL,
                                  owner_extension_id,
                                  0, GURL()) {}
@@ -62,6 +76,7 @@ class CompleteHandler {
   MOCK_METHOD1(OnComplete, void(int result_code));
   MOCK_METHOD2(OnReadComplete, void(int result_code,
       scoped_refptr<net::IOBuffer> io_buffer));
+  MOCK_METHOD2(OnAccept, void(int, net::TCPClientSocket*));
  private:
   DISALLOW_COPY_AND_ASSIGN(CompleteHandler);
 };
@@ -227,4 +242,27 @@ TEST(SocketTest, TestTCPSocketSetKeepAlive) {
   EXPECT_EQ(0, delay);
 }
 
+TEST(SocketTest, TestTCPServerSocketListenAccept) {
+  MockTCPServerSocket* tcp_server_socket = new MockTCPServerSocket();
+  CompleteHandler handler;
+
+  scoped_ptr<TCPSocket> socket(TCPSocket::CreateServerSocketForTesting(
+      tcp_server_socket, FAKE_ID));
+
+  EXPECT_CALL(*tcp_server_socket, Accept(_, _)).Times(1);
+  EXPECT_CALL(handler, OnAccept(_, _));
+
+  // Should fail if we try to listen without calling bind() first.
+  std::string err_msg;
+  EXPECT_EQ(net::ERR_FAILED, socket->Listen(10, &err_msg));
+
+  EXPECT_EQ(net::OK, socket->Bind("127.0.0.1", 9999));
+  // This error is because we created a mock class with a TCPServerSocket
+  // already initialized (which means we are already listening), and we don't
+  // allow calling the listen function again.
+  EXPECT_EQ(net::ERR_ADDRESS_IN_USE, socket->Listen(10, &err_msg));
+  socket->Accept(base::Bind(&CompleteHandler::OnAccept,
+        base::Unretained(&handler)));
+}
+
 }  // namespace extensions