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 <string> | 5 #include <string> |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/bind_helpers.h" | 8 #include "base/bind_helpers.h" |
9 #include "base/callback.h" | 9 #include "base/callback.h" |
10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
11 #include "base/run_loop.h" | 11 #include "base/run_loop.h" |
12 #include "base/stl_util.h" | |
13 #include "base/synchronization/waitable_event.h" | 12 #include "base/synchronization/waitable_event.h" |
14 #include "net/base/net_errors.h" | 13 #include "net/base/net_errors.h" |
15 #include "net/socket/socket.h" | 14 #include "net/socket/socket.h" |
16 #include "remoting/protocol/fake_stream_socket.h" | 15 #include "remoting/protocol/fake_stream_socket.h" |
17 #include "remoting/protocol/message_reader.h" | 16 #include "remoting/protocol/message_reader.h" |
18 #include "testing/gmock/include/gmock/gmock.h" | 17 #include "testing/gmock/include/gmock/gmock.h" |
19 #include "testing/gtest/include/gtest/gtest.h" | 18 #include "testing/gtest/include/gtest/gtest.h" |
20 #include "third_party/webrtc/base/byteorder.h" | 19 #include "third_party/webrtc/base/byteorder.h" |
21 | 20 |
22 using testing::_; | 21 using testing::_; |
(...skipping 20 matching lines...) Expand all Loading... |
43 void AddSecondMessage() { AddMessage(kTestMessage2); } | 42 void AddSecondMessage() { AddMessage(kTestMessage2); } |
44 | 43 |
45 // Used by the DeleteFromCallback() test. | 44 // Used by the DeleteFromCallback() test. |
46 void DeleteReader() { reader_.reset(); } | 45 void DeleteReader() { reader_.reset(); } |
47 | 46 |
48 protected: | 47 protected: |
49 void SetUp() override { | 48 void SetUp() override { |
50 reader_.reset(new MessageReader()); | 49 reader_.reset(new MessageReader()); |
51 } | 50 } |
52 | 51 |
53 void TearDown() override { base::STLDeleteElements(&messages_); } | |
54 | |
55 void InitReader() { | 52 void InitReader() { |
56 reader_->StartReading( | 53 reader_->StartReading( |
57 &socket_, | 54 &socket_, |
58 base::Bind(&MessageReaderTest::OnMessage, base::Unretained(this)), | 55 base::Bind(&MessageReaderTest::OnMessage, base::Unretained(this)), |
59 base::Bind(&MessageReaderTest::OnReadError, base::Unretained(this))); | 56 base::Bind(&MessageReaderTest::OnReadError, base::Unretained(this))); |
60 } | 57 } |
61 | 58 |
62 void AddMessage(const std::string& message) { | 59 void AddMessage(const std::string& message) { |
63 std::string data = std::string(4, ' ') + message; | 60 std::string data = std::string(4, ' ') + message; |
64 rtc::SetBE32(const_cast<char*>(data.data()), message.size()); | 61 rtc::SetBE32(const_cast<char*>(data.data()), message.size()); |
65 | 62 |
66 socket_.AppendInputData(data); | 63 socket_.AppendInputData(data); |
67 } | 64 } |
68 | 65 |
69 bool CompareResult(CompoundBuffer* buffer, const std::string& expected) { | 66 bool CompareResult(CompoundBuffer* buffer, const std::string& expected) { |
70 std::string result(buffer->total_bytes(), ' '); | 67 std::string result(buffer->total_bytes(), ' '); |
71 buffer->CopyTo(const_cast<char*>(result.data()), result.size()); | 68 buffer->CopyTo(const_cast<char*>(result.data()), result.size()); |
72 return result == expected; | 69 return result == expected; |
73 } | 70 } |
74 | 71 |
75 void OnReadError(int error) { | 72 void OnReadError(int error) { |
76 read_error_ = error; | 73 read_error_ = error; |
77 reader_.reset(); | 74 reader_.reset(); |
78 } | 75 } |
79 | 76 |
80 void OnMessage(std::unique_ptr<CompoundBuffer> buffer) { | 77 void OnMessage(std::unique_ptr<CompoundBuffer> buffer) { |
81 messages_.push_back(buffer.release()); | 78 messages_.push_back(std::move(buffer)); |
82 callback_.OnMessage(); | 79 callback_.OnMessage(); |
83 } | 80 } |
84 | 81 |
85 base::MessageLoop message_loop_; | 82 base::MessageLoop message_loop_; |
86 std::unique_ptr<MessageReader> reader_; | 83 std::unique_ptr<MessageReader> reader_; |
87 FakeStreamSocket socket_; | 84 FakeStreamSocket socket_; |
88 MockMessageReceivedCallback callback_; | 85 MockMessageReceivedCallback callback_; |
89 int read_error_ = 0; | 86 int read_error_ = 0; |
90 std::vector<CompoundBuffer*> messages_; | 87 std::vector<std::unique_ptr<CompoundBuffer>> messages_; |
91 }; | 88 }; |
92 | 89 |
93 // Receive one message. | 90 // Receive one message. |
94 TEST_F(MessageReaderTest, OneMessage) { | 91 TEST_F(MessageReaderTest, OneMessage) { |
95 AddMessage(kTestMessage1); | 92 AddMessage(kTestMessage1); |
96 | 93 |
97 EXPECT_CALL(callback_, OnMessage()).Times(1); | 94 EXPECT_CALL(callback_, OnMessage()).Times(1); |
98 | 95 |
99 InitReader(); | 96 InitReader(); |
100 base::RunLoop().RunUntilIdle(); | 97 base::RunLoop().RunUntilIdle(); |
101 | 98 |
102 EXPECT_TRUE(socket_.read_pending()); | 99 EXPECT_TRUE(socket_.read_pending()); |
103 EXPECT_EQ(1U, messages_.size()); | 100 EXPECT_EQ(1U, messages_.size()); |
104 } | 101 } |
105 | 102 |
106 // Receive two messages in one packet. | 103 // Receive two messages in one packet. |
107 TEST_F(MessageReaderTest, TwoMessages_Together) { | 104 TEST_F(MessageReaderTest, TwoMessages_Together) { |
108 AddMessage(kTestMessage1); | 105 AddMessage(kTestMessage1); |
109 AddMessage(kTestMessage2); | 106 AddMessage(kTestMessage2); |
110 | 107 |
111 EXPECT_CALL(callback_, OnMessage()).Times(2); | 108 EXPECT_CALL(callback_, OnMessage()).Times(2); |
112 | 109 |
113 InitReader(); | 110 InitReader(); |
114 base::RunLoop().RunUntilIdle(); | 111 base::RunLoop().RunUntilIdle(); |
115 | 112 |
116 Mock::VerifyAndClearExpectations(&callback_); | 113 Mock::VerifyAndClearExpectations(&callback_); |
117 Mock::VerifyAndClearExpectations(&socket_); | 114 Mock::VerifyAndClearExpectations(&socket_); |
118 | 115 |
119 EXPECT_TRUE(CompareResult(messages_[0], kTestMessage1)); | 116 EXPECT_TRUE(CompareResult(messages_[0].get(), kTestMessage1)); |
120 EXPECT_TRUE(CompareResult(messages_[1], kTestMessage2)); | 117 EXPECT_TRUE(CompareResult(messages_[1].get(), kTestMessage2)); |
121 | 118 |
122 EXPECT_TRUE(socket_.read_pending()); | 119 EXPECT_TRUE(socket_.read_pending()); |
123 } | 120 } |
124 | 121 |
125 // Receive two messages in separate packets. | 122 // Receive two messages in separate packets. |
126 TEST_F(MessageReaderTest, TwoMessages_Separately) { | 123 TEST_F(MessageReaderTest, TwoMessages_Separately) { |
127 AddMessage(kTestMessage1); | 124 AddMessage(kTestMessage1); |
128 | 125 |
129 EXPECT_CALL(callback_, OnMessage()) | 126 EXPECT_CALL(callback_, OnMessage()) |
130 .Times(1); | 127 .Times(1); |
131 | 128 |
132 InitReader(); | 129 InitReader(); |
133 base::RunLoop().RunUntilIdle(); | 130 base::RunLoop().RunUntilIdle(); |
134 | 131 |
135 Mock::VerifyAndClearExpectations(&callback_); | 132 Mock::VerifyAndClearExpectations(&callback_); |
136 Mock::VerifyAndClearExpectations(&socket_); | 133 Mock::VerifyAndClearExpectations(&socket_); |
137 | 134 |
138 EXPECT_TRUE(CompareResult(messages_[0], kTestMessage1)); | 135 EXPECT_TRUE(CompareResult(messages_[0].get(), kTestMessage1)); |
139 | 136 |
140 EXPECT_TRUE(socket_.read_pending()); | 137 EXPECT_TRUE(socket_.read_pending()); |
141 | 138 |
142 // Write another message and verify that we receive it. | 139 // Write another message and verify that we receive it. |
143 EXPECT_CALL(callback_, OnMessage()) | 140 EXPECT_CALL(callback_, OnMessage()) |
144 .Times(1); | 141 .Times(1); |
145 AddMessage(kTestMessage2); | 142 AddMessage(kTestMessage2); |
146 base::RunLoop().RunUntilIdle(); | 143 base::RunLoop().RunUntilIdle(); |
147 | 144 |
148 EXPECT_TRUE(CompareResult(messages_[1], kTestMessage2)); | 145 EXPECT_TRUE(CompareResult(messages_[1].get(), kTestMessage2)); |
149 | 146 |
150 EXPECT_TRUE(socket_.read_pending()); | 147 EXPECT_TRUE(socket_.read_pending()); |
151 } | 148 } |
152 | 149 |
153 // Read() returns error. | 150 // Read() returns error. |
154 TEST_F(MessageReaderTest, ReadError) { | 151 TEST_F(MessageReaderTest, ReadError) { |
155 socket_.AppendReadError(net::ERR_FAILED); | 152 socket_.AppendReadError(net::ERR_FAILED); |
156 | 153 |
157 EXPECT_CALL(callback_, OnMessage()).Times(0); | 154 EXPECT_CALL(callback_, OnMessage()).Times(0); |
158 | 155 |
(...skipping 26 matching lines...) Expand all Loading... |
185 EXPECT_CALL(callback_, OnMessage()) | 182 EXPECT_CALL(callback_, OnMessage()) |
186 .Times(1) | 183 .Times(1) |
187 .WillOnce(Invoke(this, &MessageReaderTest::DeleteReader)); | 184 .WillOnce(Invoke(this, &MessageReaderTest::DeleteReader)); |
188 | 185 |
189 InitReader(); | 186 InitReader(); |
190 base::RunLoop().RunUntilIdle(); | 187 base::RunLoop().RunUntilIdle(); |
191 } | 188 } |
192 | 189 |
193 } // namespace protocol | 190 } // namespace protocol |
194 } // namespace remoting | 191 } // namespace remoting |
OLD | NEW |