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 "net/socket/buffered_write_stream_socket.h" | 5 #include "net/socket/buffered_write_stream_socket.h" |
6 | 6 |
7 #include "base/message_loop.h" | 7 #include "base/message_loop.h" |
8 #include "base/memory/ref_counted.h" | 8 #include "base/memory/ref_counted.h" |
| 9 #include "base/memory/scoped_ptr.h" |
9 #include "net/base/net_errors.h" | 10 #include "net/base/net_errors.h" |
10 #include "net/base/net_log.h" | 11 #include "net/base/net_log.h" |
11 #include "net/socket/socket_test_util.h" | 12 #include "net/socket/socket_test_util.h" |
12 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
13 | 14 |
14 namespace net { | 15 namespace net { |
15 | 16 |
16 namespace { | 17 namespace { |
17 | 18 |
18 class BufferedWriteStreamSocketTest : public testing::Test { | 19 class BufferedWriteStreamSocketTest : public testing::Test { |
19 public: | 20 public: |
20 void Finish() { | 21 void Finish() { |
21 MessageLoop::current()->RunAllPending(); | 22 MessageLoop::current()->RunAllPending(); |
22 EXPECT_TRUE(data_->at_read_eof()); | 23 EXPECT_TRUE(data_->at_read_eof()); |
23 EXPECT_TRUE(data_->at_write_eof()); | 24 EXPECT_TRUE(data_->at_write_eof()); |
24 } | 25 } |
25 | 26 |
26 void Initialize(MockWrite* writes, size_t writes_count) { | 27 void Initialize(MockWrite* writes, size_t writes_count) { |
27 data_ = new DeterministicSocketData(NULL, 0, writes, writes_count); | 28 data_.reset(new DeterministicSocketData(NULL, 0, writes, writes_count)); |
28 data_->set_connect_data(MockConnect(SYNCHRONOUS, 0)); | 29 data_->set_connect_data(MockConnect(SYNCHRONOUS, 0)); |
29 if (writes_count) { | 30 if (writes_count) { |
30 data_->StopAfter(writes_count); | 31 data_->StopAfter(writes_count); |
31 } | 32 } |
32 DeterministicMockTCPClientSocket* wrapped_socket = | 33 DeterministicMockTCPClientSocket* wrapped_socket = |
33 new DeterministicMockTCPClientSocket(net_log_.net_log(), data_.get()); | 34 new DeterministicMockTCPClientSocket(net_log_.net_log(), data_.get()); |
34 data_->set_socket(wrapped_socket->AsWeakPtr()); | 35 data_->set_socket(wrapped_socket->AsWeakPtr()); |
35 socket_.reset(new BufferedWriteStreamSocket(wrapped_socket)); | 36 socket_.reset(new BufferedWriteStreamSocket(wrapped_socket)); |
36 socket_->Connect(callback_.callback()); | 37 socket_->Connect(callback_.callback()); |
37 } | 38 } |
38 | 39 |
39 void TestWrite(const char* text) { | 40 void TestWrite(const char* text) { |
40 scoped_refptr<StringIOBuffer> buf(new StringIOBuffer(text)); | 41 scoped_refptr<StringIOBuffer> buf(new StringIOBuffer(text)); |
41 EXPECT_EQ(buf->size(), | 42 EXPECT_EQ(buf->size(), |
42 socket_->Write(buf.get(), buf->size(), callback_.callback())); | 43 socket_->Write(buf.get(), buf->size(), callback_.callback())); |
43 } | 44 } |
44 | 45 |
45 scoped_ptr<BufferedWriteStreamSocket> socket_; | 46 scoped_ptr<BufferedWriteStreamSocket> socket_; |
46 scoped_refptr<DeterministicSocketData> data_; | 47 scoped_ptr<DeterministicSocketData> data_; |
47 BoundNetLog net_log_; | 48 BoundNetLog net_log_; |
48 TestCompletionCallback callback_; | 49 TestCompletionCallback callback_; |
49 }; | 50 }; |
50 | 51 |
51 TEST_F(BufferedWriteStreamSocketTest, SingleWrite) { | 52 TEST_F(BufferedWriteStreamSocketTest, SingleWrite) { |
52 MockWrite writes[] = { | 53 MockWrite writes[] = { |
53 MockWrite(SYNCHRONOUS, 0, "abc"), | 54 MockWrite(SYNCHRONOUS, 0, "abc"), |
54 }; | 55 }; |
55 Initialize(writes, arraysize(writes)); | 56 Initialize(writes, arraysize(writes)); |
56 TestWrite("abc"); | 57 TestWrite("abc"); |
(...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
113 TestWrite("abc"); | 114 TestWrite("abc"); |
114 data_->RunFor(1); | 115 data_->RunFor(1); |
115 TestWrite("def"); | 116 TestWrite("def"); |
116 data_->RunFor(1); | 117 data_->RunFor(1); |
117 Finish(); | 118 Finish(); |
118 } | 119 } |
119 | 120 |
120 } // anonymous namespace | 121 } // anonymous namespace |
121 | 122 |
122 } // namespace net | 123 } // namespace net |
OLD | NEW |