| 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 |