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 "remoting/protocol/buffered_socket_writer.h" | 5 #include "remoting/protocol/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 30 matching lines...) Expand all Loading... |
41 base::MessageLoop::current()->PostTask(FROM_HERE, | 41 base::MessageLoop::current()->PostTask(FROM_HERE, |
42 base::MessageLoop::QuitClosure()); | 42 base::MessageLoop::QuitClosure()); |
43 } | 43 } |
44 | 44 |
45 void Unexpected() { | 45 void Unexpected() { |
46 EXPECT_TRUE(false); | 46 EXPECT_TRUE(false); |
47 } | 47 } |
48 | 48 |
49 protected: | 49 protected: |
50 virtual void SetUp() OVERRIDE { | 50 virtual void SetUp() OVERRIDE { |
51 socket_.reset(new FakeSocket()); | 51 socket_.reset(new FakeStreamSocket()); |
52 writer_.reset(new BufferedSocketWriter()); | 52 writer_.reset(new BufferedSocketWriter()); |
53 writer_->Init(socket_.get(), base::Bind( | 53 writer_->Init(socket_.get(), base::Bind( |
54 &BufferedSocketWriterTest::OnWriteFailed, base::Unretained(this))); | 54 &BufferedSocketWriterTest::OnWriteFailed, base::Unretained(this))); |
55 test_buffer_ = new net::IOBufferWithSize(kTestBufferSize); | 55 test_buffer_ = new net::IOBufferWithSize(kTestBufferSize); |
56 test_buffer_2_ = new net::IOBufferWithSize(kTestBufferSize); | 56 test_buffer_2_ = new net::IOBufferWithSize(kTestBufferSize); |
57 for (int i = 0; i< kTestBufferSize; ++i) { | 57 for (int i = 0; i< kTestBufferSize; ++i) { |
58 test_buffer_->data()[i] = rand() % 256; | 58 test_buffer_->data()[i] = rand() % 256; |
59 test_buffer_2_->data()[i] = rand() % 256; | 59 test_buffer_2_->data()[i] = rand() % 256; |
60 } | 60 } |
61 } | 61 } |
(...skipping 29 matching lines...) Expand all Loading... |
91 test_buffer_2_->size()), | 91 test_buffer_2_->size()), |
92 socket_->written_data().size()); | 92 socket_->written_data().size()); |
93 EXPECT_EQ(0, memcmp(test_buffer_->data(), socket_->written_data().data(), | 93 EXPECT_EQ(0, memcmp(test_buffer_->data(), socket_->written_data().data(), |
94 test_buffer_->size())); | 94 test_buffer_->size())); |
95 EXPECT_EQ(0, memcmp(test_buffer_2_->data(), | 95 EXPECT_EQ(0, memcmp(test_buffer_2_->data(), |
96 socket_->written_data().data() + test_buffer_->size(), | 96 socket_->written_data().data() + test_buffer_->size(), |
97 test_buffer_2_->size())); | 97 test_buffer_2_->size())); |
98 } | 98 } |
99 | 99 |
100 base::MessageLoop message_loop_; | 100 base::MessageLoop message_loop_; |
101 scoped_ptr<FakeSocket> socket_; | 101 scoped_ptr<FakeStreamSocket> socket_; |
102 scoped_ptr<BufferedSocketWriter> writer_; | 102 scoped_ptr<BufferedSocketWriter> writer_; |
103 scoped_refptr<net::IOBufferWithSize> test_buffer_; | 103 scoped_refptr<net::IOBufferWithSize> test_buffer_; |
104 scoped_refptr<net::IOBufferWithSize> test_buffer_2_; | 104 scoped_refptr<net::IOBufferWithSize> test_buffer_2_; |
105 std::string written_data_; | 105 std::string written_data_; |
106 int write_error_; | 106 int write_error_; |
107 }; | 107 }; |
108 | 108 |
109 // Test synchronous write. | 109 // Test synchronous write. |
110 TEST_F(BufferedSocketWriterTest, WriteFull) { | 110 TEST_F(BufferedSocketWriterTest, WriteFull) { |
111 TestWrite(); | 111 TestWrite(); |
(...skipping 68 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
180 socket_->set_next_write_error(net::ERR_FAILED); | 180 socket_->set_next_write_error(net::ERR_FAILED); |
181 base::RunLoop().RunUntilIdle(); | 181 base::RunLoop().RunUntilIdle(); |
182 EXPECT_EQ(net::ERR_FAILED, write_error_); | 182 EXPECT_EQ(net::ERR_FAILED, write_error_); |
183 EXPECT_EQ(static_cast<size_t>(test_buffer_->size()), | 183 EXPECT_EQ(static_cast<size_t>(test_buffer_->size()), |
184 socket_->written_data().size()); | 184 socket_->written_data().size()); |
185 } | 185 } |
186 | 186 |
187 } // namespace protocol | 187 } // namespace protocol |
188 } // namespace remoting | 188 } // namespace remoting |
189 | 189 |
OLD | NEW |