OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2015 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 |
(...skipping 80 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
91 EXPECT_CALL(*channel_proxy_, EnableReceiveTransportSequenceNumber( | 91 EXPECT_CALL(*channel_proxy_, EnableReceiveTransportSequenceNumber( |
92 kTransportSequenceNumberId)) | 92 kTransportSequenceNumberId)) |
93 .Times(1); | 93 .Times(1); |
94 EXPECT_CALL(*channel_proxy_, | 94 EXPECT_CALL(*channel_proxy_, |
95 RegisterReceiverCongestionControlObjects(&packet_router_)) | 95 RegisterReceiverCongestionControlObjects(&packet_router_)) |
96 .Times(1); | 96 .Times(1); |
97 EXPECT_CALL(congestion_controller_, packet_router()) | 97 EXPECT_CALL(congestion_controller_, packet_router()) |
98 .WillOnce(Return(&packet_router_)); | 98 .WillOnce(Return(&packet_router_)); |
99 EXPECT_CALL(*channel_proxy_, ResetCongestionControlObjects()) | 99 EXPECT_CALL(*channel_proxy_, ResetCongestionControlObjects()) |
100 .Times(1); | 100 .Times(1); |
| 101 EXPECT_CALL(*channel_proxy_, RegisterExternalTransport(nullptr)) |
| 102 .Times(1); |
| 103 EXPECT_CALL(*channel_proxy_, DeRegisterExternalTransport()) |
| 104 .Times(1); |
101 return channel_proxy_; | 105 return channel_proxy_; |
102 })); | 106 })); |
103 stream_config_.voe_channel_id = kChannelId; | 107 stream_config_.voe_channel_id = kChannelId; |
104 stream_config_.rtp.local_ssrc = kLocalSsrc; | 108 stream_config_.rtp.local_ssrc = kLocalSsrc; |
105 stream_config_.rtp.remote_ssrc = kRemoteSsrc; | 109 stream_config_.rtp.remote_ssrc = kRemoteSsrc; |
106 stream_config_.rtp.extensions.push_back( | 110 stream_config_.rtp.extensions.push_back( |
107 RtpExtension(RtpExtension::kAbsSendTime, kAbsSendTimeId)); | 111 RtpExtension(RtpExtension::kAbsSendTime, kAbsSendTimeId)); |
108 stream_config_.rtp.extensions.push_back( | 112 stream_config_.rtp.extensions.push_back( |
109 RtpExtension(RtpExtension::kAudioLevel, kAudioLevelId)); | 113 RtpExtension(RtpExtension::kAudioLevel, kAudioLevelId)); |
110 stream_config_.rtp.extensions.push_back(RtpExtension( | 114 stream_config_.rtp.extensions.push_back(RtpExtension( |
111 RtpExtension::kTransportSequenceNumber, kTransportSequenceNumberId)); | 115 RtpExtension::kTransportSequenceNumber, kTransportSequenceNumberId)); |
112 } | 116 } |
113 | 117 |
114 MockCongestionController* congestion_controller() { | 118 MockCongestionController* congestion_controller() { |
115 return &congestion_controller_; | 119 return &congestion_controller_; |
116 } | 120 } |
117 MockRemoteBitrateEstimator* remote_bitrate_estimator() { | 121 MockRemoteBitrateEstimator* remote_bitrate_estimator() { |
118 return &remote_bitrate_estimator_; | 122 return &remote_bitrate_estimator_; |
119 } | 123 } |
120 AudioReceiveStream::Config& config() { return stream_config_; } | 124 AudioReceiveStream::Config& config() { return stream_config_; } |
121 rtc::scoped_refptr<AudioState> audio_state() { return audio_state_; } | 125 rtc::scoped_refptr<AudioState> audio_state() { return audio_state_; } |
122 MockVoiceEngine& voice_engine() { return voice_engine_; } | 126 MockVoiceEngine& voice_engine() { return voice_engine_; } |
| 127 MockVoEChannelProxy* channel_proxy() { return channel_proxy_; } |
123 | 128 |
124 void SetupMockForBweFeedback(bool send_side_bwe) { | 129 void SetupMockForBweFeedback(bool send_side_bwe) { |
125 EXPECT_CALL(congestion_controller_, | 130 EXPECT_CALL(congestion_controller_, |
126 GetRemoteBitrateEstimator(send_side_bwe)) | 131 GetRemoteBitrateEstimator(send_side_bwe)) |
127 .WillOnce(Return(&remote_bitrate_estimator_)); | 132 .WillOnce(Return(&remote_bitrate_estimator_)); |
128 EXPECT_CALL(remote_bitrate_estimator_, | 133 EXPECT_CALL(remote_bitrate_estimator_, |
129 RemoveStream(stream_config_.rtp.remote_ssrc)); | 134 RemoveStream(stream_config_.rtp.remote_ssrc)); |
130 } | 135 } |
131 | 136 |
132 void SetupMockForGetStats() { | 137 void SetupMockForGetStats() { |
(...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
174 it += 2; | 179 it += 2; |
175 const size_t kExtensionDataLength = kOneByteExtensionLength - 1; | 180 const size_t kExtensionDataLength = kOneByteExtensionLength - 1; |
176 uint32_t shifted_value = extension_value | 181 uint32_t shifted_value = extension_value |
177 << (8 * (kExtensionDataLength - value_length)); | 182 << (8 * (kExtensionDataLength - value_length)); |
178 *it = (id << 4) + (value_length - 1); | 183 *it = (id << 4) + (value_length - 1); |
179 ++it; | 184 ++it; |
180 ByteWriter<uint32_t, kExtensionDataLength>::WriteBigEndian(&(*it), | 185 ByteWriter<uint32_t, kExtensionDataLength>::WriteBigEndian(&(*it), |
181 shifted_value); | 186 shifted_value); |
182 } | 187 } |
183 | 188 |
184 std::vector<uint8_t> CreateRtpHeaderWithOneByteExtension( | 189 const std::vector<uint8_t> CreateRtpHeaderWithOneByteExtension( |
185 int extension_id, | 190 int extension_id, |
186 uint32_t extension_value, | 191 uint32_t extension_value, |
187 size_t value_length) { | 192 size_t value_length) { |
188 std::vector<uint8_t> header; | 193 std::vector<uint8_t> header; |
189 header.resize(webrtc::kRtpHeaderSize + kOneByteExtensionHeaderLength + | 194 header.resize(webrtc::kRtpHeaderSize + kOneByteExtensionHeaderLength + |
190 kOneByteExtensionLength); | 195 kOneByteExtensionLength); |
191 header[0] = 0x80; // Version 2. | 196 header[0] = 0x80; // Version 2. |
192 header[0] |= 0x10; // Set extension bit. | 197 header[0] |= 0x10; // Set extension bit. |
193 header[1] = 100; // Payload type. | 198 header[1] = 100; // Payload type. |
194 header[1] |= 0x80; // Marker bit is set. | 199 header[1] |= 0x80; // Marker bit is set. |
195 ByteWriter<uint16_t>::WriteBigEndian(&header[2], 0x1234); // Sequence number. | 200 ByteWriter<uint16_t>::WriteBigEndian(&header[2], 0x1234); // Sequence number. |
196 ByteWriter<uint32_t>::WriteBigEndian(&header[4], 0x5678); // Timestamp. | 201 ByteWriter<uint32_t>::WriteBigEndian(&header[4], 0x5678); // Timestamp. |
197 ByteWriter<uint32_t>::WriteBigEndian(&header[8], 0x4321); // SSRC. | 202 ByteWriter<uint32_t>::WriteBigEndian(&header[8], 0x4321); // SSRC. |
198 | 203 |
199 BuildOneByteExtension(header.begin() + webrtc::kRtpHeaderSize, extension_id, | 204 BuildOneByteExtension(header.begin() + webrtc::kRtpHeaderSize, extension_id, |
200 extension_value, value_length); | 205 extension_value, value_length); |
201 return header; | 206 return header; |
202 } | 207 } |
| 208 |
| 209 const std::vector<uint8_t> CreateRtcpSenderReport() { |
| 210 std::vector<uint8_t> packet; |
| 211 const size_t kRtcpSrLength = 28; // In bytes. |
| 212 packet.resize(kRtcpSrLength); |
| 213 packet[0] = 0x80; // Version 2. |
| 214 packet[1] = 0xc8; // PT = 200, SR. |
| 215 // Length in number of 32-bit words - 1. |
| 216 ByteWriter<uint16_t>::WriteBigEndian(&packet[2], 6); |
| 217 ByteWriter<uint32_t>::WriteBigEndian(&packet[4], kLocalSsrc); |
| 218 return packet; |
| 219 } |
203 } // namespace | 220 } // namespace |
204 | 221 |
205 TEST(AudioReceiveStreamTest, ConfigToString) { | 222 TEST(AudioReceiveStreamTest, ConfigToString) { |
206 AudioReceiveStream::Config config; | 223 AudioReceiveStream::Config config; |
207 config.rtp.remote_ssrc = kRemoteSsrc; | 224 config.rtp.remote_ssrc = kRemoteSsrc; |
208 config.rtp.local_ssrc = kLocalSsrc; | 225 config.rtp.local_ssrc = kLocalSsrc; |
209 config.rtp.extensions.push_back( | 226 config.rtp.extensions.push_back( |
210 RtpExtension(RtpExtension::kAbsSendTime, kAbsSendTimeId)); | 227 RtpExtension(RtpExtension::kAbsSendTime, kAbsSendTimeId)); |
211 config.voe_channel_id = kChannelId; | 228 config.voe_channel_id = kChannelId; |
212 EXPECT_EQ( | 229 EXPECT_EQ( |
213 "{rtp: {remote_ssrc: 1234, local_ssrc: 5678, extensions: [{name: " | 230 "{rtp: {remote_ssrc: 1234, local_ssrc: 5678, extensions: [{name: " |
214 "http://www.webrtc.org/experiments/rtp-hdrext/abs-send-time, id: 2}], " | 231 "http://www.webrtc.org/experiments/rtp-hdrext/abs-send-time, id: 2}], " |
215 "transport_cc: off}, " | 232 "transport_cc: off}, " |
216 "receive_transport: nullptr, rtcp_send_transport: nullptr, " | 233 "rtcp_send_transport: nullptr, " |
217 "voe_channel_id: 2}", | 234 "voe_channel_id: 2}", |
218 config.ToString()); | 235 config.ToString()); |
219 } | 236 } |
220 | 237 |
221 TEST(AudioReceiveStreamTest, ConstructDestruct) { | 238 TEST(AudioReceiveStreamTest, ConstructDestruct) { |
222 ConfigHelper helper; | 239 ConfigHelper helper; |
223 internal::AudioReceiveStream recv_stream( | 240 internal::AudioReceiveStream recv_stream( |
224 helper.congestion_controller(), helper.config(), helper.audio_state()); | 241 helper.congestion_controller(), helper.config(), helper.audio_state()); |
225 } | 242 } |
226 | 243 |
227 MATCHER_P(VerifyHeaderExtension, expected_extension, "") { | 244 MATCHER_P(VerifyHeaderExtension, expected_extension, "") { |
228 return arg.extension.hasAbsoluteSendTime == | 245 return arg.extension.hasAbsoluteSendTime == |
229 expected_extension.hasAbsoluteSendTime && | 246 expected_extension.hasAbsoluteSendTime && |
230 arg.extension.absoluteSendTime == | 247 arg.extension.absoluteSendTime == |
231 expected_extension.absoluteSendTime && | 248 expected_extension.absoluteSendTime && |
232 arg.extension.hasTransportSequenceNumber == | 249 arg.extension.hasTransportSequenceNumber == |
233 expected_extension.hasTransportSequenceNumber && | 250 expected_extension.hasTransportSequenceNumber && |
234 arg.extension.transportSequenceNumber == | 251 arg.extension.transportSequenceNumber == |
235 expected_extension.transportSequenceNumber; | 252 expected_extension.transportSequenceNumber; |
236 } | 253 } |
237 | 254 |
238 TEST(AudioReceiveStreamTest, AudioPacketUpdatesBweFeedback) { | 255 TEST(AudioReceiveStreamTest, ReceiveRtpPacket) { |
239 ConfigHelper helper; | 256 ConfigHelper helper; |
240 helper.config().rtp.transport_cc = true; | 257 helper.config().rtp.transport_cc = true; |
241 helper.SetupMockForBweFeedback(true); | 258 helper.SetupMockForBweFeedback(true); |
242 internal::AudioReceiveStream recv_stream( | 259 internal::AudioReceiveStream recv_stream( |
243 helper.congestion_controller(), helper.config(), helper.audio_state()); | 260 helper.congestion_controller(), helper.config(), helper.audio_state()); |
244 const int kTransportSequenceNumberValue = 1234; | 261 const int kTransportSequenceNumberValue = 1234; |
245 std::vector<uint8_t> rtp_packet = CreateRtpHeaderWithOneByteExtension( | 262 std::vector<uint8_t> rtp_packet = CreateRtpHeaderWithOneByteExtension( |
246 kTransportSequenceNumberId, kTransportSequenceNumberValue, 2); | 263 kTransportSequenceNumberId, kTransportSequenceNumberValue, 2); |
247 PacketTime packet_time(5678000, 0); | 264 PacketTime packet_time(5678000, 0); |
248 const size_t kExpectedHeaderLength = 20; | 265 const size_t kExpectedHeaderLength = 20; |
249 RTPHeaderExtension expected_extension; | 266 RTPHeaderExtension expected_extension; |
250 expected_extension.hasTransportSequenceNumber = true; | 267 expected_extension.hasTransportSequenceNumber = true; |
251 expected_extension.transportSequenceNumber = kTransportSequenceNumberValue; | 268 expected_extension.transportSequenceNumber = kTransportSequenceNumberValue; |
252 EXPECT_CALL(*helper.remote_bitrate_estimator(), | 269 EXPECT_CALL(*helper.remote_bitrate_estimator(), |
253 IncomingPacket(packet_time.timestamp / 1000, | 270 IncomingPacket(packet_time.timestamp / 1000, |
254 rtp_packet.size() - kExpectedHeaderLength, | 271 rtp_packet.size() - kExpectedHeaderLength, |
255 VerifyHeaderExtension(expected_extension), false)) | 272 VerifyHeaderExtension(expected_extension), false)) |
256 .Times(1); | 273 .Times(1); |
| 274 EXPECT_CALL(*helper.channel_proxy(), |
| 275 ReceivedRTPPacket(&rtp_packet[0], |
| 276 rtp_packet.size(), |
| 277 _)) |
| 278 .WillOnce(Return(true)); |
257 EXPECT_TRUE( | 279 EXPECT_TRUE( |
258 recv_stream.DeliverRtp(&rtp_packet[0], rtp_packet.size(), packet_time)); | 280 recv_stream.DeliverRtp(&rtp_packet[0], rtp_packet.size(), packet_time)); |
259 } | 281 } |
260 | 282 |
| 283 TEST(AudioReceiveStreamTest, ReceiveRtcpPacket) { |
| 284 ConfigHelper helper; |
| 285 helper.config().rtp.transport_cc = true; |
| 286 helper.SetupMockForBweFeedback(true); |
| 287 internal::AudioReceiveStream recv_stream( |
| 288 helper.congestion_controller(), helper.config(), helper.audio_state()); |
| 289 |
| 290 std::vector<uint8_t> rtcp_packet = CreateRtcpSenderReport(); |
| 291 EXPECT_CALL(*helper.channel_proxy(), |
| 292 ReceivedRTCPPacket(&rtcp_packet[0], rtcp_packet.size())) |
| 293 .WillOnce(Return(true)); |
| 294 EXPECT_TRUE(recv_stream.DeliverRtcp(&rtcp_packet[0], rtcp_packet.size())); |
| 295 } |
| 296 |
| 297 |
261 TEST(AudioReceiveStreamTest, GetStats) { | 298 TEST(AudioReceiveStreamTest, GetStats) { |
262 ConfigHelper helper; | 299 ConfigHelper helper; |
263 internal::AudioReceiveStream recv_stream( | 300 internal::AudioReceiveStream recv_stream( |
264 helper.congestion_controller(), helper.config(), helper.audio_state()); | 301 helper.congestion_controller(), helper.config(), helper.audio_state()); |
265 helper.SetupMockForGetStats(); | 302 helper.SetupMockForGetStats(); |
266 AudioReceiveStream::Stats stats = recv_stream.GetStats(); | 303 AudioReceiveStream::Stats stats = recv_stream.GetStats(); |
267 EXPECT_EQ(kRemoteSsrc, stats.remote_ssrc); | 304 EXPECT_EQ(kRemoteSsrc, stats.remote_ssrc); |
268 EXPECT_EQ(static_cast<int64_t>(kCallStats.bytesReceived), stats.bytes_rcvd); | 305 EXPECT_EQ(static_cast<int64_t>(kCallStats.bytesReceived), stats.bytes_rcvd); |
269 EXPECT_EQ(static_cast<uint32_t>(kCallStats.packetsReceived), | 306 EXPECT_EQ(static_cast<uint32_t>(kCallStats.packetsReceived), |
270 stats.packets_rcvd); | 307 stats.packets_rcvd); |
(...skipping 23 matching lines...) Expand all Loading... |
294 EXPECT_EQ(kAudioDecodeStats.calls_to_neteq, stats.decoding_calls_to_neteq); | 331 EXPECT_EQ(kAudioDecodeStats.calls_to_neteq, stats.decoding_calls_to_neteq); |
295 EXPECT_EQ(kAudioDecodeStats.decoded_normal, stats.decoding_normal); | 332 EXPECT_EQ(kAudioDecodeStats.decoded_normal, stats.decoding_normal); |
296 EXPECT_EQ(kAudioDecodeStats.decoded_plc, stats.decoding_plc); | 333 EXPECT_EQ(kAudioDecodeStats.decoded_plc, stats.decoding_plc); |
297 EXPECT_EQ(kAudioDecodeStats.decoded_cng, stats.decoding_cng); | 334 EXPECT_EQ(kAudioDecodeStats.decoded_cng, stats.decoding_cng); |
298 EXPECT_EQ(kAudioDecodeStats.decoded_plc_cng, stats.decoding_plc_cng); | 335 EXPECT_EQ(kAudioDecodeStats.decoded_plc_cng, stats.decoding_plc_cng); |
299 EXPECT_EQ(kCallStats.capture_start_ntp_time_ms_, | 336 EXPECT_EQ(kCallStats.capture_start_ntp_time_ms_, |
300 stats.capture_start_ntp_time_ms); | 337 stats.capture_start_ntp_time_ms); |
301 } | 338 } |
302 } // namespace test | 339 } // namespace test |
303 } // namespace webrtc | 340 } // namespace webrtc |
OLD | NEW |