blob: fa8b97b9e4dc2a2b11db235417344ea1e235276e [file] [log] [blame]
/*
* Copyright (c) 2018 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 "modules/rtp_rtcp/source/contributing_sources.h"
#include "rtc_base/time_utils.h"
#include "test/gmock.h"
#include "test/gtest.h"
namespace webrtc {
namespace {
using ::testing::UnorderedElementsAre;
constexpr uint32_t kCsrc1 = 111;
constexpr uint32_t kCsrc2 = 222;
constexpr uint32_t kCsrc3 = 333;
constexpr uint32_t kRtpTimestamp1 = 314;
constexpr uint32_t kRtpTimestamp2 = 315;
constexpr uint32_t kRtpTimestamp3 = 316;
} // namespace
TEST(ContributingSourcesTest, RecordSources) {
ContributingSources csrcs;
constexpr uint32_t kCsrcs[] = {kCsrc1, kCsrc2};
constexpr int64_t kTime1 = 10;
csrcs.Update(kTime1, kCsrcs, absl::nullopt, kRtpTimestamp1);
EXPECT_THAT(
csrcs.GetSources(kTime1),
UnorderedElementsAre(RtpSource(kTime1, kCsrc1, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp1),
RtpSource(kTime1, kCsrc2, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp1)));
}
TEST(ContributingSourcesTest, UpdateSources) {
ContributingSources csrcs;
// TODO(nisse): When migrating to absl::Span, the named constant arrays should
// be replaced by unnamed literals where they are passed to csrcs.Update(...).
constexpr uint32_t kCsrcs1[] = {kCsrc1, kCsrc2};
constexpr uint32_t kCsrcs2[] = {kCsrc2, kCsrc3};
constexpr int64_t kTime1 = 10;
constexpr int64_t kTime2 = kTime1 + 5 * rtc::kNumMillisecsPerSec;
csrcs.Update(kTime1, kCsrcs1, absl::nullopt, kRtpTimestamp1);
EXPECT_THAT(
csrcs.GetSources(kTime1),
UnorderedElementsAre(RtpSource(kTime1, kCsrc1, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp1),
RtpSource(kTime1, kCsrc2, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp1)));
csrcs.Update(kTime2, kCsrcs2, absl::nullopt, kRtpTimestamp2);
EXPECT_THAT(
csrcs.GetSources(kTime2),
UnorderedElementsAre(RtpSource(kTime1, kCsrc1, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp1),
RtpSource(kTime2, kCsrc2, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp2),
RtpSource(kTime2, kCsrc3, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp2)));
}
TEST(ContributingSourcesTest, ReturnRecentOnly) {
ContributingSources csrcs;
constexpr uint32_t kCsrcs1[] = {kCsrc1, kCsrc2};
constexpr uint32_t kCsrcs2[] = {kCsrc2, kCsrc3};
constexpr int64_t kTime1 = 10;
constexpr int64_t kTime2 = kTime1 + 5 * rtc::kNumMillisecsPerSec;
constexpr int64_t kTime3 = kTime1 + 12 * rtc::kNumMillisecsPerSec;
csrcs.Update(kTime1, kCsrcs1, absl::nullopt, kRtpTimestamp1);
EXPECT_THAT(
csrcs.GetSources(kTime1),
UnorderedElementsAre(RtpSource(kTime1, kCsrc1, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp1),
RtpSource(kTime1, kCsrc2, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp1)));
csrcs.Update(kTime2, kCsrcs2, absl::nullopt, kRtpTimestamp2);
EXPECT_THAT(
csrcs.GetSources(kTime3),
UnorderedElementsAre(RtpSource(kTime2, kCsrc2, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp2),
RtpSource(kTime2, kCsrc3, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp2)));
}
TEST(ContributingSourcesTest, PurgeOldSources) {
ContributingSources csrcs;
constexpr uint32_t kCsrcs1[] = {kCsrc1, kCsrc2};
constexpr uint32_t kCsrcs2[] = {kCsrc2, kCsrc3};
constexpr int64_t kTime1 = 10;
constexpr int64_t kTime2 = kTime1 + 10 * rtc::kNumMillisecsPerSec;
constexpr int64_t kTime3 = kTime1 + 20 * rtc::kNumMillisecsPerSec;
csrcs.Update(kTime1, kCsrcs1, absl::nullopt, kRtpTimestamp1);
EXPECT_THAT(
csrcs.GetSources(kTime2),
UnorderedElementsAre(RtpSource(kTime1, kCsrc1, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp1),
RtpSource(kTime1, kCsrc2, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp1)));
csrcs.Update(kTime2, kCsrcs2, absl::nullopt, kRtpTimestamp2);
EXPECT_THAT(
csrcs.GetSources(kTime2),
UnorderedElementsAre(RtpSource(kTime1, kCsrc1, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp1),
RtpSource(kTime2, kCsrc2, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp2),
RtpSource(kTime2, kCsrc3, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp2)));
csrcs.Update(kTime3, kCsrcs2, absl::nullopt, kRtpTimestamp3);
EXPECT_THAT(
csrcs.GetSources(kTime3),
UnorderedElementsAre(RtpSource(kTime3, kCsrc2, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp3),
RtpSource(kTime3, kCsrc3, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp3)));
// Query at an earlier time; check that old sources really have been purged
// and don't reappear.
EXPECT_THAT(
csrcs.GetSources(kTime2),
UnorderedElementsAre(RtpSource(kTime3, kCsrc2, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp3),
RtpSource(kTime3, kCsrc3, RtpSourceType::CSRC,
absl::nullopt, kRtpTimestamp3)));
}
TEST(ContributingSourcesTest, AudioLevel) {
ContributingSources csrcs;
constexpr uint32_t kCsrcs[] = {kCsrc1, kCsrc2};
constexpr int64_t kTime1 = 10;
csrcs.Update(kTime1, kCsrcs, 47, kRtpTimestamp1);
EXPECT_THAT(
csrcs.GetSources(kTime1),
UnorderedElementsAre(
RtpSource(kTime1, kCsrc1, RtpSourceType::CSRC, 47, kRtpTimestamp1),
RtpSource(kTime1, kCsrc2, RtpSourceType::CSRC, 47, kRtpTimestamp1)));
constexpr uint32_t kCsrcsSubset[] = {kCsrc1};
csrcs.Update(kTime1 + 1, kCsrcsSubset, absl::nullopt, kRtpTimestamp2);
EXPECT_THAT(
csrcs.GetSources(kTime1 + 1),
UnorderedElementsAre(
RtpSource(kTime1 + 1, kCsrc1, RtpSourceType::CSRC, absl::nullopt,
kRtpTimestamp2),
RtpSource(kTime1, kCsrc2, RtpSourceType::CSRC, 47, kRtpTimestamp1)));
}
} // namespace webrtc