Index: webrtc/modules/video_coding/protection_bitrate_calculator_unittest.cc |
diff --git a/webrtc/modules/video_coding/protection_bitrate_calculator_unittest.cc b/webrtc/modules/video_coding/protection_bitrate_calculator_unittest.cc |
index 667dd6a359f3d888f93d44145575658fd5a792ea..4783aaf050716c7b39cd310c6c30945b8fa2540d 100644 |
--- a/webrtc/modules/video_coding/protection_bitrate_calculator_unittest.cc |
+++ b/webrtc/modules/video_coding/protection_bitrate_calculator_unittest.cc |
@@ -67,6 +67,7 @@ TEST_F(ProtectionBitrateCalculatorTest, ProtectsUsingFecBitrate) { |
protection_callback_.fec_rate_bps_ = kCodecBitrateBps; |
target_bitrate = media_opt_.SetTargetRates(kMaxBitrateBps, 30, 128, 100); |
EXPECT_EQ(kMaxBitrateBps / 2, target_bitrate); |
+ EXPECT_EQ(target_bitrate, media_opt_.GetEncoderTargetRate()); |
} |
TEST_F(ProtectionBitrateCalculatorTest, ProtectsUsingNackBitrate) { |
@@ -84,6 +85,19 @@ TEST_F(ProtectionBitrateCalculatorTest, ProtectsUsingNackBitrate) { |
protection_callback_.nack_rate_bps_ = kMaxBitrateBps; |
target_bitrate = media_opt_.SetTargetRates(kMaxBitrateBps, 30, 128, 100); |
EXPECT_EQ(kMaxBitrateBps / 2, target_bitrate); |
+ EXPECT_EQ(target_bitrate, media_opt_.GetEncoderTargetRate()); |
+} |
+ |
+TEST_F(ProtectionBitrateCalculatorTest, NoProtection) { |
+ static const uint32_t kMaxBitrateBps = 130000; |
+ |
+ media_opt_.SetProtectionMethod(false /*enable_fec*/, false /* enable_nack */); |
+ media_opt_.SetEncodingData(kCodecBitrateBps, 640, 480, 30, 1, 1000); |
+ |
+ uint32_t target_bitrate = |
+ media_opt_.SetTargetRates(kMaxBitrateBps, 30, 128, 100); |
+ EXPECT_EQ(kMaxBitrateBps, target_bitrate); |
+ EXPECT_EQ(target_bitrate, media_opt_.GetEncoderTargetRate()); |
} |
} // namespace webrtc |