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 |