OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/extensions/api/socket/tcp_socket.h" | 5 #include "chrome/browser/extensions/api/socket/tcp_socket.h" |
6 | 6 |
7 #include "base/memory/scoped_ptr.h" | 7 #include "base/memory/scoped_ptr.h" |
8 #include "chrome/browser/extensions/api/api_resource_event_notifier.h" | 8 #include "chrome/browser/extensions/api/api_resource_event_notifier.h" |
9 #include "net/base/address_list.h" | 9 #include "net/base/address_list.h" |
10 #include "net/base/completion_callback.h" | 10 #include "net/base/completion_callback.h" |
11 #include "net/base/io_buffer.h" | 11 #include "net/base/io_buffer.h" |
12 #include "net/base/net_errors.h" | 12 #include "net/base/net_errors.h" |
13 #include "net/base/rand_callback.h" | 13 #include "net/base/rand_callback.h" |
14 #include "net/socket/tcp_client_socket.h" | 14 #include "net/socket/tcp_client_socket.h" |
| 15 #include "net/socket/tcp_server_socket.h" |
15 #include "testing/gmock/include/gmock/gmock.h" | 16 #include "testing/gmock/include/gmock/gmock.h" |
16 | 17 |
17 using testing::_; | 18 using testing::_; |
18 using testing::DoAll; | 19 using testing::DoAll; |
19 using testing::Return; | 20 using testing::Return; |
20 using testing::SaveArg; | 21 using testing::SaveArg; |
21 | 22 |
22 namespace extensions { | 23 namespace extensions { |
23 | 24 |
24 class MockTCPSocket : public net::TCPClientSocket { | 25 class MockTCPSocket : public net::TCPClientSocket { |
25 public: | 26 public: |
26 explicit MockTCPSocket(const net::AddressList& address_list) | 27 explicit MockTCPSocket(const net::AddressList& address_list) |
27 : net::TCPClientSocket(address_list, NULL, net::NetLog::Source()) { | 28 : net::TCPClientSocket(address_list, NULL, net::NetLog::Source()) { |
28 } | 29 } |
29 | 30 |
30 MOCK_METHOD3(Read, int(net::IOBuffer* buf, int buf_len, | 31 MOCK_METHOD3(Read, int(net::IOBuffer* buf, int buf_len, |
31 const net::CompletionCallback& callback)); | 32 const net::CompletionCallback& callback)); |
32 MOCK_METHOD3(Write, int(net::IOBuffer* buf, int buf_len, | 33 MOCK_METHOD3(Write, int(net::IOBuffer* buf, int buf_len, |
33 const net::CompletionCallback& callback)); | 34 const net::CompletionCallback& callback)); |
34 MOCK_METHOD2(SetKeepAlive, bool(bool enable, int delay)); | 35 MOCK_METHOD2(SetKeepAlive, bool(bool enable, int delay)); |
35 MOCK_METHOD1(SetNoDelay, bool(bool no_delay)); | 36 MOCK_METHOD1(SetNoDelay, bool(bool no_delay)); |
36 virtual bool IsConnected() const OVERRIDE { | 37 virtual bool IsConnected() const OVERRIDE { |
37 return true; | 38 return true; |
38 } | 39 } |
39 | 40 |
40 private: | 41 private: |
41 DISALLOW_COPY_AND_ASSIGN(MockTCPSocket); | 42 DISALLOW_COPY_AND_ASSIGN(MockTCPSocket); |
42 }; | 43 }; |
43 | 44 |
| 45 class MockTCPServerSocket : public net::TCPServerSocket { |
| 46 public: |
| 47 explicit MockTCPServerSocket() |
| 48 : net::TCPServerSocket(NULL, net::NetLog::Source()) { |
| 49 } |
| 50 MOCK_METHOD2(Listen, int(const net::IPEndPoint& address, int backlog)); |
| 51 MOCK_METHOD2(Accept, int(scoped_ptr<net::StreamSocket>* socket, |
| 52 const net::CompletionCallback& callback)); |
| 53 |
| 54 private: |
| 55 DISALLOW_COPY_AND_ASSIGN(MockTCPServerSocket); |
| 56 }; |
| 57 |
44 class MockApiResourceEventNotifier : public ApiResourceEventNotifier { | 58 class MockApiResourceEventNotifier : public ApiResourceEventNotifier { |
45 public: | 59 public: |
46 MockApiResourceEventNotifier() : ApiResourceEventNotifier(NULL, NULL, | 60 MockApiResourceEventNotifier() : ApiResourceEventNotifier(NULL, NULL, |
47 std::string(), | 61 std::string(), |
48 0, GURL()) {} | 62 0, GURL()) {} |
49 | 63 |
50 MOCK_METHOD2(OnReadComplete, void(int result_code, | 64 MOCK_METHOD2(OnReadComplete, void(int result_code, |
51 const std::string& message)); | 65 const std::string& message)); |
52 MOCK_METHOD1(OnWriteComplete, void(int result_code)); | 66 MOCK_METHOD1(OnWriteComplete, void(int result_code)); |
53 | 67 |
54 protected: | 68 protected: |
55 virtual ~MockApiResourceEventNotifier() {} | 69 virtual ~MockApiResourceEventNotifier() {} |
56 }; | 70 }; |
57 | 71 |
58 class CompleteHandler { | 72 class CompleteHandler { |
59 public: | 73 public: |
60 CompleteHandler() {} | 74 CompleteHandler() {} |
61 MOCK_METHOD1(OnComplete, void(int result_code)); | 75 MOCK_METHOD1(OnComplete, void(int result_code)); |
62 MOCK_METHOD2(OnReadComplete, void(int result_code, | 76 MOCK_METHOD2(OnReadComplete, void(int result_code, |
63 scoped_refptr<net::IOBuffer> io_buffer)); | 77 scoped_refptr<net::IOBuffer> io_buffer)); |
| 78 MOCK_METHOD2(OnAccept, void(int, net::TCPClientSocket*)); |
64 private: | 79 private: |
65 DISALLOW_COPY_AND_ASSIGN(CompleteHandler); | 80 DISALLOW_COPY_AND_ASSIGN(CompleteHandler); |
66 }; | 81 }; |
67 | 82 |
68 | 83 |
69 TEST(SocketTest, TestTCPSocketRead) { | 84 TEST(SocketTest, TestTCPSocketRead) { |
70 net::AddressList address_list; | 85 net::AddressList address_list; |
71 MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list); | 86 MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list); |
72 ApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier(); | 87 ApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier(); |
73 CompleteHandler handler; | 88 CompleteHandler handler; |
(...skipping 138 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
212 EXPECT_CALL(*tcp_client_socket, SetKeepAlive(_, _)) | 227 EXPECT_CALL(*tcp_client_socket, SetKeepAlive(_, _)) |
213 .WillOnce(testing::DoAll(SaveArg<0>(&enable), | 228 .WillOnce(testing::DoAll(SaveArg<0>(&enable), |
214 SaveArg<1>(&delay), | 229 SaveArg<1>(&delay), |
215 Return(false))); | 230 Return(false))); |
216 result = socket->SetKeepAlive(false, 0); | 231 result = socket->SetKeepAlive(false, 0); |
217 EXPECT_FALSE(result); | 232 EXPECT_FALSE(result); |
218 EXPECT_FALSE(enable); | 233 EXPECT_FALSE(enable); |
219 EXPECT_EQ(0, delay); | 234 EXPECT_EQ(0, delay); |
220 } | 235 } |
221 | 236 |
| 237 TEST(SocketTest, TestTCPServerSocketListenAccept) { |
| 238 MockTCPServerSocket* tcp_server_socket = new MockTCPServerSocket(); |
| 239 CompleteHandler handler; |
| 240 |
| 241 scoped_ptr<TCPSocket> socket(TCPSocket::CreateServerSocketForTesting( |
| 242 tcp_server_socket)); |
| 243 |
| 244 EXPECT_CALL(*tcp_server_socket, Listen(_, _)).Times(1); |
| 245 EXPECT_CALL(*tcp_server_socket, Accept(_, _)).Times(1); |
| 246 EXPECT_CALL(handler, OnAccept(_, _)); |
| 247 |
| 248 // Should fail if we try to listen without calling bind() first. |
| 249 std::string err_msg; |
| 250 EXPECT_EQ(net::ERR_FAILED, socket->Listen(10, err_msg)); |
| 251 |
| 252 EXPECT_TRUE(socket->Bind("127.0.0.1", 9999)); |
| 253 EXPECT_EQ(0, socket->Listen(10, err_msg)); |
| 254 socket->Accept(base::Bind(&CompleteHandler::OnAccept, |
| 255 base::Unretained(&handler))); |
| 256 } |
| 257 |
222 } // namespace extensions | 258 } // namespace extensions |
OLD | NEW |