Index: webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc |
diff --git a/webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc b/webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc |
index 3f467ef8a46e16ecb668c6ac35131fbd26d7e2bf..610d1fb7621df0fa97f0e2cc64191f8168127c27 100644 |
--- a/webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc |
+++ b/webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc |
@@ -13,6 +13,7 @@ |
#include "testing/gtest/include/gtest/gtest.h" |
+#include "webrtc/call/mock/mock_rtc_event_log.h" |
#include "webrtc/modules/bitrate_controller/include/bitrate_controller.h" |
#include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" |
@@ -61,8 +62,8 @@ class BitrateControllerTest : public ::testing::Test { |
~BitrateControllerTest() {} |
virtual void SetUp() { |
- controller_ = |
- BitrateController::CreateBitrateController(&clock_, &bitrate_observer_); |
+ controller_ = BitrateController::CreateBitrateController( |
+ &clock_, &bitrate_observer_, &event_log_); |
controller_->SetStartBitrate(kStartBitrateBps); |
EXPECT_EQ(kStartBitrateBps, bitrate_observer_.last_bitrate_); |
controller_->SetMinMaxBitrate(kMinBitrateBps, kMaxBitrateBps); |
@@ -86,6 +87,7 @@ class BitrateControllerTest : public ::testing::Test { |
TestBitrateObserver bitrate_observer_; |
BitrateController* controller_; |
RtcpBandwidthObserver* bandwidth_observer_; |
+ webrtc::MockRtcEventLog event_log_; |
}; |
TEST_F(BitrateControllerTest, DefaultMinMaxBitrate) { |
@@ -102,6 +104,7 @@ TEST_F(BitrateControllerTest, DefaultMinMaxBitrate) { |
TEST_F(BitrateControllerTest, OneBitrateObserverOneRtcpObserver) { |
// First REMB applies immediately. |
+ EXPECT_CALL(event_log_, LogBwePacketLossEvent(testing::Gt(0), 0, 0)).Times(8); |
int64_t time_ms = 1001; |
webrtc::ReportBlockList report_blocks; |
report_blocks.push_back(CreateReportBlock(1, 2, 0, 1)); |
@@ -178,6 +181,7 @@ TEST_F(BitrateControllerTest, OneBitrateObserverOneRtcpObserver) { |
TEST_F(BitrateControllerTest, OneBitrateObserverTwoRtcpObservers) { |
// REMBs during the first 2 seconds apply immediately. |
+ EXPECT_CALL(event_log_, LogBwePacketLossEvent(testing::Gt(0), 0, 0)).Times(9); |
int64_t time_ms = 1; |
webrtc::ReportBlockList report_blocks; |
report_blocks.push_back(CreateReportBlock(1, 2, 0, 1)); |
@@ -273,6 +277,13 @@ TEST_F(BitrateControllerTest, OneBitrateObserverTwoRtcpObservers) { |
} |
TEST_F(BitrateControllerTest, OneBitrateObserverMultipleReportBlocks) { |
+ testing::Expectation first_calls = |
+ EXPECT_CALL(event_log_, LogBwePacketLossEvent(testing::Gt(0), 0, 0)) |
+ .Times(7); |
+ EXPECT_CALL(event_log_, |
+ LogBwePacketLossEvent(testing::Gt(0), testing::Gt(0), 0)) |
+ .Times(2) |
+ .After(first_calls); |
uint32_t sequence_number[2] = {0, 0xFF00}; |
const int kStartBitrate = 200000; |
const int kMinBitrate = 100000; |