| Index: webrtc/call/bitrate_allocator_unittest.cc
|
| diff --git a/webrtc/call/bitrate_allocator_unittest.cc b/webrtc/call/bitrate_allocator_unittest.cc
|
| index 86f75a4380ae6ff71167395ad5b37af2414f0edd..fc4f1702bb3949e17738921ec96aae4b198dbc0c 100644
|
| --- a/webrtc/call/bitrate_allocator_unittest.cc
|
| +++ b/webrtc/call/bitrate_allocator_unittest.cc
|
| @@ -17,12 +17,12 @@
|
|
|
| namespace webrtc {
|
|
|
| -class TestBitrateObserver : public BitrateObserver {
|
| +class TestBitrateObserver : public BitrateAllocatorObserver {
|
| public:
|
| TestBitrateObserver()
|
| : last_bitrate_(0), last_fraction_loss_(0), last_rtt_(0) {}
|
|
|
| - virtual void OnNetworkChanged(uint32_t bitrate,
|
| + virtual void OnBitrateUpdated(uint32_t bitrate,
|
| uint8_t fraction_loss,
|
| int64_t rtt) {
|
| last_bitrate_ = bitrate;
|
| @@ -47,7 +47,7 @@ class BitrateAllocatorTest : public ::testing::Test {
|
| TEST_F(BitrateAllocatorTest, UpdatingBitrateObserver) {
|
| TestBitrateObserver bitrate_observer;
|
| int start_bitrate =
|
| - allocator_->AddBitrateObserver(&bitrate_observer, 100000, 1500000);
|
| + allocator_->AddObserver(&bitrate_observer, 100000, 1500000);
|
| EXPECT_EQ(300000, start_bitrate);
|
| allocator_->OnNetworkChanged(200000, 0, 0);
|
| EXPECT_EQ(200000u, bitrate_observer.last_bitrate_);
|
| @@ -56,12 +56,10 @@ TEST_F(BitrateAllocatorTest, UpdatingBitrateObserver) {
|
| // bitrate for FEC/retransmissions (see todo in BitrateAllocator).
|
| allocator_->OnNetworkChanged(4000000, 0, 0);
|
| EXPECT_EQ(3000000u, bitrate_observer.last_bitrate_);
|
| - start_bitrate =
|
| - allocator_->AddBitrateObserver(&bitrate_observer, 100000, 4000000);
|
| + start_bitrate = allocator_->AddObserver(&bitrate_observer, 100000, 4000000);
|
| EXPECT_EQ(4000000, start_bitrate);
|
|
|
| - start_bitrate =
|
| - allocator_->AddBitrateObserver(&bitrate_observer, 100000, 1500000);
|
| + start_bitrate = allocator_->AddObserver(&bitrate_observer, 100000, 1500000);
|
| EXPECT_EQ(3000000, start_bitrate);
|
| EXPECT_EQ(3000000u, bitrate_observer.last_bitrate_);
|
| allocator_->OnNetworkChanged(1500000, 0, 0);
|
| @@ -72,10 +70,9 @@ TEST_F(BitrateAllocatorTest, TwoBitrateObserversOneRtcpObserver) {
|
| TestBitrateObserver bitrate_observer_1;
|
| TestBitrateObserver bitrate_observer_2;
|
| int start_bitrate =
|
| - allocator_->AddBitrateObserver(&bitrate_observer_1, 100000, 300000);
|
| + allocator_->AddObserver(&bitrate_observer_1, 100000, 300000);
|
| EXPECT_EQ(300000, start_bitrate);
|
| - start_bitrate =
|
| - allocator_->AddBitrateObserver(&bitrate_observer_2, 200000, 300000);
|
| + start_bitrate = allocator_->AddObserver(&bitrate_observer_2, 200000, 300000);
|
| EXPECT_EQ(200000, start_bitrate);
|
|
|
| // Test too low start bitrate, hence lower than sum of min. Min bitrates will
|
| @@ -116,7 +113,7 @@ class BitrateAllocatorTestNoEnforceMin : public ::testing::Test {
|
| TEST_F(BitrateAllocatorTestNoEnforceMin, OneBitrateObserver) {
|
| TestBitrateObserver bitrate_observer_1;
|
| int start_bitrate =
|
| - allocator_->AddBitrateObserver(&bitrate_observer_1, 100000, 400000);
|
| + allocator_->AddObserver(&bitrate_observer_1, 100000, 400000);
|
| EXPECT_EQ(300000, start_bitrate);
|
|
|
| // High REMB.
|
| @@ -127,7 +124,7 @@ TEST_F(BitrateAllocatorTestNoEnforceMin, OneBitrateObserver) {
|
| allocator_->OnNetworkChanged(10000, 0, 0);
|
| EXPECT_EQ(10000u, bitrate_observer_1.last_bitrate_);
|
|
|
| - allocator_->RemoveBitrateObserver(&bitrate_observer_1);
|
| + allocator_->RemoveObserver(&bitrate_observer_1);
|
| }
|
|
|
| TEST_F(BitrateAllocatorTestNoEnforceMin, ThreeBitrateObservers) {
|
| @@ -136,16 +133,14 @@ TEST_F(BitrateAllocatorTestNoEnforceMin, ThreeBitrateObservers) {
|
| TestBitrateObserver bitrate_observer_3;
|
| // Set up the observers with min bitrates at 100000, 200000, and 300000.
|
| int start_bitrate =
|
| - allocator_->AddBitrateObserver(&bitrate_observer_1, 100000, 400000);
|
| + allocator_->AddObserver(&bitrate_observer_1, 100000, 400000);
|
| EXPECT_EQ(300000, start_bitrate);
|
|
|
| - start_bitrate =
|
| - allocator_->AddBitrateObserver(&bitrate_observer_2, 200000, 400000);
|
| + start_bitrate = allocator_->AddObserver(&bitrate_observer_2, 200000, 400000);
|
| EXPECT_EQ(200000, start_bitrate);
|
| EXPECT_EQ(100000u, bitrate_observer_1.last_bitrate_);
|
|
|
| - start_bitrate =
|
| - allocator_->AddBitrateObserver(&bitrate_observer_3, 300000, 400000);
|
| + start_bitrate = allocator_->AddObserver(&bitrate_observer_3, 300000, 400000);
|
| EXPECT_EQ(0, start_bitrate);
|
| EXPECT_EQ(100000u, bitrate_observer_1.last_bitrate_);
|
| EXPECT_EQ(200000u, bitrate_observer_2.last_bitrate_);
|
| @@ -175,9 +170,9 @@ TEST_F(BitrateAllocatorTestNoEnforceMin, ThreeBitrateObservers) {
|
| EXPECT_EQ(0u, bitrate_observer_2.last_bitrate_);
|
| EXPECT_EQ(0u, bitrate_observer_3.last_bitrate_);
|
|
|
| - allocator_->RemoveBitrateObserver(&bitrate_observer_1);
|
| - allocator_->RemoveBitrateObserver(&bitrate_observer_2);
|
| - allocator_->RemoveBitrateObserver(&bitrate_observer_3);
|
| + allocator_->RemoveObserver(&bitrate_observer_1);
|
| + allocator_->RemoveObserver(&bitrate_observer_2);
|
| + allocator_->RemoveObserver(&bitrate_observer_3);
|
| }
|
|
|
| TEST_F(BitrateAllocatorTest, ThreeBitrateObserversLowRembEnforceMin) {
|
| @@ -185,16 +180,14 @@ TEST_F(BitrateAllocatorTest, ThreeBitrateObserversLowRembEnforceMin) {
|
| TestBitrateObserver bitrate_observer_2;
|
| TestBitrateObserver bitrate_observer_3;
|
| int start_bitrate =
|
| - allocator_->AddBitrateObserver(&bitrate_observer_1, 100000, 400000);
|
| + allocator_->AddObserver(&bitrate_observer_1, 100000, 400000);
|
| EXPECT_EQ(300000, start_bitrate);
|
|
|
| - start_bitrate =
|
| - allocator_->AddBitrateObserver(&bitrate_observer_2, 200000, 400000);
|
| + start_bitrate = allocator_->AddObserver(&bitrate_observer_2, 200000, 400000);
|
| EXPECT_EQ(200000, start_bitrate);
|
| EXPECT_EQ(100000u, bitrate_observer_1.last_bitrate_);
|
|
|
| - start_bitrate =
|
| - allocator_->AddBitrateObserver(&bitrate_observer_3, 300000, 400000);
|
| + start_bitrate = allocator_->AddObserver(&bitrate_observer_3, 300000, 400000);
|
| EXPECT_EQ(300000, start_bitrate);
|
| EXPECT_EQ(100000, static_cast<int>(bitrate_observer_1.last_bitrate_));
|
| EXPECT_EQ(200000, static_cast<int>(bitrate_observer_2.last_bitrate_));
|
| @@ -205,8 +198,8 @@ TEST_F(BitrateAllocatorTest, ThreeBitrateObserversLowRembEnforceMin) {
|
| EXPECT_EQ(200000u, bitrate_observer_2.last_bitrate_); // Min cap.
|
| EXPECT_EQ(300000u, bitrate_observer_3.last_bitrate_); // Min cap.
|
|
|
| - allocator_->RemoveBitrateObserver(&bitrate_observer_1);
|
| - allocator_->RemoveBitrateObserver(&bitrate_observer_2);
|
| - allocator_->RemoveBitrateObserver(&bitrate_observer_3);
|
| + allocator_->RemoveObserver(&bitrate_observer_1);
|
| + allocator_->RemoveObserver(&bitrate_observer_2);
|
| + allocator_->RemoveObserver(&bitrate_observer_3);
|
| }
|
| } // namespace webrtc
|
|
|