OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "remoting/base/buffered_socket_writer.h" | 5 #include "remoting/base/buffered_socket_writer.h" |
6 | 6 |
7 #include <stdlib.h> | 7 #include <stdlib.h> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
(...skipping 75 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
86 } | 86 } |
87 | 87 |
88 protected: | 88 protected: |
89 void SetUp() override { | 89 void SetUp() override { |
90 socket_.reset(new net::MockTCPClientSocket(net::AddressList(), &net_log_, | 90 socket_.reset(new net::MockTCPClientSocket(net::AddressList(), &net_log_, |
91 &socket_data_provider_)); | 91 &socket_data_provider_)); |
92 socket_data_provider_.set_connect_data( | 92 socket_data_provider_.set_connect_data( |
93 net::MockConnect(net::SYNCHRONOUS, net::OK)); | 93 net::MockConnect(net::SYNCHRONOUS, net::OK)); |
94 EXPECT_EQ(net::OK, socket_->Connect(net::CompletionCallback())); | 94 EXPECT_EQ(net::OK, socket_->Connect(net::CompletionCallback())); |
95 | 95 |
96 writer_.reset(new BufferedSocketWriter()); | 96 writer_ = BufferedSocketWriter::CreateForSocket( |
97 writer_->Init(socket_.get(), base::Bind( | 97 socket_.get(), base::Bind(&BufferedSocketWriterTest::OnWriteFailed, |
98 &BufferedSocketWriterTest::OnWriteFailed, base::Unretained(this))); | 98 base::Unretained(this))); |
99 test_buffer_ = new net::IOBufferWithSize(kTestBufferSize); | 99 test_buffer_ = new net::IOBufferWithSize(kTestBufferSize); |
100 test_buffer_2_ = new net::IOBufferWithSize(kTestBufferSize); | 100 test_buffer_2_ = new net::IOBufferWithSize(kTestBufferSize); |
101 for (int i = 0; i< kTestBufferSize; ++i) { | 101 for (int i = 0; i< kTestBufferSize; ++i) { |
102 test_buffer_->data()[i] = rand() % 256; | 102 test_buffer_->data()[i] = rand() % 256; |
103 test_buffer_2_->data()[i] = rand() % 256; | 103 test_buffer_2_->data()[i] = rand() % 256; |
104 } | 104 } |
105 } | 105 } |
106 | 106 |
107 void OnWriteFailed(int error) { | 107 void OnWriteFailed(int error) { |
108 write_error_ = error; | 108 write_error_ = error; |
(...skipping 10 matching lines...) Expand all Loading... |
119 socket_data_provider_.written_data().data(), | 119 socket_data_provider_.written_data().data(), |
120 test_buffer_->size())); | 120 test_buffer_->size())); |
121 EXPECT_EQ(0, memcmp(test_buffer_2_->data(), | 121 EXPECT_EQ(0, memcmp(test_buffer_2_->data(), |
122 socket_data_provider_.written_data().data() + | 122 socket_data_provider_.written_data().data() + |
123 test_buffer_->size(), | 123 test_buffer_->size(), |
124 test_buffer_2_->size())); | 124 test_buffer_2_->size())); |
125 } | 125 } |
126 | 126 |
127 void TestAppendInCallback() { | 127 void TestAppendInCallback() { |
128 writer_->Write(test_buffer_, base::Bind( | 128 writer_->Write(test_buffer_, base::Bind( |
129 base::IgnoreResult(&BufferedSocketWriterBase::Write), | 129 base::IgnoreResult(&BufferedSocketWriter::Write), |
130 base::Unretained(writer_.get()), test_buffer_2_, | 130 base::Unretained(writer_.get()), test_buffer_2_, |
131 base::Closure())); | 131 base::Closure())); |
132 base::RunLoop().RunUntilIdle(); | 132 base::RunLoop().RunUntilIdle(); |
133 ASSERT_EQ(static_cast<size_t>(test_buffer_->size() + | 133 ASSERT_EQ(static_cast<size_t>(test_buffer_->size() + |
134 test_buffer_2_->size()), | 134 test_buffer_2_->size()), |
135 socket_data_provider_.written_data().size()); | 135 socket_data_provider_.written_data().size()); |
136 EXPECT_EQ(0, memcmp(test_buffer_->data(), | 136 EXPECT_EQ(0, memcmp(test_buffer_->data(), |
137 socket_data_provider_.written_data().data(), | 137 socket_data_provider_.written_data().data(), |
138 test_buffer_->size())); | 138 test_buffer_->size())); |
139 EXPECT_EQ(0, memcmp(test_buffer_2_->data(), | 139 EXPECT_EQ(0, memcmp(test_buffer_2_->data(), |
(...skipping 85 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
225 writer_->Write(test_buffer_2_, | 225 writer_->Write(test_buffer_2_, |
226 base::Bind(&BufferedSocketWriterTest::Unexpected, | 226 base::Bind(&BufferedSocketWriterTest::Unexpected, |
227 base::Unretained(this))); | 227 base::Unretained(this))); |
228 base::RunLoop().RunUntilIdle(); | 228 base::RunLoop().RunUntilIdle(); |
229 EXPECT_EQ(net::ERR_FAILED, write_error_); | 229 EXPECT_EQ(net::ERR_FAILED, write_error_); |
230 EXPECT_EQ(static_cast<size_t>(test_buffer_->size()), | 230 EXPECT_EQ(static_cast<size_t>(test_buffer_->size()), |
231 socket_data_provider_.written_data().size()); | 231 socket_data_provider_.written_data().size()); |
232 } | 232 } |
233 | 233 |
234 } // namespace remoting | 234 } // namespace remoting |
OLD | NEW |