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 "blimp/net/blimp_message_checkpointer.h" | 5 #include "blimp/net/blimp_message_checkpointer.h" |
6 | 6 |
7 #include "base/memory/ref_counted.h" | 7 #include "base/memory/ref_counted.h" |
8 #include "base/test/test_mock_time_task_runner.h" | 8 #include "base/test/test_mock_time_task_runner.h" |
9 #include "base/thread_task_runner_handle.h" | 9 #include "base/thread_task_runner_handle.h" |
10 #include "blimp/common/create_blimp_message.h" | 10 #include "blimp/common/create_blimp_message.h" |
(...skipping 10 matching lines...) Expand all Loading... |
21 using testing::SaveArg; | 21 using testing::SaveArg; |
22 | 22 |
23 namespace blimp { | 23 namespace blimp { |
24 | 24 |
25 namespace { | 25 namespace { |
26 class MockCheckpointObserver : public BlimpMessageCheckpointObserver { | 26 class MockCheckpointObserver : public BlimpMessageCheckpointObserver { |
27 public: | 27 public: |
28 MockCheckpointObserver() {} | 28 MockCheckpointObserver() {} |
29 ~MockCheckpointObserver() override {} | 29 ~MockCheckpointObserver() override {} |
30 | 30 |
31 MOCK_METHOD1(OnMessageCheckpoint, void(int64)); | 31 MOCK_METHOD1(OnMessageCheckpoint, void(int64_t)); |
32 }; | 32 }; |
33 | 33 |
34 } // namespace | 34 } // namespace |
35 | 35 |
36 static scoped_ptr<BlimpMessage> CreateExpectedAckMessage(int64 id) { | 36 static scoped_ptr<BlimpMessage> CreateExpectedAckMessage(int64_t id) { |
37 scoped_ptr<BlimpMessage> message = make_scoped_ptr(new BlimpMessage); | 37 scoped_ptr<BlimpMessage> message = make_scoped_ptr(new BlimpMessage); |
38 message->set_type(BlimpMessage::PROTOCOL_CONTROL); | 38 message->set_type(BlimpMessage::PROTOCOL_CONTROL); |
39 ProtocolControlMessage* protocol_control = | 39 ProtocolControlMessage* protocol_control = |
40 message->mutable_protocol_control(); | 40 message->mutable_protocol_control(); |
41 protocol_control->set_type(ProtocolControlMessage::CHECKPOINT_ACK); | 41 protocol_control->set_type(ProtocolControlMessage::CHECKPOINT_ACK); |
42 CheckpointAckMessage* checkpoint_ack = | 42 CheckpointAckMessage* checkpoint_ack = |
43 protocol_control->mutable_checkpoint_ack(); | 43 protocol_control->mutable_checkpoint_ack(); |
44 checkpoint_ack->set_checkpoint_id(id); | 44 checkpoint_ack->set_checkpoint_id(id); |
45 return message; | 45 return message; |
46 } | 46 } |
47 | 47 |
48 class BlimpMessageCheckpointerTest : public testing::Test { | 48 class BlimpMessageCheckpointerTest : public testing::Test { |
49 public: | 49 public: |
50 BlimpMessageCheckpointerTest() | 50 BlimpMessageCheckpointerTest() |
51 : runner_(new base::TestMockTimeTaskRunner), runner_handle_(runner_) {} | 51 : runner_(new base::TestMockTimeTaskRunner), runner_handle_(runner_) {} |
52 | 52 |
53 ~BlimpMessageCheckpointerTest() override {} | 53 ~BlimpMessageCheckpointerTest() override {} |
54 | 54 |
55 int64 SimulateIncomingMessage() { | 55 int64_t SimulateIncomingMessage() { |
56 scoped_ptr<BlimpMessage> message(new BlimpMessage); | 56 scoped_ptr<BlimpMessage> message(new BlimpMessage); |
57 message->set_message_id(++message_id_); | 57 message->set_message_id(++message_id_); |
58 message->set_type(BlimpMessage::INPUT); | 58 message->set_type(BlimpMessage::INPUT); |
59 checkpointer_->ProcessMessage( | 59 checkpointer_->ProcessMessage( |
60 std::move(message), | 60 std::move(message), |
61 base::Bind(&BlimpMessageCheckpointerTest::IncomingCompletionCallback, | 61 base::Bind(&BlimpMessageCheckpointerTest::IncomingCompletionCallback, |
62 base::Unretained(this))); | 62 base::Unretained(this))); |
63 return message_id_; | 63 return message_id_; |
64 } | 64 } |
65 | 65 |
66 void SetUp() override { | 66 void SetUp() override { |
67 checkpointer_ = make_scoped_ptr(new BlimpMessageCheckpointer( | 67 checkpointer_ = make_scoped_ptr(new BlimpMessageCheckpointer( |
68 &incoming_processor_, &outgoing_processor_, &checkpoint_observer_)); | 68 &incoming_processor_, &outgoing_processor_, &checkpoint_observer_)); |
69 } | 69 } |
70 | 70 |
71 MOCK_METHOD1(IncomingCompletionCallback, void(int)); | 71 MOCK_METHOD1(IncomingCompletionCallback, void(int)); |
72 | 72 |
73 protected: | 73 protected: |
74 // Used to verify timing-specific behaviours. | 74 // Used to verify timing-specific behaviours. |
75 scoped_refptr<base::TestMockTimeTaskRunner> runner_; | 75 scoped_refptr<base::TestMockTimeTaskRunner> runner_; |
76 base::ThreadTaskRunnerHandle runner_handle_; | 76 base::ThreadTaskRunnerHandle runner_handle_; |
77 | 77 |
78 int64 message_id_ = 0; | 78 int64_t message_id_ = 0; |
79 | 79 |
80 testing::StrictMock<MockBlimpMessageProcessor> incoming_processor_; | 80 testing::StrictMock<MockBlimpMessageProcessor> incoming_processor_; |
81 testing::StrictMock<MockBlimpMessageProcessor> outgoing_processor_; | 81 testing::StrictMock<MockBlimpMessageProcessor> outgoing_processor_; |
82 testing::StrictMock<MockCheckpointObserver> checkpoint_observer_; | 82 testing::StrictMock<MockCheckpointObserver> checkpoint_observer_; |
83 net::CompletionCallback captured_cb_; | 83 net::CompletionCallback captured_cb_; |
84 | 84 |
85 scoped_ptr<BlimpMessageCheckpointer> checkpointer_; | 85 scoped_ptr<BlimpMessageCheckpointer> checkpointer_; |
86 }; | 86 }; |
87 | 87 |
88 TEST_F(BlimpMessageCheckpointerTest, CallbackPropagates) { | 88 TEST_F(BlimpMessageCheckpointerTest, CallbackPropagates) { |
(...skipping 86 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
175 | 175 |
176 // Simulate an incoming message. | 176 // Simulate an incoming message. |
177 scoped_ptr<BlimpMessage> ack_message = CreateCheckpointAckMessage(10); | 177 scoped_ptr<BlimpMessage> ack_message = CreateCheckpointAckMessage(10); |
178 checkpointer_->ProcessMessage( | 178 checkpointer_->ProcessMessage( |
179 std::move(ack_message), | 179 std::move(ack_message), |
180 base::Bind(&BlimpMessageCheckpointerTest::IncomingCompletionCallback, | 180 base::Bind(&BlimpMessageCheckpointerTest::IncomingCompletionCallback, |
181 base::Unretained(this))); | 181 base::Unretained(this))); |
182 } | 182 } |
183 | 183 |
184 } // namespace blimp | 184 } // namespace blimp |
OLD | NEW |