diff options
Diffstat (limited to 'osp/impl/quic/quic_server_unittest.cc')
-rw-r--r-- | osp/impl/quic/quic_server_unittest.cc | 36 |
1 files changed, 17 insertions, 19 deletions
diff --git a/osp/impl/quic/quic_server_unittest.cc b/osp/impl/quic/quic_server_unittest.cc index ed1c3362..12ad8c33 100644 --- a/osp/impl/quic/quic_server_unittest.cc +++ b/osp/impl/quic/quic_server_unittest.cc @@ -49,12 +49,11 @@ class MockConnectionObserver final : public ProtocolConnection::Observer { class QuicServerTest : public Test { public: QuicServerTest() { - fake_clock_ = std::make_unique<platform::FakeClock>( - platform::Clock::time_point(std::chrono::milliseconds(1298424))); - task_runner_ = - std::make_unique<platform::FakeTaskRunner>(fake_clock_.get()); - quic_bridge_ = std::make_unique<FakeQuicBridge>(task_runner_.get(), - platform::FakeClock::now); + fake_clock_ = std::make_unique<FakeClock>( + Clock::time_point(std::chrono::milliseconds(1298424))); + task_runner_ = std::make_unique<FakeTaskRunner>(fake_clock_.get()); + quic_bridge_ = + std::make_unique<FakeQuicBridge>(task_runner_.get(), FakeClock::now); } protected: @@ -103,17 +102,16 @@ class QuicServerTest : public Test { EXPECT_CALL(mock_message_callback, OnStreamMessage( 0, _, msgs::Type::kPresentationConnectionMessage, _, _, _)) - .WillOnce( - Invoke([&decode_result, &received_message]( - uint64_t endpoint_id, uint64_t connection_id, - msgs::Type message_type, const uint8_t* buffer, - size_t buffer_size, platform::Clock::time_point now) { - decode_result = msgs::DecodePresentationConnectionMessage( - buffer, buffer_size, &received_message); - if (decode_result < 0) - return ErrorOr<size_t>(Error::Code::kCborParsing); - return ErrorOr<size_t>(decode_result); - })); + .WillOnce(Invoke([&decode_result, &received_message]( + uint64_t endpoint_id, uint64_t connection_id, + msgs::Type message_type, const uint8_t* buffer, + size_t buffer_size, Clock::time_point now) { + decode_result = msgs::DecodePresentationConnectionMessage( + buffer, buffer_size, &received_message); + if (decode_result < 0) + return ErrorOr<size_t>(Error::Code::kCborParsing); + return ErrorOr<size_t>(decode_result); + })); quic_bridge_->RunTasksUntilIdle(); ASSERT_GT(decode_result, 0); @@ -124,8 +122,8 @@ class QuicServerTest : public Test { EXPECT_EQ(received_message.message.str, message.message.str); } - std::unique_ptr<platform::FakeClock> fake_clock_; - std::unique_ptr<platform::FakeTaskRunner> task_runner_; + std::unique_ptr<FakeClock> fake_clock_; + std::unique_ptr<FakeTaskRunner> task_runner_; std::unique_ptr<FakeQuicBridge> quic_bridge_; QuicServer* server_; }; |