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