| 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 16ae87967d50ba59a2c58493410d0759be5257b4..f8819faee4fc2f9b7402dd47ef399059e790da36 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,6 +42,19 @@ 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() : ApiResourceEventNotifier(NULL, NULL,
|
| @@ -61,6 +75,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);
|
| };
|
| @@ -219,4 +234,25 @@ 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));
|
| +
|
| + EXPECT_CALL(*tcp_server_socket, Listen(_, _)).Times(1);
|
| + 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_TRUE(socket->Bind("127.0.0.1", 9999));
|
| + EXPECT_EQ(0, socket->Listen(10, err_msg));
|
| + socket->Accept(base::Bind(&CompleteHandler::OnAccept,
|
| + base::Unretained(&handler)));
|
| +}
|
| +
|
| } // namespace extensions
|
|
|