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

Side by Side Diff: webrtc/modules/rtp_rtcp/source/rtcp_packet/bye_unittest.cc

Issue 2372713005: Revert of Unify rtcp packet setters (Closed)
Patch Set: Created 4 years, 2 months 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) 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
11 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/bye.h" 11 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/bye.h"
12 12
13 #include "testing/gmock/include/gmock/gmock.h" 13 #include "testing/gmock/include/gmock/gmock.h"
14 #include "testing/gtest/include/gtest/gtest.h" 14 #include "testing/gtest/include/gtest/gtest.h"
15 #include "webrtc/test/rtcp_packet_parser.h" 15 #include "webrtc/test/rtcp_packet_parser.h"
16 16
17 using testing::ElementsAre; 17 using testing::ElementsAre;
18 using webrtc::rtcp::Bye; 18 using webrtc::rtcp::Bye;
19 19
20 namespace webrtc { 20 namespace webrtc {
21 namespace { 21 namespace {
22 const uint32_t kSenderSsrc = 0x12345678; 22 const uint32_t kSenderSsrc = 0x12345678;
23 const uint32_t kCsrc1 = 0x22232425; 23 const uint32_t kCsrc1 = 0x22232425;
24 const uint32_t kCsrc2 = 0x33343536; 24 const uint32_t kCsrc2 = 0x33343536;
25 } // namespace 25 } // namespace
26 26
27 TEST(RtcpPacketByeTest, CreateAndParseWithoutReason) { 27 TEST(RtcpPacketByeTest, CreateAndParseWithoutReason) {
28 Bye bye; 28 Bye bye;
29 bye.SetSenderSsrc(kSenderSsrc); 29 bye.From(kSenderSsrc);
30 30
31 rtc::Buffer raw = bye.Build(); 31 rtc::Buffer raw = bye.Build();
32 Bye parsed_bye; 32 Bye parsed_bye;
33 EXPECT_TRUE(test::ParseSinglePacket(raw, &parsed_bye)); 33 EXPECT_TRUE(test::ParseSinglePacket(raw, &parsed_bye));
34 34
35 EXPECT_EQ(kSenderSsrc, parsed_bye.sender_ssrc()); 35 EXPECT_EQ(kSenderSsrc, parsed_bye.sender_ssrc());
36 EXPECT_TRUE(parsed_bye.csrcs().empty()); 36 EXPECT_TRUE(parsed_bye.csrcs().empty());
37 EXPECT_TRUE(parsed_bye.reason().empty()); 37 EXPECT_TRUE(parsed_bye.reason().empty());
38 } 38 }
39 39
40 TEST(RtcpPacketByeTest, CreateAndParseWithCsrcs) { 40 TEST(RtcpPacketByeTest, CreateAndParseWithCsrcs) {
41 Bye bye; 41 Bye bye;
42 bye.SetSenderSsrc(kSenderSsrc); 42 bye.From(kSenderSsrc);
43 EXPECT_TRUE(bye.SetCsrcs({kCsrc1, kCsrc2})); 43 EXPECT_TRUE(bye.WithCsrc(kCsrc1));
44 EXPECT_TRUE(bye.WithCsrc(kCsrc2));
44 EXPECT_TRUE(bye.reason().empty()); 45 EXPECT_TRUE(bye.reason().empty());
45 46
46 rtc::Buffer raw = bye.Build(); 47 rtc::Buffer raw = bye.Build();
47 Bye parsed_bye; 48 Bye parsed_bye;
48 EXPECT_TRUE(test::ParseSinglePacket(raw, &parsed_bye)); 49 EXPECT_TRUE(test::ParseSinglePacket(raw, &parsed_bye));
49 50
50 EXPECT_EQ(kSenderSsrc, parsed_bye.sender_ssrc()); 51 EXPECT_EQ(kSenderSsrc, parsed_bye.sender_ssrc());
51 EXPECT_THAT(parsed_bye.csrcs(), ElementsAre(kCsrc1, kCsrc2)); 52 EXPECT_THAT(parsed_bye.csrcs(), ElementsAre(kCsrc1, kCsrc2));
52 EXPECT_TRUE(parsed_bye.reason().empty()); 53 EXPECT_TRUE(parsed_bye.reason().empty());
53 } 54 }
54 55
55 TEST(RtcpPacketByeTest, CreateAndParseWithCsrcsAndAReason) { 56 TEST(RtcpPacketByeTest, CreateAndParseWithCsrcsAndAReason) {
56 Bye bye; 57 Bye bye;
57 const std::string kReason = "Some Reason"; 58 const std::string kReason = "Some Reason";
58 59
59 bye.SetSenderSsrc(kSenderSsrc); 60 bye.From(kSenderSsrc);
60 EXPECT_TRUE(bye.SetCsrcs({kCsrc1, kCsrc2})); 61 EXPECT_TRUE(bye.WithCsrc(kCsrc1));
61 bye.SetReason(kReason); 62 EXPECT_TRUE(bye.WithCsrc(kCsrc2));
63 bye.WithReason(kReason);
62 64
63 rtc::Buffer raw = bye.Build(); 65 rtc::Buffer raw = bye.Build();
64 Bye parsed_bye; 66 Bye parsed_bye;
65 EXPECT_TRUE(test::ParseSinglePacket(raw, &parsed_bye)); 67 EXPECT_TRUE(test::ParseSinglePacket(raw, &parsed_bye));
66 68
67 EXPECT_EQ(kSenderSsrc, parsed_bye.sender_ssrc()); 69 EXPECT_EQ(kSenderSsrc, parsed_bye.sender_ssrc());
68 EXPECT_THAT(parsed_bye.csrcs(), ElementsAre(kCsrc1, kCsrc2)); 70 EXPECT_THAT(parsed_bye.csrcs(), ElementsAre(kCsrc1, kCsrc2));
69 EXPECT_EQ(kReason, parsed_bye.reason()); 71 EXPECT_EQ(kReason, parsed_bye.reason());
70 } 72 }
71 73
72 TEST(RtcpPacketByeTest, CreateWithTooManyCsrcs) { 74 TEST(RtcpPacketByeTest, CreateWithTooManyCsrcs) {
73 Bye bye; 75 Bye bye;
74 bye.SetSenderSsrc(kSenderSsrc); 76 bye.From(kSenderSsrc);
75 const int kMaxCsrcs = (1 << 5) - 2; // 5 bit len, first item is sender SSRC. 77 const int kMaxCsrcs = (1 << 5) - 2; // 5 bit len, first item is sender SSRC.
76 EXPECT_TRUE(bye.SetCsrcs(std::vector<uint32_t>(kMaxCsrcs, kCsrc1))); 78 for (int i = 0; i < kMaxCsrcs; ++i) {
77 EXPECT_FALSE(bye.SetCsrcs(std::vector<uint32_t>(kMaxCsrcs + 1, kCsrc1))); 79 EXPECT_TRUE(bye.WithCsrc(i));
80 }
81 EXPECT_FALSE(bye.WithCsrc(kMaxCsrcs));
78 } 82 }
79 83
80 TEST(RtcpPacketByeTest, CreateAndParseWithAReason) { 84 TEST(RtcpPacketByeTest, CreateAndParseWithAReason) {
81 Bye bye; 85 Bye bye;
82 const std::string kReason = "Some Random Reason"; 86 const std::string kReason = "Some Random Reason";
83 87
84 bye.SetSenderSsrc(kSenderSsrc); 88 bye.From(kSenderSsrc);
85 bye.SetReason(kReason); 89 bye.WithReason(kReason);
86 90
87 rtc::Buffer raw = bye.Build(); 91 rtc::Buffer raw = bye.Build();
88 Bye parsed_bye; 92 Bye parsed_bye;
89 EXPECT_TRUE(test::ParseSinglePacket(raw, &parsed_bye)); 93 EXPECT_TRUE(test::ParseSinglePacket(raw, &parsed_bye));
90 94
91 EXPECT_EQ(kSenderSsrc, parsed_bye.sender_ssrc()); 95 EXPECT_EQ(kSenderSsrc, parsed_bye.sender_ssrc());
92 EXPECT_TRUE(parsed_bye.csrcs().empty()); 96 EXPECT_TRUE(parsed_bye.csrcs().empty());
93 EXPECT_EQ(kReason, parsed_bye.reason()); 97 EXPECT_EQ(kReason, parsed_bye.reason());
94 } 98 }
95 99
96 TEST(RtcpPacketByeTest, CreateAndParseWithReasons) { 100 TEST(RtcpPacketByeTest, CreateAndParseWithReasons) {
97 // Test that packet creation/parsing behave with reasons of different length 101 // Test that packet creation/parsing behave with reasons of different length
98 // both when it require padding and when it does not. 102 // both when it require padding and when it does not.
99 for (size_t reminder = 0; reminder < 4; ++reminder) { 103 for (size_t reminder = 0; reminder < 4; ++reminder) {
100 const std::string kReason(4 + reminder, 'a' + reminder); 104 const std::string kReason(4 + reminder, 'a' + reminder);
101 Bye bye; 105 Bye bye;
102 bye.SetSenderSsrc(kSenderSsrc); 106 bye.From(kSenderSsrc);
103 bye.SetReason(kReason); 107 bye.WithReason(kReason);
104 108
105 rtc::Buffer raw = bye.Build(); 109 rtc::Buffer raw = bye.Build();
106 Bye parsed_bye; 110 Bye parsed_bye;
107 EXPECT_TRUE(test::ParseSinglePacket(raw, &parsed_bye)); 111 EXPECT_TRUE(test::ParseSinglePacket(raw, &parsed_bye));
108 112
109 EXPECT_EQ(kReason, parsed_bye.reason()); 113 EXPECT_EQ(kReason, parsed_bye.reason());
110 } 114 }
111 } 115 }
112 116
113 TEST(RtcpPacketByeTest, ParseEmptyPacket) { 117 TEST(RtcpPacketByeTest, ParseEmptyPacket) {
114 uint8_t kEmptyPacket[] = {0x80, Bye::kPacketType, 0, 0}; 118 uint8_t kEmptyPacket[] = {0x80, Bye::kPacketType, 0, 0};
115 Bye parsed_bye; 119 Bye parsed_bye;
116 EXPECT_TRUE(test::ParseSinglePacket(kEmptyPacket, &parsed_bye)); 120 EXPECT_TRUE(test::ParseSinglePacket(kEmptyPacket, &parsed_bye));
117 EXPECT_EQ(0u, parsed_bye.sender_ssrc()); 121 EXPECT_EQ(0u, parsed_bye.sender_ssrc());
118 EXPECT_TRUE(parsed_bye.csrcs().empty()); 122 EXPECT_TRUE(parsed_bye.csrcs().empty());
119 EXPECT_TRUE(parsed_bye.reason().empty()); 123 EXPECT_TRUE(parsed_bye.reason().empty());
120 } 124 }
121 125
122 TEST(RtcpPacketByeTest, ParseFailOnInvalidSrcCount) { 126 TEST(RtcpPacketByeTest, ParseFailOnInvalidSrcCount) {
123 Bye bye; 127 Bye bye;
124 bye.SetSenderSsrc(kSenderSsrc); 128 bye.From(kSenderSsrc);
125 129
126 rtc::Buffer raw = bye.Build(); 130 rtc::Buffer raw = bye.Build();
127 raw[0]++; // Damage the packet: increase ssrc count by one. 131 raw[0]++; // Damage the packet: increase ssrc count by one.
128 132
129 Bye parsed_bye; 133 Bye parsed_bye;
130 EXPECT_FALSE(test::ParseSinglePacket(raw, &parsed_bye)); 134 EXPECT_FALSE(test::ParseSinglePacket(raw, &parsed_bye));
131 } 135 }
132 136
133 TEST(RtcpPacketByeTest, ParseFailOnInvalidReasonLength) { 137 TEST(RtcpPacketByeTest, ParseFailOnInvalidReasonLength) {
134 Bye bye; 138 Bye bye;
135 bye.SetSenderSsrc(kSenderSsrc); 139 bye.From(kSenderSsrc);
136 bye.SetReason("18 characters long"); 140 bye.WithReason("18 characters long");
137 141
138 rtc::Buffer raw = bye.Build(); 142 rtc::Buffer raw = bye.Build();
139 // Damage the packet: decrease payload size by 4 bytes 143 // Damage the packet: decrease payload size by 4 bytes
140 raw[3]--; 144 raw[3]--;
141 raw.SetSize(raw.size() - 4); 145 raw.SetSize(raw.size() - 4);
142 146
143 Bye parsed_bye; 147 Bye parsed_bye;
144 EXPECT_FALSE(test::ParseSinglePacket(raw, &parsed_bye)); 148 EXPECT_FALSE(test::ParseSinglePacket(raw, &parsed_bye));
145 } 149 }
146 150
147 } // namespace webrtc 151 } // namespace webrtc
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698