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 76 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
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_.reset(new BufferedSocketWriter()); |
97 writer_->Init(socket_.get(), base::Bind( | 97 writer_->Init( |
98 &BufferedSocketWriterTest::OnWriteFailed, base::Unretained(this))); | 98 base::Bind(&net::Socket::Write, base::Unretained(socket_.get())), |
| 99 base::Bind(&BufferedSocketWriterTest::OnWriteFailed, |
| 100 base::Unretained(this))); |
99 test_buffer_ = new net::IOBufferWithSize(kTestBufferSize); | 101 test_buffer_ = new net::IOBufferWithSize(kTestBufferSize); |
100 test_buffer_2_ = new net::IOBufferWithSize(kTestBufferSize); | 102 test_buffer_2_ = new net::IOBufferWithSize(kTestBufferSize); |
101 for (int i = 0; i< kTestBufferSize; ++i) { | 103 for (int i = 0; i< kTestBufferSize; ++i) { |
102 test_buffer_->data()[i] = rand() % 256; | 104 test_buffer_->data()[i] = rand() % 256; |
103 test_buffer_2_->data()[i] = rand() % 256; | 105 test_buffer_2_->data()[i] = rand() % 256; |
104 } | 106 } |
105 } | 107 } |
106 | 108 |
107 void OnWriteFailed(int error) { | 109 void OnWriteFailed(int error) { |
108 write_error_ = error; | 110 write_error_ = error; |
(...skipping 116 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
225 writer_->Write(test_buffer_2_, | 227 writer_->Write(test_buffer_2_, |
226 base::Bind(&BufferedSocketWriterTest::Unexpected, | 228 base::Bind(&BufferedSocketWriterTest::Unexpected, |
227 base::Unretained(this))); | 229 base::Unretained(this))); |
228 base::RunLoop().RunUntilIdle(); | 230 base::RunLoop().RunUntilIdle(); |
229 EXPECT_EQ(net::ERR_FAILED, write_error_); | 231 EXPECT_EQ(net::ERR_FAILED, write_error_); |
230 EXPECT_EQ(static_cast<size_t>(test_buffer_->size()), | 232 EXPECT_EQ(static_cast<size_t>(test_buffer_->size()), |
231 socket_data_provider_.written_data().size()); | 233 socket_data_provider_.written_data().size()); |
232 } | 234 } |
233 | 235 |
234 } // namespace remoting | 236 } // namespace remoting |
OLD | NEW |