| /* |
| * Copyright (c) 2022 The WebRTC project authors. All Rights Reserved. |
| * |
| * Use of this source code is governed by a BSD-style license |
| * that can be found in the LICENSE file in the root of the source |
| * tree. An additional intellectual property rights grant can be found |
| * in the file PATENTS. All contributing project authors may |
| * be found in the AUTHORS file in the root of the source tree. |
| */ |
| |
| #include "test/pc/e2e/analyzer/video/default_video_quality_analyzer_stream_state.h" |
| |
| #include <set> |
| |
| #include "api/units/timestamp.h" |
| #include "system_wrappers/include/clock.h" |
| #include "test/gtest.h" |
| |
| namespace webrtc { |
| namespace { |
| |
| TEST(StreamStateTest, PopFrontAndFrontIndependentForEachPeer) { |
| StreamState state(/*sender=*/0, |
| /*receivers=*/std::set<size_t>{1, 2}, Timestamp::Seconds(1), |
| Clock::GetRealTimeClock()); |
| state.PushBack(/*frame_id=*/1); |
| state.PushBack(/*frame_id=*/2); |
| |
| EXPECT_EQ(state.Front(/*peer=*/1), 1); |
| EXPECT_EQ(state.PopFront(/*peer=*/1), 1); |
| EXPECT_EQ(state.Front(/*peer=*/1), 2); |
| EXPECT_EQ(state.PopFront(/*peer=*/1), 2); |
| EXPECT_EQ(state.Front(/*peer=*/2), 1); |
| EXPECT_EQ(state.PopFront(/*peer=*/2), 1); |
| EXPECT_EQ(state.Front(/*peer=*/2), 2); |
| EXPECT_EQ(state.PopFront(/*peer=*/2), 2); |
| } |
| |
| TEST(StreamStateTest, IsEmpty) { |
| StreamState state(/*sender=*/0, |
| /*receivers=*/std::set<size_t>{1, 2}, Timestamp::Seconds(1), |
| Clock::GetRealTimeClock()); |
| state.PushBack(/*frame_id=*/1); |
| |
| EXPECT_FALSE(state.IsEmpty(/*peer=*/1)); |
| |
| state.PopFront(/*peer=*/1); |
| |
| EXPECT_TRUE(state.IsEmpty(/*peer=*/1)); |
| } |
| |
| TEST(StreamStateTest, PopFrontForOnlyOnePeerDontChangeAliveFramesCount) { |
| StreamState state(/*sender=*/0, |
| /*receivers=*/std::set<size_t>{1, 2}, Timestamp::Seconds(1), |
| Clock::GetRealTimeClock()); |
| state.PushBack(/*frame_id=*/1); |
| state.PushBack(/*frame_id=*/2); |
| |
| EXPECT_EQ(state.GetAliveFramesCount(), 2lu); |
| |
| state.PopFront(/*peer=*/1); |
| state.PopFront(/*peer=*/1); |
| |
| EXPECT_EQ(state.GetAliveFramesCount(), 2lu); |
| } |
| |
| TEST(StreamStateTest, PopFrontForAllPeersReducesAliveFramesCount) { |
| StreamState state(/*sender=*/0, |
| /*receivers=*/std::set<size_t>{1, 2}, Timestamp::Seconds(1), |
| Clock::GetRealTimeClock()); |
| state.PushBack(/*frame_id=*/1); |
| state.PushBack(/*frame_id=*/2); |
| |
| EXPECT_EQ(state.GetAliveFramesCount(), 2lu); |
| |
| state.PopFront(/*peer=*/1); |
| state.PopFront(/*peer=*/2); |
| |
| EXPECT_EQ(state.GetAliveFramesCount(), 1lu); |
| } |
| |
| TEST(StreamStateTest, RemovePeerForLastExpectedReceiverUpdatesAliveFrames) { |
| StreamState state(/*sender=*/0, |
| /*receivers=*/std::set<size_t>{1, 2}, Timestamp::Seconds(1), |
| Clock::GetRealTimeClock()); |
| state.PushBack(/*frame_id=*/1); |
| state.PushBack(/*frame_id=*/2); |
| |
| state.PopFront(/*peer=*/1); |
| |
| EXPECT_EQ(state.GetAliveFramesCount(), 2lu); |
| |
| state.RemovePeer(/*peer=*/2); |
| |
| EXPECT_EQ(state.GetAliveFramesCount(), 1lu); |
| } |
| |
| } // namespace |
| } // namespace webrtc |