Chromium Code Reviews| Index: webrtc/call/flexfec_receive_stream_unittest.cc |
| diff --git a/webrtc/call/flexfec_receive_stream_unittest.cc b/webrtc/call/flexfec_receive_stream_unittest.cc |
| index 68f2a2d0203640c6fb3df8b065614f8d7bf03951..170362916a1916351c413fad4eb2dd2e6ca407e5 100644 |
| --- a/webrtc/call/flexfec_receive_stream_unittest.cc |
| +++ b/webrtc/call/flexfec_receive_stream_unittest.cc |
| @@ -8,24 +8,40 @@ |
| * be found in the AUTHORS file in the root of the source tree. |
| */ |
| +#include "webrtc/base/array_view.h" |
| #include "webrtc/base/basictypes.h" |
| #include "webrtc/call/flexfec_receive_stream.h" |
| #include "webrtc/modules/rtp_rtcp/include/flexfec_receiver.h" |
| #include "webrtc/modules/rtp_rtcp/source/byte_io.h" |
| +#include "webrtc/modules/rtp_rtcp/source/rtp_packet_received.h" |
| #include "webrtc/modules/rtp_rtcp/mocks/mock_recovered_packet_receiver.h" |
| #include "webrtc/test/gmock.h" |
| #include "webrtc/test/gtest.h" |
| namespace webrtc { |
| +namespace { |
| + |
| +using ::testing::_; |
| +using ::testing::Return; |
| + |
| +RtpPacketReceived ParsePacket(rtc::ArrayView<const uint8_t> packet) { |
| + RtpPacketReceived parsed_packet(nullptr); |
| + EXPECT_TRUE(parsed_packet.Parse(packet)); |
| + return parsed_packet; |
| +} |
| + |
| +} // namespace |
| + |
| TEST(FlexfecReceiveStreamTest, ConstructDestruct) { |
| FlexfecReceiveStream::Config config; |
| config.payload_type = 118; |
| config.remote_ssrc = 424223; |
| config.protected_media_ssrcs = {912512}; |
| - MockRecoveredPacketReceiver callback; |
| + MockRecoveredPacketReceiver recovered_packet_receiver; |
| - internal::FlexfecReceiveStream receive_stream(config, &callback); |
| + internal::FlexfecReceiveStream receive_stream(config, |
| + &recovered_packet_receiver); |
| } |
| TEST(FlexfecReceiveStreamTest, StartStop) { |
| @@ -33,27 +49,14 @@ TEST(FlexfecReceiveStreamTest, StartStop) { |
| config.payload_type = 118; |
| config.remote_ssrc = 1652392; |
| config.protected_media_ssrcs = {23300443}; |
| - MockRecoveredPacketReceiver callback; |
| - internal::FlexfecReceiveStream receive_stream(config, &callback); |
| + MockRecoveredPacketReceiver recovered_packet_receiver; |
| + internal::FlexfecReceiveStream receive_stream(config, |
| + &recovered_packet_receiver); |
| receive_stream.Start(); |
| receive_stream.Stop(); |
| } |
| -TEST(FlexfecReceiveStreamTest, DoesNotProcessPacketWhenNoMediaSsrcGiven) { |
| - FlexfecReceiveStream::Config config; |
| - config.payload_type = 118; |
| - config.remote_ssrc = 424223; |
| - config.protected_media_ssrcs = {}; |
| - MockRecoveredPacketReceiver callback; |
| - internal::FlexfecReceiveStream receive_stream(config, &callback); |
| - const uint8_t packet[] = {0x00, 0x11, 0x22, 0x33}; |
| - const size_t packet_length = sizeof(packet); |
| - |
| - EXPECT_FALSE( |
| - receive_stream.AddAndProcessReceivedPacket(packet, packet_length)); |
| -} |
| - |
| // Create a FlexFEC packet that protects a single media packet and ensure |
| // that the callback is called. Correctness of recovery is checked in the |
| // FlexfecReceiver unit tests. |
| @@ -103,16 +106,15 @@ TEST(FlexfecReceiveStreamTest, RecoversPacketWhenStarted) { |
| &recovered_packet_receiver); |
| // Do not call back before being started. |
| - receive_stream.AddAndProcessReceivedPacket(kFlexfecPacket, |
| - kFlexfecPacketLength); |
| + receive_stream.AddAndProcessReceivedPacket(ParsePacket( |
| + rtc::ArrayView<const uint8_t>(kFlexfecPacket, kFlexfecPacketLength))); |
|
danilchap
2016/12/14 13:39:02
just ParsePacket(kFlexfecPacket)
ArrayView would d
brandtr
2016/12/14 14:08:31
Nice!
|
| // Call back after being started. |
| receive_stream.Start(); |
| - EXPECT_CALL( |
| - recovered_packet_receiver, |
| - OnRecoveredPacket(::testing::_, kRtpHeaderSize + kPayloadLength[1])); |
| - receive_stream.AddAndProcessReceivedPacket(kFlexfecPacket, |
| - kFlexfecPacketLength); |
| + EXPECT_CALL(recovered_packet_receiver, |
| + OnRecoveredPacket(_, kRtpHeaderSize + kPayloadLength[1])); |
| + receive_stream.AddAndProcessReceivedPacket(ParsePacket( |
|
danilchap
2016/12/14 13:39:02
ditto
brandtr
2016/12/14 14:08:31
Done.
|
| + rtc::ArrayView<const uint8_t>(kFlexfecPacket, kFlexfecPacketLength))); |
| } |
| } // namespace webrtc |