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 "extensions/browser/api/cast_channel/keep_alive_delegate.h" | 5 #include "extensions/browser/api/cast_channel/keep_alive_delegate.h" |
6 | 6 |
7 #include <stdint.h> | 7 #include <stdint.h> |
8 | 8 |
9 #include "base/macros.h" | 9 #include "base/macros.h" |
10 #include "base/memory/ptr_util.h" | 10 #include "base/memory/ptr_util.h" |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
42 // Passes through the Reset call to the base MockTimer and visits the mock | 42 // Passes through the Reset call to the base MockTimer and visits the mock |
43 // ResetTriggered method. | 43 // ResetTriggered method. |
44 void Reset() override { | 44 void Reset() override { |
45 base::MockTimer::Reset(); | 45 base::MockTimer::Reset(); |
46 ResetTriggered(); | 46 ResetTriggered(); |
47 } | 47 } |
48 }; | 48 }; |
49 | 49 |
50 class KeepAliveDelegateTest : public testing::Test { | 50 class KeepAliveDelegateTest : public testing::Test { |
51 public: | 51 public: |
| 52 using ChannelError = ::cast_channel::ChannelError; |
| 53 |
52 KeepAliveDelegateTest() {} | 54 KeepAliveDelegateTest() {} |
53 ~KeepAliveDelegateTest() override {} | 55 ~KeepAliveDelegateTest() override {} |
54 | 56 |
55 protected: | 57 protected: |
56 void SetUp() override { | 58 void SetUp() override { |
57 inner_delegate_ = new MockCastTransportDelegate; | 59 inner_delegate_ = new MockCastTransportDelegate; |
58 logger_ = new Logger(); | 60 logger_ = new Logger(); |
59 keep_alive_.reset(new KeepAliveDelegate( | 61 keep_alive_.reset(new KeepAliveDelegate( |
60 &socket_, logger_, base::WrapUnique(inner_delegate_), | 62 &socket_, logger_, base::WrapUnique(inner_delegate_), |
61 base::TimeDelta::FromMilliseconds(kTestPingTimeoutMillis), | 63 base::TimeDelta::FromMilliseconds(kTestPingTimeoutMillis), |
(...skipping 18 matching lines...) Expand all Loading... |
80 scoped_refptr<Logger> logger_; | 82 scoped_refptr<Logger> logger_; |
81 MockCastTransportDelegate* inner_delegate_; | 83 MockCastTransportDelegate* inner_delegate_; |
82 MockTimerWithMonitoredReset* liveness_timer_; | 84 MockTimerWithMonitoredReset* liveness_timer_; |
83 MockTimerWithMonitoredReset* ping_timer_; | 85 MockTimerWithMonitoredReset* ping_timer_; |
84 | 86 |
85 private: | 87 private: |
86 DISALLOW_COPY_AND_ASSIGN(KeepAliveDelegateTest); | 88 DISALLOW_COPY_AND_ASSIGN(KeepAliveDelegateTest); |
87 }; | 89 }; |
88 | 90 |
89 TEST_F(KeepAliveDelegateTest, TestErrorHandledBeforeStarting) { | 91 TEST_F(KeepAliveDelegateTest, TestErrorHandledBeforeStarting) { |
90 keep_alive_->OnError(CHANNEL_ERROR_CONNECT_ERROR); | 92 keep_alive_->OnError(ChannelError::CONNECT_ERROR); |
91 } | 93 } |
92 | 94 |
93 TEST_F(KeepAliveDelegateTest, TestPing) { | 95 TEST_F(KeepAliveDelegateTest, TestPing) { |
94 EXPECT_CALL(*socket_.mock_transport(), | 96 EXPECT_CALL(*socket_.mock_transport(), |
95 SendMessage(EqualsProto(KeepAliveDelegate::CreateKeepAliveMessage( | 97 SendMessage(EqualsProto(KeepAliveDelegate::CreateKeepAliveMessage( |
96 KeepAliveDelegate::kHeartbeatPingType)), | 98 KeepAliveDelegate::kHeartbeatPingType)), |
97 _)) | 99 _)) |
98 .WillOnce(PostCompletionCallbackTask<1>(net::OK)); | 100 .WillOnce(PostCompletionCallbackTask<1>(net::OK)); |
99 EXPECT_CALL(*inner_delegate_, Start()); | 101 EXPECT_CALL(*inner_delegate_, Start()); |
100 EXPECT_CALL(*ping_timer_, ResetTriggered()).Times(2); | 102 EXPECT_CALL(*ping_timer_, ResetTriggered()).Times(2); |
101 EXPECT_CALL(*liveness_timer_, ResetTriggered()).Times(2); | 103 EXPECT_CALL(*liveness_timer_, ResetTriggered()).Times(2); |
102 EXPECT_CALL(*ping_timer_, Stop()); | 104 EXPECT_CALL(*ping_timer_, Stop()); |
103 | 105 |
104 keep_alive_->Start(); | 106 keep_alive_->Start(); |
105 ping_timer_->Fire(); | 107 ping_timer_->Fire(); |
106 keep_alive_->OnMessage(KeepAliveDelegate::CreateKeepAliveMessage( | 108 keep_alive_->OnMessage(KeepAliveDelegate::CreateKeepAliveMessage( |
107 KeepAliveDelegate::kHeartbeatPongType)); | 109 KeepAliveDelegate::kHeartbeatPongType)); |
108 RunPendingTasks(); | 110 RunPendingTasks(); |
109 } | 111 } |
110 | 112 |
111 TEST_F(KeepAliveDelegateTest, TestPingFailed) { | 113 TEST_F(KeepAliveDelegateTest, TestPingFailed) { |
112 EXPECT_CALL(*socket_.mock_transport(), | 114 EXPECT_CALL(*socket_.mock_transport(), |
113 SendMessage(EqualsProto(KeepAliveDelegate::CreateKeepAliveMessage( | 115 SendMessage(EqualsProto(KeepAliveDelegate::CreateKeepAliveMessage( |
114 KeepAliveDelegate::kHeartbeatPingType)), | 116 KeepAliveDelegate::kHeartbeatPingType)), |
115 _)) | 117 _)) |
116 .WillOnce(PostCompletionCallbackTask<1>(net::ERR_CONNECTION_RESET)); | 118 .WillOnce(PostCompletionCallbackTask<1>(net::ERR_CONNECTION_RESET)); |
117 EXPECT_CALL(*inner_delegate_, Start()); | 119 EXPECT_CALL(*inner_delegate_, Start()); |
118 EXPECT_CALL(*inner_delegate_, OnError(CHANNEL_ERROR_SOCKET_ERROR)); | 120 EXPECT_CALL(*inner_delegate_, OnError(ChannelError::CAST_SOCKET_ERROR)); |
119 EXPECT_CALL(*ping_timer_, ResetTriggered()).Times(1); | 121 EXPECT_CALL(*ping_timer_, ResetTriggered()).Times(1); |
120 EXPECT_CALL(*liveness_timer_, ResetTriggered()).Times(1); | 122 EXPECT_CALL(*liveness_timer_, ResetTriggered()).Times(1); |
121 EXPECT_CALL(*liveness_timer_, Stop()); | 123 EXPECT_CALL(*liveness_timer_, Stop()); |
122 EXPECT_CALL(*ping_timer_, Stop()).Times(2); | 124 EXPECT_CALL(*ping_timer_, Stop()).Times(2); |
123 | 125 |
124 keep_alive_->Start(); | 126 keep_alive_->Start(); |
125 ping_timer_->Fire(); | 127 ping_timer_->Fire(); |
126 RunPendingTasks(); | 128 RunPendingTasks(); |
127 EXPECT_EQ(proto::PING_WRITE_ERROR, | 129 EXPECT_EQ(proto::PING_WRITE_ERROR, |
128 logger_->GetLastErrors(socket_.id()).event_type); | 130 logger_->GetLastErrors(socket_.id()).event_type); |
129 EXPECT_EQ(net::ERR_CONNECTION_RESET, | 131 EXPECT_EQ(net::ERR_CONNECTION_RESET, |
130 logger_->GetLastErrors(socket_.id()).net_return_value); | 132 logger_->GetLastErrors(socket_.id()).net_return_value); |
131 } | 133 } |
132 | 134 |
133 TEST_F(KeepAliveDelegateTest, TestPingAndLivenessTimeout) { | 135 TEST_F(KeepAliveDelegateTest, TestPingAndLivenessTimeout) { |
134 EXPECT_CALL(*socket_.mock_transport(), | 136 EXPECT_CALL(*socket_.mock_transport(), |
135 SendMessage(EqualsProto(KeepAliveDelegate::CreateKeepAliveMessage( | 137 SendMessage(EqualsProto(KeepAliveDelegate::CreateKeepAliveMessage( |
136 KeepAliveDelegate::kHeartbeatPingType)), | 138 KeepAliveDelegate::kHeartbeatPingType)), |
137 _)) | 139 _)) |
138 .WillOnce(PostCompletionCallbackTask<1>(net::OK)); | 140 .WillOnce(PostCompletionCallbackTask<1>(net::OK)); |
139 EXPECT_CALL(*inner_delegate_, OnError(CHANNEL_ERROR_PING_TIMEOUT)); | 141 EXPECT_CALL(*inner_delegate_, OnError(ChannelError::PING_TIMEOUT)); |
140 EXPECT_CALL(*inner_delegate_, Start()); | 142 EXPECT_CALL(*inner_delegate_, Start()); |
141 EXPECT_CALL(*ping_timer_, ResetTriggered()).Times(1); | 143 EXPECT_CALL(*ping_timer_, ResetTriggered()).Times(1); |
142 EXPECT_CALL(*liveness_timer_, ResetTriggered()).Times(1); | 144 EXPECT_CALL(*liveness_timer_, ResetTriggered()).Times(1); |
143 EXPECT_CALL(*liveness_timer_, Stop()).Times(2); | 145 EXPECT_CALL(*liveness_timer_, Stop()).Times(2); |
144 EXPECT_CALL(*ping_timer_, Stop()).Times(2); | 146 EXPECT_CALL(*ping_timer_, Stop()).Times(2); |
145 | 147 |
146 keep_alive_->Start(); | 148 keep_alive_->Start(); |
147 ping_timer_->Fire(); | 149 ping_timer_->Fire(); |
148 liveness_timer_->Fire(); | 150 liveness_timer_->Fire(); |
149 RunPendingTasks(); | 151 RunPendingTasks(); |
(...skipping 25 matching lines...) Expand all Loading... |
175 EXPECT_CALL(*ping_timer_, ResetTriggered()).Times(2); | 177 EXPECT_CALL(*ping_timer_, ResetTriggered()).Times(2); |
176 EXPECT_CALL(*liveness_timer_, ResetTriggered()).Times(2); | 178 EXPECT_CALL(*liveness_timer_, ResetTriggered()).Times(2); |
177 EXPECT_CALL(*liveness_timer_, Stop()).Times(1); | 179 EXPECT_CALL(*liveness_timer_, Stop()).Times(1); |
178 EXPECT_CALL(*ping_timer_, Stop()).Times(1); | 180 EXPECT_CALL(*ping_timer_, Stop()).Times(1); |
179 | 181 |
180 Sequence message_and_error_sequence; | 182 Sequence message_and_error_sequence; |
181 EXPECT_CALL(*inner_delegate_, OnMessage(EqualsProto(message))) | 183 EXPECT_CALL(*inner_delegate_, OnMessage(EqualsProto(message))) |
182 .Times(1) | 184 .Times(1) |
183 .InSequence(message_and_error_sequence) | 185 .InSequence(message_and_error_sequence) |
184 .RetiresOnSaturation(); | 186 .RetiresOnSaturation(); |
185 EXPECT_CALL(*inner_delegate_, OnError(CHANNEL_ERROR_INVALID_MESSAGE)) | 187 EXPECT_CALL(*inner_delegate_, OnError(ChannelError::INVALID_MESSAGE)) |
186 .Times(1) | 188 .Times(1) |
187 .InSequence(message_and_error_sequence); | 189 .InSequence(message_and_error_sequence); |
188 EXPECT_CALL(*inner_delegate_, OnMessage(EqualsProto(message_after_error))) | 190 EXPECT_CALL(*inner_delegate_, OnMessage(EqualsProto(message_after_error))) |
189 .Times(1) | 191 .Times(1) |
190 .InSequence(message_and_error_sequence) | 192 .InSequence(message_and_error_sequence) |
191 .RetiresOnSaturation(); | 193 .RetiresOnSaturation(); |
192 EXPECT_CALL(*inner_delegate_, OnMessage(EqualsProto(late_ping_message))) | 194 EXPECT_CALL(*inner_delegate_, OnMessage(EqualsProto(late_ping_message))) |
193 .Times(0) | 195 .Times(0) |
194 .InSequence(message_and_error_sequence) | 196 .InSequence(message_and_error_sequence) |
195 .RetiresOnSaturation(); | 197 .RetiresOnSaturation(); |
196 | 198 |
197 // Start, process one message, then error-out. KeepAliveDelegate will | 199 // Start, process one message, then error-out. KeepAliveDelegate will |
198 // automatically stop itself. | 200 // automatically stop itself. |
199 keep_alive_->Start(); | 201 keep_alive_->Start(); |
200 keep_alive_->OnMessage(message); | 202 keep_alive_->OnMessage(message); |
201 RunPendingTasks(); | 203 RunPendingTasks(); |
202 keep_alive_->OnError(CHANNEL_ERROR_INVALID_MESSAGE); | 204 keep_alive_->OnError(ChannelError::INVALID_MESSAGE); |
203 RunPendingTasks(); | 205 RunPendingTasks(); |
204 | 206 |
205 // Process a non-PING/PONG message and expect it to pass through. | 207 // Process a non-PING/PONG message and expect it to pass through. |
206 keep_alive_->OnMessage(message_after_error); | 208 keep_alive_->OnMessage(message_after_error); |
207 RunPendingTasks(); | 209 RunPendingTasks(); |
208 | 210 |
209 // Process a late-arriving PING/PONG message, which should have no effect. | 211 // Process a late-arriving PING/PONG message, which should have no effect. |
210 keep_alive_->OnMessage(late_ping_message); | 212 keep_alive_->OnMessage(late_ping_message); |
211 RunPendingTasks(); | 213 RunPendingTasks(); |
212 } | 214 } |
213 | 215 |
214 } // namespace | 216 } // namespace |
215 } // namespace cast_channel | 217 } // namespace cast_channel |
216 } // namespace api | 218 } // namespace api |
217 } // namespace extensions | 219 } // namespace extensions |
OLD | NEW |