| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| 11 #include <stdint.h> | 11 #include <stdint.h> |
| 12 | 12 |
| 13 #include "webrtc/base/array_view.h" | 13 #include "webrtc/base/array_view.h" |
| 14 #include "webrtc/call/flexfec_receive_stream_impl.h" | 14 #include "webrtc/call/flexfec_receive_stream_impl.h" |
| 15 #include "webrtc/call/rtp_stream_receiver_controller.h" |
| 15 #include "webrtc/modules/pacing/packet_router.h" | 16 #include "webrtc/modules/pacing/packet_router.h" |
| 16 #include "webrtc/modules/rtp_rtcp/include/flexfec_receiver.h" | 17 #include "webrtc/modules/rtp_rtcp/include/flexfec_receiver.h" |
| 17 #include "webrtc/modules/rtp_rtcp/mocks/mock_recovered_packet_receiver.h" | 18 #include "webrtc/modules/rtp_rtcp/mocks/mock_recovered_packet_receiver.h" |
| 18 #include "webrtc/modules/rtp_rtcp/mocks/mock_rtcp_rtt_stats.h" | 19 #include "webrtc/modules/rtp_rtcp/mocks/mock_rtcp_rtt_stats.h" |
| 19 #include "webrtc/modules/rtp_rtcp/source/byte_io.h" | 20 #include "webrtc/modules/rtp_rtcp/source/byte_io.h" |
| 20 #include "webrtc/modules/rtp_rtcp/source/rtp_header_extensions.h" | 21 #include "webrtc/modules/rtp_rtcp/source/rtp_header_extensions.h" |
| 21 #include "webrtc/modules/utility/include/mock/mock_process_thread.h" | 22 #include "webrtc/modules/utility/include/mock/mock_process_thread.h" |
| 22 #include "webrtc/test/gmock.h" | 23 #include "webrtc/test/gmock.h" |
| 23 #include "webrtc/test/gtest.h" | 24 #include "webrtc/test/gtest.h" |
| 24 #include "webrtc/test/mock_transport.h" | 25 #include "webrtc/test/mock_transport.h" |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 70 EXPECT_TRUE(config.IsCompleteAndEnabled()); | 71 EXPECT_TRUE(config.IsCompleteAndEnabled()); |
| 71 | 72 |
| 72 config.protected_media_ssrcs.push_back(33423423); | 73 config.protected_media_ssrcs.push_back(33423423); |
| 73 EXPECT_FALSE(config.IsCompleteAndEnabled()); | 74 EXPECT_FALSE(config.IsCompleteAndEnabled()); |
| 74 } | 75 } |
| 75 | 76 |
| 76 class FlexfecReceiveStreamTest : public ::testing::Test { | 77 class FlexfecReceiveStreamTest : public ::testing::Test { |
| 77 protected: | 78 protected: |
| 78 FlexfecReceiveStreamTest() | 79 FlexfecReceiveStreamTest() |
| 79 : config_(CreateDefaultConfig(&rtcp_send_transport_)), | 80 : config_(CreateDefaultConfig(&rtcp_send_transport_)), |
| 80 receive_stream_(config_, | 81 receive_stream_(&rtp_stream_receiver_controller_, |
| 82 config_, |
| 81 &recovered_packet_receiver_, | 83 &recovered_packet_receiver_, |
| 82 &rtt_stats_, | 84 &rtt_stats_, |
| 83 &process_thread_) {} | 85 &process_thread_) {} |
| 84 | 86 |
| 85 MockTransport rtcp_send_transport_; | 87 MockTransport rtcp_send_transport_; |
| 86 FlexfecReceiveStream::Config config_; | 88 FlexfecReceiveStream::Config config_; |
| 87 MockRecoveredPacketReceiver recovered_packet_receiver_; | 89 MockRecoveredPacketReceiver recovered_packet_receiver_; |
| 88 MockRtcpRttStats rtt_stats_; | 90 MockRtcpRttStats rtt_stats_; |
| 89 MockProcessThread process_thread_; | 91 MockProcessThread process_thread_; |
| 90 | 92 RtpStreamReceiverController rtp_stream_receiver_controller_; |
| 91 FlexfecReceiveStreamImpl receive_stream_; | 93 FlexfecReceiveStreamImpl receive_stream_; |
| 92 }; | 94 }; |
| 93 | 95 |
| 94 TEST_F(FlexfecReceiveStreamTest, ConstructDestruct) {} | 96 TEST_F(FlexfecReceiveStreamTest, ConstructDestruct) {} |
| 95 | 97 |
| 96 TEST_F(FlexfecReceiveStreamTest, StartStop) { | 98 TEST_F(FlexfecReceiveStreamTest, StartStop) { |
| 97 receive_stream_.Start(); | 99 receive_stream_.Start(); |
| 98 receive_stream_.Stop(); | 100 receive_stream_.Stop(); |
| 99 } | 101 } |
| 100 | 102 |
| (...skipping 26 matching lines...) Expand all Loading... |
| 127 0x00, kMediaPlType, kPayloadLength[0], kPayloadLength[1], | 129 0x00, kMediaPlType, kPayloadLength[0], kPayloadLength[1], |
| 128 kMediaTs[0], kMediaTs[1], kMediaTs[2], kMediaTs[3], | 130 kMediaTs[0], kMediaTs[1], kMediaTs[2], kMediaTs[3], |
| 129 kSsrcCount, kReservedBits, kReservedBits, kReservedBits, | 131 kSsrcCount, kReservedBits, kReservedBits, kReservedBits, |
| 130 kMediaSsrc[0], kMediaSsrc[1], kMediaSsrc[2], kMediaSsrc[3], | 132 kMediaSsrc[0], kMediaSsrc[1], kMediaSsrc[2], kMediaSsrc[3], |
| 131 kMediaSeqNum[0], kMediaSeqNum[1], kFlexfecPktMask[0], kFlexfecPktMask[1], | 133 kMediaSeqNum[0], kMediaSeqNum[1], kFlexfecPktMask[0], kFlexfecPktMask[1], |
| 132 // FEC payload. | 134 // FEC payload. |
| 133 kPayloadBits, kPayloadBits, kPayloadBits, kPayloadBits}; | 135 kPayloadBits, kPayloadBits, kPayloadBits, kPayloadBits}; |
| 134 // clang-format on | 136 // clang-format on |
| 135 | 137 |
| 136 testing::StrictMock<MockRecoveredPacketReceiver> recovered_packet_receiver; | 138 testing::StrictMock<MockRecoveredPacketReceiver> recovered_packet_receiver; |
| 137 FlexfecReceiveStreamImpl receive_stream(config_, &recovered_packet_receiver, | 139 FlexfecReceiveStreamImpl receive_stream(&rtp_stream_receiver_controller_, |
| 140 config_, &recovered_packet_receiver, |
| 138 &rtt_stats_, &process_thread_); | 141 &rtt_stats_, &process_thread_); |
| 139 | 142 |
| 140 // Do not call back before being started. | 143 // Do not call back before being started. |
| 141 receive_stream.OnRtpPacket(ParsePacket(kFlexfecPacket)); | 144 receive_stream.OnRtpPacket(ParsePacket(kFlexfecPacket)); |
| 142 | 145 |
| 143 // Call back after being started. | 146 // Call back after being started. |
| 144 receive_stream.Start(); | 147 receive_stream.Start(); |
| 145 EXPECT_CALL( | 148 EXPECT_CALL( |
| 146 recovered_packet_receiver, | 149 recovered_packet_receiver, |
| 147 OnRecoveredPacket(::testing::_, kRtpHeaderSize + kPayloadLength[1])); | 150 OnRecoveredPacket(::testing::_, kRtpHeaderSize + kPayloadLength[1])); |
| 148 receive_stream.OnRtpPacket(ParsePacket(kFlexfecPacket)); | 151 receive_stream.OnRtpPacket(ParsePacket(kFlexfecPacket)); |
| 149 } | 152 } |
| 150 | 153 |
| 151 } // namespace webrtc | 154 } // namespace webrtc |
| OLD | NEW |