| 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/protocol/channel_socket_adapter.h" | 5 #include "remoting/protocol/channel_socket_adapter.h" |
| 6 | 6 |
| 7 #include "base/memory/ref_counted.h" | 7 #include "base/memory/ref_counted.h" |
| 8 #include "base/memory/scoped_ptr.h" | 8 #include "base/memory/scoped_ptr.h" |
| 9 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
| 10 #include "net/base/io_buffer.h" | 10 #include "net/base/io_buffer.h" |
| (...skipping 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 81 scoped_ptr<TransportChannelSocketAdapter> target_; | 81 scoped_ptr<TransportChannelSocketAdapter> target_; |
| 82 net::CompletionCallback callback_; | 82 net::CompletionCallback callback_; |
| 83 int callback_result_; | 83 int callback_result_; |
| 84 base::MessageLoopForIO message_loop_; | 84 base::MessageLoopForIO message_loop_; |
| 85 }; | 85 }; |
| 86 | 86 |
| 87 // Verify that Read() returns net::ERR_IO_PENDING. | 87 // Verify that Read() returns net::ERR_IO_PENDING. |
| 88 TEST_F(TransportChannelSocketAdapterTest, Read) { | 88 TEST_F(TransportChannelSocketAdapterTest, Read) { |
| 89 scoped_refptr<IOBuffer> buffer(new IOBuffer(kBufferSize)); | 89 scoped_refptr<IOBuffer> buffer(new IOBuffer(kBufferSize)); |
| 90 | 90 |
| 91 int result = target_->Read(buffer.get(), kBufferSize, callback_); | 91 int result = target_->Recv(buffer.get(), kBufferSize, callback_); |
| 92 ASSERT_EQ(net::ERR_IO_PENDING, result); | 92 ASSERT_EQ(net::ERR_IO_PENDING, result); |
| 93 | 93 |
| 94 channel_.SignalReadPacket(&channel_, kTestData, kTestDataSize, | 94 channel_.SignalReadPacket(&channel_, kTestData, kTestDataSize, |
| 95 rtc::CreatePacketTime(0), 0); | 95 rtc::CreatePacketTime(0), 0); |
| 96 EXPECT_EQ(kTestDataSize, callback_result_); | 96 EXPECT_EQ(kTestDataSize, callback_result_); |
| 97 } | 97 } |
| 98 | 98 |
| 99 // Verify that Read() after Close() returns error. | 99 // Verify that Read() after Close() returns error. |
| 100 TEST_F(TransportChannelSocketAdapterTest, ReadClose) { | 100 TEST_F(TransportChannelSocketAdapterTest, ReadClose) { |
| 101 scoped_refptr<IOBuffer> buffer(new IOBuffer(kBufferSize)); | 101 scoped_refptr<IOBuffer> buffer(new IOBuffer(kBufferSize)); |
| 102 | 102 |
| 103 int result = target_->Read(buffer.get(), kBufferSize, callback_); | 103 int result = target_->Recv(buffer.get(), kBufferSize, callback_); |
| 104 ASSERT_EQ(net::ERR_IO_PENDING, result); | 104 ASSERT_EQ(net::ERR_IO_PENDING, result); |
| 105 | 105 |
| 106 target_->Close(kTestError); | 106 target_->Close(kTestError); |
| 107 EXPECT_EQ(kTestError, callback_result_); | 107 EXPECT_EQ(kTestError, callback_result_); |
| 108 | 108 |
| 109 // All Read() calls after Close() should return the error. | 109 // All Recv() calls after Close() should return the error. |
| 110 EXPECT_EQ(kTestError, target_->Read(buffer.get(), kBufferSize, callback_)); | 110 EXPECT_EQ(kTestError, target_->Recv(buffer.get(), kBufferSize, callback_)); |
| 111 } | 111 } |
| 112 | 112 |
| 113 // Verify that Write sends the packet and returns correct result. | 113 // Verify that Send sends the packet and returns correct result. |
| 114 TEST_F(TransportChannelSocketAdapterTest, Write) { | 114 TEST_F(TransportChannelSocketAdapterTest, Send) { |
| 115 scoped_refptr<IOBuffer> buffer(new IOBuffer(kTestDataSize)); | 115 scoped_refptr<IOBuffer> buffer(new IOBuffer(kTestDataSize)); |
| 116 | 116 |
| 117 EXPECT_CALL(channel_, SendPacket(buffer->data(), kTestDataSize, _, 0)) | 117 EXPECT_CALL(channel_, SendPacket(buffer->data(), kTestDataSize, _, 0)) |
| 118 .WillOnce(Return(kTestDataSize)); | 118 .WillOnce(Return(kTestDataSize)); |
| 119 | 119 |
| 120 int result = target_->Write(buffer.get(), kTestDataSize, callback_); | 120 int result = target_->Send(buffer.get(), kTestDataSize, callback_); |
| 121 EXPECT_EQ(kTestDataSize, result); | 121 EXPECT_EQ(kTestDataSize, result); |
| 122 } | 122 } |
| 123 | 123 |
| 124 // Verify that the message is still sent if Write() is called while | 124 // Verify that the message is still sent if Send() is called while |
| 125 // socket is not open yet. The result is the packet is lost. | 125 // socket is not open yet. The result is the packet is lost. |
| 126 TEST_F(TransportChannelSocketAdapterTest, WritePending) { | 126 TEST_F(TransportChannelSocketAdapterTest, SendPending) { |
| 127 scoped_refptr<IOBuffer> buffer(new IOBuffer(kTestDataSize)); | 127 scoped_refptr<IOBuffer> buffer(new IOBuffer(kTestDataSize)); |
| 128 | 128 |
| 129 EXPECT_CALL(channel_, SendPacket(buffer->data(), kTestDataSize, _, 0)) | 129 EXPECT_CALL(channel_, SendPacket(buffer->data(), kTestDataSize, _, 0)) |
| 130 .Times(1) | 130 .Times(1) |
| 131 .WillOnce(Return(SOCKET_ERROR)); | 131 .WillOnce(Return(SOCKET_ERROR)); |
| 132 | 132 |
| 133 EXPECT_CALL(channel_, GetError()) | 133 EXPECT_CALL(channel_, GetError()) |
| 134 .WillOnce(Return(EWOULDBLOCK)); | 134 .WillOnce(Return(EWOULDBLOCK)); |
| 135 | 135 |
| 136 int result = target_->Write(buffer.get(), kTestDataSize, callback_); | 136 int result = target_->Send(buffer.get(), kTestDataSize, callback_); |
| 137 ASSERT_EQ(net::OK, result); | 137 ASSERT_EQ(net::OK, result); |
| 138 } | 138 } |
| 139 | 139 |
| 140 } // namespace protocol | 140 } // namespace protocol |
| 141 } // namespace remoting | 141 } // namespace remoting |
| OLD | NEW |