| // Copyright 2019 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "remoting/signaling/message_tracker.h" |
| |
| #include "base/test/task_environment.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace remoting { |
| |
| class MessageTrackerTest : public testing::Test { |
| public: |
| protected: |
| static constexpr base::TimeDelta GetCleanupInterval(); |
| |
| base::test::TaskEnvironment task_environment_{ |
| base::test::TaskEnvironment::TimeSource::MOCK_TIME}; |
| MessageTracker message_tracker_; |
| }; |
| |
| // static |
| constexpr base::TimeDelta MessageTrackerTest::GetCleanupInterval() { |
| return MessageTracker::kCleanupInterval; |
| } |
| |
| TEST_F(MessageTrackerTest, TestIsIdTracked_UntrackedIdReturnsTrue) { |
| ASSERT_FALSE(message_tracker_.IsIdTracked("1")); |
| } |
| |
| TEST_F(MessageTrackerTest, TrackIdOnce_IdIsTracked) { |
| message_tracker_.TrackId("1"); |
| ASSERT_TRUE(message_tracker_.IsIdTracked("1")); |
| } |
| |
| TEST_F(MessageTrackerTest, TrackIdAndAdvanceTimer_ExpiredIdNotRejected) { |
| message_tracker_.TrackId("1"); |
| ASSERT_TRUE(message_tracker_.IsIdTracked("1")); |
| task_environment_.FastForwardBy(GetCleanupInterval() * 2); |
| ASSERT_FALSE(message_tracker_.IsIdTracked("1")); |
| } |
| |
| TEST_F(MessageTrackerTest, TrackIdAndAdvanceTimer_NotExpiredIdRejected) { |
| message_tracker_.TrackId("1"); |
| ASSERT_TRUE(message_tracker_.IsIdTracked("1")); |
| task_environment_.FastForwardBy(GetCleanupInterval() / 2); |
| ASSERT_TRUE(message_tracker_.IsIdTracked("1")); |
| } |
| |
| } // namespace remoting |