Index: webrtc/video/payload_router_unittest.cc |
diff --git a/webrtc/video/payload_router_unittest.cc b/webrtc/video/payload_router_unittest.cc |
index 41e173bf5fb930669be163d0749ea2d39717aeb6..f29a8c2da98f392c392d93826978087dbab23cc0 100644 |
--- a/webrtc/video/payload_router_unittest.cc |
+++ b/webrtc/video/payload_router_unittest.cc |
@@ -27,6 +27,8 @@ namespace webrtc { |
TEST(PayloadRouterTest, SendOnOneModule) { |
MockRtpRtcp rtp; |
std::vector<RtpRtcp*> modules(1, &rtp); |
+ VideoCodec codec; |
+ codec.numberOfSimulcastStreams = 1; |
uint8_t payload = 'a'; |
int8_t payload_type = 96; |
@@ -38,7 +40,7 @@ TEST(PayloadRouterTest, SendOnOneModule) { |
encoded_image._length = 1; |
PayloadRouter payload_router(modules, payload_type); |
- payload_router.SetSendingRtpModules(modules.size()); |
+ payload_router.SetSendCodec(codec); |
EXPECT_CALL(rtp, SendOutgoingData(encoded_image._frameType, payload_type, |
encoded_image._timeStamp, |
@@ -71,7 +73,8 @@ TEST(PayloadRouterTest, SendOnOneModule) { |
.Times(1); |
EXPECT_EQ(0, payload_router.Encoded(encoded_image, nullptr, nullptr)); |
- payload_router.SetSendingRtpModules(0); |
+ codec.numberOfSimulcastStreams = 0; |
+ payload_router.SetSendCodec(codec); |
EXPECT_CALL(rtp, SendOutgoingData(encoded_image._frameType, payload_type, |
encoded_image._timeStamp, |
encoded_image.capture_time_ms_, &payload, |
@@ -86,6 +89,8 @@ TEST(PayloadRouterTest, SendSimulcast) { |
std::vector<RtpRtcp*> modules; |
modules.push_back(&rtp_1); |
modules.push_back(&rtp_2); |
+ VideoCodec codec; |
+ codec.numberOfSimulcastStreams = 2; |
int8_t payload_type = 96; |
uint8_t payload = 'a'; |
@@ -97,7 +102,7 @@ TEST(PayloadRouterTest, SendSimulcast) { |
encoded_image._length = 1; |
PayloadRouter payload_router(modules, payload_type); |
- payload_router.SetSendingRtpModules(modules.size()); |
+ payload_router.SetSendCodec(codec); |
CodecSpecificInfo codec_info_1; |
memset(&codec_info_1, 0, sizeof(CodecSpecificInfo)); |
@@ -138,7 +143,8 @@ TEST(PayloadRouterTest, SendSimulcast) { |
EXPECT_EQ(-1, payload_router.Encoded(encoded_image, &codec_info_2, nullptr)); |
// Invalid simulcast index. |
- payload_router.SetSendingRtpModules(1); |
+ codec.numberOfSimulcastStreams = 1; |
+ payload_router.SetSendCodec(codec); |
payload_router.set_active(true); |
EXPECT_CALL(rtp_1, SendOutgoingData(_, _, _, _, _, _, _, _)) |
.Times(0); |
@@ -160,7 +166,9 @@ TEST(PayloadRouterTest, MaxPayloadLength) { |
PayloadRouter payload_router(modules, 42); |
EXPECT_EQ(kDefaultMaxLength, PayloadRouter::DefaultMaxPayloadLength()); |
- payload_router.SetSendingRtpModules(modules.size()); |
+ VideoCodec codec; |
+ codec.numberOfSimulcastStreams = 2; |
+ payload_router.SetSendCodec(codec); |
// Modules return a higher length than the default value. |
EXPECT_CALL(rtp_1, MaxDataPayloadLength()) |
@@ -189,23 +197,18 @@ TEST(PayloadRouterTest, SetTargetSendBitrates) { |
modules.push_back(&rtp_1); |
modules.push_back(&rtp_2); |
PayloadRouter payload_router(modules, 42); |
- payload_router.SetSendingRtpModules(modules.size()); |
+ VideoCodec codec; |
+ codec.numberOfSimulcastStreams = 2; |
+ codec.simulcastStream[0].maxBitrate = 10; |
+ codec.simulcastStream[1].maxBitrate = 100; |
+ payload_router.SetSendCodec(codec); |
const uint32_t bitrate_1 = 10000; |
const uint32_t bitrate_2 = 76543; |
- std::vector<uint32_t> bitrates(2, bitrate_1); |
- bitrates[1] = bitrate_2; |
EXPECT_CALL(rtp_1, SetTargetSendBitrate(bitrate_1)) |
.Times(1); |
EXPECT_CALL(rtp_2, SetTargetSendBitrate(bitrate_2)) |
.Times(1); |
- payload_router.SetTargetSendBitrates(bitrates); |
- |
- bitrates.resize(1); |
- EXPECT_CALL(rtp_1, SetTargetSendBitrate(bitrate_1)) |
- .Times(1); |
- EXPECT_CALL(rtp_2, SetTargetSendBitrate(bitrate_2)) |
- .Times(0); |
- payload_router.SetTargetSendBitrates(bitrates); |
+ payload_router.SetTargetSendBitrate(bitrate_1 + bitrate_2); |
} |
} // namespace webrtc |