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 |