Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(28)

Side by Side Diff: webrtc/call/flexfec_receive_stream_unittest.cc

Issue 2553863003: Parse FlexFEC RTP headers in Call and add integration with BWE. (Closed)
Patch Set: danilchap comments 2. Created 4 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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 "webrtc/base/array_view.h"
11 #include "webrtc/base/basictypes.h" 12 #include "webrtc/base/basictypes.h"
12 #include "webrtc/call/flexfec_receive_stream.h" 13 #include "webrtc/call/flexfec_receive_stream.h"
13 #include "webrtc/modules/rtp_rtcp/include/flexfec_receiver.h" 14 #include "webrtc/modules/rtp_rtcp/include/flexfec_receiver.h"
14 #include "webrtc/modules/rtp_rtcp/source/byte_io.h" 15 #include "webrtc/modules/rtp_rtcp/source/byte_io.h"
16 #include "webrtc/modules/rtp_rtcp/source/rtp_packet_received.h"
15 #include "webrtc/modules/rtp_rtcp/mocks/mock_recovered_packet_receiver.h" 17 #include "webrtc/modules/rtp_rtcp/mocks/mock_recovered_packet_receiver.h"
16 #include "webrtc/test/gmock.h" 18 #include "webrtc/test/gmock.h"
17 #include "webrtc/test/gtest.h" 19 #include "webrtc/test/gtest.h"
18 20
19 namespace webrtc { 21 namespace webrtc {
20 22
23 namespace {
24
25 using ::testing::_;
26 using ::testing::Return;
27
28 RtpPacketReceived ParsePacket(rtc::ArrayView<const uint8_t> packet) {
29 RtpPacketReceived parsed_packet(nullptr);
30 EXPECT_TRUE(parsed_packet.Parse(packet));
31 return parsed_packet;
32 }
33
34 } // namespace
35
21 TEST(FlexfecReceiveStreamTest, ConstructDestruct) { 36 TEST(FlexfecReceiveStreamTest, ConstructDestruct) {
22 FlexfecReceiveStream::Config config; 37 FlexfecReceiveStream::Config config;
23 config.payload_type = 118; 38 config.payload_type = 118;
24 config.remote_ssrc = 424223; 39 config.remote_ssrc = 424223;
25 config.protected_media_ssrcs = {912512}; 40 config.protected_media_ssrcs = {912512};
26 MockRecoveredPacketReceiver callback; 41 MockRecoveredPacketReceiver recovered_packet_receiver;
27 42
28 internal::FlexfecReceiveStream receive_stream(config, &callback); 43 internal::FlexfecReceiveStream receive_stream(config,
44 &recovered_packet_receiver);
29 } 45 }
30 46
31 TEST(FlexfecReceiveStreamTest, StartStop) { 47 TEST(FlexfecReceiveStreamTest, StartStop) {
32 FlexfecReceiveStream::Config config; 48 FlexfecReceiveStream::Config config;
33 config.payload_type = 118; 49 config.payload_type = 118;
34 config.remote_ssrc = 1652392; 50 config.remote_ssrc = 1652392;
35 config.protected_media_ssrcs = {23300443}; 51 config.protected_media_ssrcs = {23300443};
36 MockRecoveredPacketReceiver callback; 52 MockRecoveredPacketReceiver recovered_packet_receiver;
37 internal::FlexfecReceiveStream receive_stream(config, &callback); 53 internal::FlexfecReceiveStream receive_stream(config,
54 &recovered_packet_receiver);
38 55
39 receive_stream.Start(); 56 receive_stream.Start();
40 receive_stream.Stop(); 57 receive_stream.Stop();
41 } 58 }
42 59
43 TEST(FlexfecReceiveStreamTest, DoesNotProcessPacketWhenNoMediaSsrcGiven) {
44 FlexfecReceiveStream::Config config;
45 config.payload_type = 118;
46 config.remote_ssrc = 424223;
47 config.protected_media_ssrcs = {};
48 MockRecoveredPacketReceiver callback;
49 internal::FlexfecReceiveStream receive_stream(config, &callback);
50 const uint8_t packet[] = {0x00, 0x11, 0x22, 0x33};
51 const size_t packet_length = sizeof(packet);
52
53 EXPECT_FALSE(
54 receive_stream.AddAndProcessReceivedPacket(packet, packet_length));
55 }
56
57 // Create a FlexFEC packet that protects a single media packet and ensure 60 // Create a FlexFEC packet that protects a single media packet and ensure
58 // that the callback is called. Correctness of recovery is checked in the 61 // that the callback is called. Correctness of recovery is checked in the
59 // FlexfecReceiver unit tests. 62 // FlexfecReceiver unit tests.
60 TEST(FlexfecReceiveStreamTest, RecoversPacketWhenStarted) { 63 TEST(FlexfecReceiveStreamTest, RecoversPacketWhenStarted) {
61 constexpr uint8_t kFlexfecPlType = 118; 64 constexpr uint8_t kFlexfecPlType = 118;
62 constexpr uint8_t kFlexfecSeqNum[] = {0x00, 0x01}; 65 constexpr uint8_t kFlexfecSeqNum[] = {0x00, 0x01};
63 constexpr uint8_t kFlexfecTs[] = {0x00, 0x11, 0x22, 0x33}; 66 constexpr uint8_t kFlexfecTs[] = {0x00, 0x11, 0x22, 0x33};
64 constexpr uint8_t kFlexfecSsrc[] = {0x00, 0x00, 0x00, 0x01}; 67 constexpr uint8_t kFlexfecSsrc[] = {0x00, 0x00, 0x00, 0x01};
65 constexpr uint8_t kMediaPlType = 107; 68 constexpr uint8_t kMediaPlType = 107;
66 constexpr uint8_t kMediaSeqNum[] = {0x00, 0x02}; 69 constexpr uint8_t kMediaSeqNum[] = {0x00, 0x02};
(...skipping 17 matching lines...) Expand all
84 kFlexfecSsrc[0], kFlexfecSsrc[1], kFlexfecSsrc[2], kFlexfecSsrc[3], 87 kFlexfecSsrc[0], kFlexfecSsrc[1], kFlexfecSsrc[2], kFlexfecSsrc[3],
85 // FlexFEC header. 88 // FlexFEC header.
86 0x00, kMediaPlType, kPayloadLength[0], kPayloadLength[1], 89 0x00, kMediaPlType, kPayloadLength[0], kPayloadLength[1],
87 kMediaTs[0], kMediaTs[1], kMediaTs[2], kMediaTs[3], 90 kMediaTs[0], kMediaTs[1], kMediaTs[2], kMediaTs[3],
88 kSsrcCount, kReservedBits, kReservedBits, kReservedBits, 91 kSsrcCount, kReservedBits, kReservedBits, kReservedBits,
89 kMediaSsrc[0], kMediaSsrc[1], kMediaSsrc[2], kMediaSsrc[3], 92 kMediaSsrc[0], kMediaSsrc[1], kMediaSsrc[2], kMediaSsrc[3],
90 kMediaSeqNum[0], kMediaSeqNum[1], kFlexfecPktMask[0], kFlexfecPktMask[1], 93 kMediaSeqNum[0], kMediaSeqNum[1], kFlexfecPktMask[0], kFlexfecPktMask[1],
91 // FEC payload. 94 // FEC payload.
92 kPayloadBits, kPayloadBits, kPayloadBits, kPayloadBits}; 95 kPayloadBits, kPayloadBits, kPayloadBits, kPayloadBits};
93 // clang-format on 96 // clang-format on
94 constexpr size_t kFlexfecPacketLength = sizeof(kFlexfecPacket);
95 97
96 FlexfecReceiveStream::Config config; 98 FlexfecReceiveStream::Config config;
97 config.payload_type = kFlexfecPlType; 99 config.payload_type = kFlexfecPlType;
98 config.remote_ssrc = ByteReader<uint32_t>::ReadBigEndian(kFlexfecSsrc); 100 config.remote_ssrc = ByteReader<uint32_t>::ReadBigEndian(kFlexfecSsrc);
99 config.protected_media_ssrcs = { 101 config.protected_media_ssrcs = {
100 ByteReader<uint32_t>::ReadBigEndian(kMediaSsrc)}; 102 ByteReader<uint32_t>::ReadBigEndian(kMediaSsrc)};
101 testing::StrictMock<MockRecoveredPacketReceiver> recovered_packet_receiver; 103 testing::StrictMock<MockRecoveredPacketReceiver> recovered_packet_receiver;
102 internal::FlexfecReceiveStream receive_stream(config, 104 internal::FlexfecReceiveStream receive_stream(config,
103 &recovered_packet_receiver); 105 &recovered_packet_receiver);
104 106
105 // Do not call back before being started. 107 // Do not call back before being started.
106 receive_stream.AddAndProcessReceivedPacket(kFlexfecPacket, 108 receive_stream.AddAndProcessReceivedPacket(ParsePacket(kFlexfecPacket));
107 kFlexfecPacketLength);
108 109
109 // Call back after being started. 110 // Call back after being started.
110 receive_stream.Start(); 111 receive_stream.Start();
111 EXPECT_CALL( 112 EXPECT_CALL(recovered_packet_receiver,
112 recovered_packet_receiver, 113 OnRecoveredPacket(_, kRtpHeaderSize + kPayloadLength[1]));
113 OnRecoveredPacket(::testing::_, kRtpHeaderSize + kPayloadLength[1])); 114 receive_stream.AddAndProcessReceivedPacket(ParsePacket(kFlexfecPacket));
114 receive_stream.AddAndProcessReceivedPacket(kFlexfecPacket,
115 kFlexfecPacketLength);
116 } 115 }
117 116
118 } // namespace webrtc 117 } // namespace webrtc
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698