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" |
(...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
45 const std::string& message)); | 45 const std::string& message)); |
46 MOCK_METHOD1(OnWriteComplete, void(int result_code)); | 46 MOCK_METHOD1(OnWriteComplete, void(int result_code)); |
47 }; | 47 }; |
48 | 48 |
49 TEST(SocketTest, TestTCPSocketRead) { | 49 TEST(SocketTest, TestTCPSocketRead) { |
50 net::AddressList address_list; | 50 net::AddressList address_list; |
51 MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list); | 51 MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list); |
52 APIResourceEventNotifier* notifier = new MockAPIResourceEventNotifier(); | 52 APIResourceEventNotifier* notifier = new MockAPIResourceEventNotifier(); |
53 | 53 |
54 scoped_ptr<TCPSocket> socket(TCPSocket::CreateSocketForTesting( | 54 scoped_ptr<TCPSocket> socket(TCPSocket::CreateSocketForTesting( |
55 tcp_client_socket, "1.2.3.4", 1, notifier)); | 55 tcp_client_socket, notifier)); |
56 | 56 |
57 EXPECT_CALL(*tcp_client_socket, Read(_, _, _)) | 57 EXPECT_CALL(*tcp_client_socket, Read(_, _, _)) |
58 .Times(1); | 58 .Times(1); |
59 | 59 |
60 scoped_refptr<net::IOBufferWithSize> io_buffer( | 60 scoped_refptr<net::IOBufferWithSize> io_buffer( |
61 new net::IOBufferWithSize(512)); | 61 new net::IOBufferWithSize(512)); |
62 socket->Read(io_buffer.get(), io_buffer->size()); | 62 socket->Read(io_buffer.get(), io_buffer->size()); |
63 } | 63 } |
64 | 64 |
65 TEST(SocketTest, TestTCPSocketWrite) { | 65 TEST(SocketTest, TestTCPSocketWrite) { |
66 net::AddressList address_list; | 66 net::AddressList address_list; |
67 MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list); | 67 MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list); |
68 APIResourceEventNotifier* notifier = new MockAPIResourceEventNotifier(); | 68 APIResourceEventNotifier* notifier = new MockAPIResourceEventNotifier(); |
69 | 69 |
70 scoped_ptr<TCPSocket> socket(TCPSocket::CreateSocketForTesting( | 70 scoped_ptr<TCPSocket> socket(TCPSocket::CreateSocketForTesting( |
71 tcp_client_socket, "1.2.3.4", 1, notifier)); | 71 tcp_client_socket, notifier)); |
72 | 72 |
73 EXPECT_CALL(*tcp_client_socket, Write(_, _, _)) | 73 EXPECT_CALL(*tcp_client_socket, Write(_, _, _)) |
74 .Times(1); | 74 .Times(1); |
75 | 75 |
76 scoped_refptr<net::IOBufferWithSize> io_buffer( | 76 scoped_refptr<net::IOBufferWithSize> io_buffer( |
77 new net::IOBufferWithSize(256)); | 77 new net::IOBufferWithSize(256)); |
78 socket->Write(io_buffer.get(), io_buffer->size()); | 78 socket->Write(io_buffer.get(), io_buffer->size()); |
79 } | 79 } |
80 | 80 |
81 TEST(SocketTest, TestTCPSocketBlockedWrite) { | 81 TEST(SocketTest, TestTCPSocketBlockedWrite) { |
82 net::AddressList address_list; | 82 net::AddressList address_list; |
83 MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list); | 83 MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list); |
84 MockAPIResourceEventNotifier* notifier = new MockAPIResourceEventNotifier(); | 84 MockAPIResourceEventNotifier* notifier = new MockAPIResourceEventNotifier(); |
85 | 85 |
86 scoped_ptr<TCPSocket> socket(TCPSocket::CreateSocketForTesting( | 86 scoped_ptr<TCPSocket> socket(TCPSocket::CreateSocketForTesting( |
87 tcp_client_socket, "1.2.3.4", 1, notifier)); | 87 tcp_client_socket, notifier)); |
88 | 88 |
89 net::CompletionCallback callback; | 89 net::CompletionCallback callback; |
90 EXPECT_CALL(*tcp_client_socket, Write(_, _, _)) | 90 EXPECT_CALL(*tcp_client_socket, Write(_, _, _)) |
91 .Times(1) | 91 .Times(1) |
92 .WillOnce(testing::DoAll(SaveArg<2>(&callback), | 92 .WillOnce(testing::DoAll(SaveArg<2>(&callback), |
93 Return(net::ERR_IO_PENDING))); | 93 Return(net::ERR_IO_PENDING))); |
94 | 94 |
95 scoped_refptr<net::IOBufferWithSize> io_buffer(new net::IOBufferWithSize( | 95 scoped_refptr<net::IOBufferWithSize> io_buffer(new net::IOBufferWithSize( |
96 1)); | 96 1)); |
97 ASSERT_EQ(net::ERR_IO_PENDING, socket->Write(io_buffer.get(), | 97 ASSERT_EQ(net::ERR_IO_PENDING, socket->Write(io_buffer.get(), |
98 io_buffer->size())); | 98 io_buffer->size())); |
99 | 99 |
100 // Good. Original call came back unable to complete. Now pretend the socket | 100 // Good. Original call came back unable to complete. Now pretend the socket |
101 // finished, and confirm that we passed the error back. | 101 // finished, and confirm that we passed the error back. |
102 EXPECT_CALL(*notifier, OnWriteComplete(42)) | 102 EXPECT_CALL(*notifier, OnWriteComplete(42)) |
103 .Times(1); | 103 .Times(1); |
104 callback.Run(42); | 104 callback.Run(42); |
105 } | 105 } |
106 | 106 |
107 } // namespace extensions | 107 } // namespace extensions |
OLD | NEW |