| Index: webrtc/modules/rtp_rtcp/source/rtp_receiver_unittest.cc
|
| diff --git a/webrtc/modules/rtp_rtcp/source/rtp_receiver_unittest.cc b/webrtc/modules/rtp_rtcp/source/rtp_receiver_unittest.cc
|
| deleted file mode 100644
|
| index 459e8e0315623841e30cfa29cedfe6f66202d048..0000000000000000000000000000000000000000
|
| --- a/webrtc/modules/rtp_rtcp/source/rtp_receiver_unittest.cc
|
| +++ /dev/null
|
| @@ -1,260 +0,0 @@
|
| -/*
|
| - * Copyright (c) 2017 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 <memory>
|
| -
|
| -#include "webrtc/common_types.h"
|
| -#include "webrtc/modules/rtp_rtcp/include/rtp_header_parser.h"
|
| -#include "webrtc/modules/rtp_rtcp/include/rtp_payload_registry.h"
|
| -#include "webrtc/modules/rtp_rtcp/include/rtp_receiver.h"
|
| -#include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h"
|
| -#include "webrtc/modules/rtp_rtcp/source/rtp_receiver_impl.h"
|
| -#include "webrtc/test/gtest.h"
|
| -
|
| -namespace webrtc {
|
| -
|
| -const uint32_t kTestRate = 64000u;
|
| -const uint8_t kTestPayload[] = {'t', 'e', 's', 't'};
|
| -const uint8_t kPcmuPayloadType = 96;
|
| -const int64_t kGetSourcesTimeoutMs = 10000;
|
| -const int kSourceListsSize = 20;
|
| -
|
| -class RtpReceiverTest : public ::testing::Test {
|
| - protected:
|
| - RtpReceiverTest()
|
| - : fake_clock_(123456),
|
| - rtp_receiver_(
|
| - RtpReceiver::CreateAudioReceiver(&fake_clock_,
|
| - nullptr,
|
| - nullptr,
|
| - &rtp_payload_registry_)) {
|
| - CodecInst voice_codec = {};
|
| - voice_codec.pltype = kPcmuPayloadType;
|
| - voice_codec.plfreq = 8000;
|
| - voice_codec.rate = kTestRate;
|
| - memcpy(voice_codec.plname, "PCMU", 5);
|
| - rtp_receiver_->RegisterReceivePayload(voice_codec);
|
| - }
|
| - ~RtpReceiverTest() {}
|
| -
|
| - bool FindSourceByIdAndType(const std::vector<RtpSource>& sources,
|
| - uint32_t source_id,
|
| - RtpSourceType type,
|
| - RtpSource* source) {
|
| - for (size_t i = 0; i < sources.size(); ++i) {
|
| - if (sources[i].source_id() == source_id &&
|
| - sources[i].source_type() == type) {
|
| - (*source) = sources[i];
|
| - return true;
|
| - }
|
| - }
|
| - return false;
|
| - }
|
| -
|
| - SimulatedClock fake_clock_;
|
| - RTPPayloadRegistry rtp_payload_registry_;
|
| - std::unique_ptr<RtpReceiver> rtp_receiver_;
|
| -};
|
| -
|
| -TEST_F(RtpReceiverTest, GetSources) {
|
| - RTPHeader header;
|
| - header.payloadType = kPcmuPayloadType;
|
| - header.ssrc = 1;
|
| - header.timestamp = fake_clock_.TimeInMilliseconds();
|
| - header.numCSRCs = 2;
|
| - header.arrOfCSRCs[0] = 111;
|
| - header.arrOfCSRCs[1] = 222;
|
| - PayloadUnion payload_specific = {AudioPayload()};
|
| - bool in_order = false;
|
| - RtpSource source(0, 0, RtpSourceType::SSRC);
|
| -
|
| - EXPECT_TRUE(rtp_receiver_->IncomingRtpPacket(header, kTestPayload, 4,
|
| - payload_specific, in_order));
|
| - auto sources = rtp_receiver_->GetSources();
|
| - // One SSRC source and two CSRC sources.
|
| - ASSERT_EQ(3u, sources.size());
|
| - ASSERT_TRUE(FindSourceByIdAndType(sources, 1u, RtpSourceType::SSRC, &source));
|
| - EXPECT_EQ(fake_clock_.TimeInMilliseconds(), source.timestamp_ms());
|
| - ASSERT_TRUE(
|
| - FindSourceByIdAndType(sources, 222u, RtpSourceType::CSRC, &source));
|
| - EXPECT_EQ(fake_clock_.TimeInMilliseconds(), source.timestamp_ms());
|
| - ASSERT_TRUE(
|
| - FindSourceByIdAndType(sources, 111u, RtpSourceType::CSRC, &source));
|
| - EXPECT_EQ(fake_clock_.TimeInMilliseconds(), source.timestamp_ms());
|
| -
|
| - // Advance the fake clock and the method is expected to return the
|
| - // contributing source object with same source id and updated timestamp.
|
| - fake_clock_.AdvanceTimeMilliseconds(1);
|
| - EXPECT_TRUE(rtp_receiver_->IncomingRtpPacket(header, kTestPayload, 4,
|
| - payload_specific, in_order));
|
| - sources = rtp_receiver_->GetSources();
|
| - ASSERT_EQ(3u, sources.size());
|
| - ASSERT_TRUE(FindSourceByIdAndType(sources, 1u, RtpSourceType::SSRC, &source));
|
| - EXPECT_EQ(fake_clock_.TimeInMilliseconds(), source.timestamp_ms());
|
| - ASSERT_TRUE(
|
| - FindSourceByIdAndType(sources, 222u, RtpSourceType::CSRC, &source));
|
| - EXPECT_EQ(fake_clock_.TimeInMilliseconds(), source.timestamp_ms());
|
| - ASSERT_TRUE(
|
| - FindSourceByIdAndType(sources, 111u, RtpSourceType::CSRC, &source));
|
| - EXPECT_EQ(fake_clock_.TimeInMilliseconds(), source.timestamp_ms());
|
| -
|
| - // Test the edge case that the sources are still there just before the
|
| - // timeout.
|
| - int64_t prev_timestamp = fake_clock_.TimeInMilliseconds();
|
| - fake_clock_.AdvanceTimeMilliseconds(kGetSourcesTimeoutMs);
|
| - sources = rtp_receiver_->GetSources();
|
| - ASSERT_EQ(3u, sources.size());
|
| - ASSERT_TRUE(FindSourceByIdAndType(sources, 1u, RtpSourceType::SSRC, &source));
|
| - EXPECT_EQ(prev_timestamp, source.timestamp_ms());
|
| - ASSERT_TRUE(
|
| - FindSourceByIdAndType(sources, 222u, RtpSourceType::CSRC, &source));
|
| - EXPECT_EQ(prev_timestamp, source.timestamp_ms());
|
| - ASSERT_TRUE(
|
| - FindSourceByIdAndType(sources, 111u, RtpSourceType::CSRC, &source));
|
| - EXPECT_EQ(prev_timestamp, source.timestamp_ms());
|
| -
|
| - // Time out.
|
| - fake_clock_.AdvanceTimeMilliseconds(1);
|
| - sources = rtp_receiver_->GetSources();
|
| - // All the sources should be out of date.
|
| - ASSERT_EQ(0u, sources.size());
|
| -}
|
| -
|
| -// Test the case that the SSRC is changed.
|
| -TEST_F(RtpReceiverTest, GetSourcesChangeSSRC) {
|
| - int64_t prev_time = -1;
|
| - int64_t cur_time = fake_clock_.TimeInMilliseconds();
|
| - RTPHeader header;
|
| - header.payloadType = kPcmuPayloadType;
|
| - header.ssrc = 1;
|
| - header.timestamp = cur_time;
|
| - PayloadUnion payload_specific = {AudioPayload()};
|
| - bool in_order = false;
|
| - RtpSource source(0, 0, RtpSourceType::SSRC);
|
| -
|
| - EXPECT_TRUE(rtp_receiver_->IncomingRtpPacket(header, kTestPayload, 4,
|
| - payload_specific, in_order));
|
| - auto sources = rtp_receiver_->GetSources();
|
| - ASSERT_EQ(1u, sources.size());
|
| - EXPECT_EQ(1u, sources[0].source_id());
|
| - EXPECT_EQ(cur_time, sources[0].timestamp_ms());
|
| -
|
| - // The SSRC is changed and the old SSRC is expected to be returned.
|
| - fake_clock_.AdvanceTimeMilliseconds(100);
|
| - prev_time = cur_time;
|
| - cur_time = fake_clock_.TimeInMilliseconds();
|
| - header.ssrc = 2;
|
| - header.timestamp = cur_time;
|
| - EXPECT_TRUE(rtp_receiver_->IncomingRtpPacket(header, kTestPayload, 4,
|
| - payload_specific, in_order));
|
| - sources = rtp_receiver_->GetSources();
|
| - ASSERT_EQ(2u, sources.size());
|
| - ASSERT_TRUE(FindSourceByIdAndType(sources, 2u, RtpSourceType::SSRC, &source));
|
| - EXPECT_EQ(cur_time, source.timestamp_ms());
|
| - ASSERT_TRUE(FindSourceByIdAndType(sources, 1u, RtpSourceType::SSRC, &source));
|
| - EXPECT_EQ(prev_time, source.timestamp_ms());
|
| -
|
| - // The SSRC is changed again and happen to be changed back to 1. No
|
| - // duplication is expected.
|
| - fake_clock_.AdvanceTimeMilliseconds(100);
|
| - header.ssrc = 1;
|
| - header.timestamp = cur_time;
|
| - prev_time = cur_time;
|
| - cur_time = fake_clock_.TimeInMilliseconds();
|
| - EXPECT_TRUE(rtp_receiver_->IncomingRtpPacket(header, kTestPayload, 4,
|
| - payload_specific, in_order));
|
| - sources = rtp_receiver_->GetSources();
|
| - ASSERT_EQ(2u, sources.size());
|
| - ASSERT_TRUE(FindSourceByIdAndType(sources, 1u, RtpSourceType::SSRC, &source));
|
| - EXPECT_EQ(cur_time, source.timestamp_ms());
|
| - ASSERT_TRUE(FindSourceByIdAndType(sources, 2u, RtpSourceType::SSRC, &source));
|
| - EXPECT_EQ(prev_time, source.timestamp_ms());
|
| -
|
| - // Old SSRC source timeout.
|
| - fake_clock_.AdvanceTimeMilliseconds(kGetSourcesTimeoutMs);
|
| - cur_time = fake_clock_.TimeInMilliseconds();
|
| - EXPECT_TRUE(rtp_receiver_->IncomingRtpPacket(header, kTestPayload, 4,
|
| - payload_specific, in_order));
|
| - sources = rtp_receiver_->GetSources();
|
| - ASSERT_EQ(1u, sources.size());
|
| - EXPECT_EQ(1u, sources[0].source_id());
|
| - EXPECT_EQ(cur_time, sources[0].timestamp_ms());
|
| - EXPECT_EQ(RtpSourceType::SSRC, sources[0].source_type());
|
| -}
|
| -
|
| -TEST_F(RtpReceiverTest, GetSourcesRemoveOutdatedSource) {
|
| - int64_t timestamp = fake_clock_.TimeInMilliseconds();
|
| - bool in_order = false;
|
| - RTPHeader header;
|
| - header.payloadType = kPcmuPayloadType;
|
| - header.timestamp = timestamp;
|
| - PayloadUnion payload_specific = {AudioPayload()};
|
| - header.numCSRCs = 1;
|
| - RtpSource source(0, 0, RtpSourceType::SSRC);
|
| -
|
| - for (size_t i = 0; i < kSourceListsSize; ++i) {
|
| - header.ssrc = i;
|
| - header.arrOfCSRCs[0] = (i + 1);
|
| - EXPECT_TRUE(rtp_receiver_->IncomingRtpPacket(header, kTestPayload, 4,
|
| - payload_specific, in_order));
|
| - }
|
| -
|
| - auto sources = rtp_receiver_->GetSources();
|
| - // Expect |kSourceListsSize| SSRC sources and |kSourceListsSize| CSRC sources.
|
| - ASSERT_TRUE(sources.size() == 2 * kSourceListsSize);
|
| - for (size_t i = 0; i < kSourceListsSize; ++i) {
|
| - // The SSRC source IDs are expected to be 19, 18, 17 ... 0
|
| - ASSERT_TRUE(
|
| - FindSourceByIdAndType(sources, i, RtpSourceType::SSRC, &source));
|
| - EXPECT_EQ(timestamp, source.timestamp_ms());
|
| -
|
| - // The CSRC source IDs are expected to be 20, 19, 18 ... 1
|
| - ASSERT_TRUE(
|
| - FindSourceByIdAndType(sources, (i + 1), RtpSourceType::CSRC, &source));
|
| - EXPECT_EQ(timestamp, source.timestamp_ms());
|
| - }
|
| -
|
| - fake_clock_.AdvanceTimeMilliseconds(kGetSourcesTimeoutMs);
|
| - for (size_t i = 0; i < kSourceListsSize; ++i) {
|
| - // The SSRC source IDs are expected to be 19, 18, 17 ... 0
|
| - ASSERT_TRUE(
|
| - FindSourceByIdAndType(sources, i, RtpSourceType::SSRC, &source));
|
| - EXPECT_EQ(timestamp, source.timestamp_ms());
|
| -
|
| - // The CSRC source IDs are expected to be 20, 19, 18 ... 1
|
| - ASSERT_TRUE(
|
| - FindSourceByIdAndType(sources, (i + 1), RtpSourceType::CSRC, &source));
|
| - EXPECT_EQ(timestamp, source.timestamp_ms());
|
| - }
|
| -
|
| - // Timeout. All the existing objects are out of date and are expected to be
|
| - // removed.
|
| - fake_clock_.AdvanceTimeMilliseconds(1);
|
| - header.ssrc = 111;
|
| - header.arrOfCSRCs[0] = 222;
|
| - EXPECT_TRUE(rtp_receiver_->IncomingRtpPacket(header, kTestPayload, 4,
|
| - payload_specific, in_order));
|
| - auto rtp_receiver_impl = static_cast<RtpReceiverImpl*>(rtp_receiver_.get());
|
| - auto ssrc_sources = rtp_receiver_impl->ssrc_sources_for_testing();
|
| - ASSERT_EQ(1u, ssrc_sources.size());
|
| - EXPECT_EQ(111u, ssrc_sources.begin()->source_id());
|
| - EXPECT_EQ(RtpSourceType::SSRC, ssrc_sources.begin()->source_type());
|
| - EXPECT_EQ(fake_clock_.TimeInMilliseconds(),
|
| - ssrc_sources.begin()->timestamp_ms());
|
| -
|
| - auto csrc_sources = rtp_receiver_impl->csrc_sources_for_testing();
|
| - ASSERT_EQ(1u, csrc_sources.size());
|
| - EXPECT_EQ(222u, csrc_sources.begin()->source_id());
|
| - EXPECT_EQ(RtpSourceType::CSRC, csrc_sources.begin()->source_type());
|
| - EXPECT_EQ(fake_clock_.TimeInMilliseconds(),
|
| - csrc_sources.begin()->timestamp_ms());
|
| -}
|
| -
|
| -} // namespace webrtc
|
|
|