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 |
11 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/app.h" | 11 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/app.h" |
12 | 12 |
13 #include <limits> | 13 #include "testing/gmock/include/gmock/gmock.h" |
14 #include "testing/gtest/include/gtest/gtest.h" | |
15 #include "webrtc/test/rtcp_packet_parser.h" | |
14 | 16 |
15 #include "testing/gtest/include/gtest/gtest.h" | 17 using testing::ElementsAreArray; |
16 #include "webrtc/modules/rtp_rtcp/source/rtcp_utility.h" | 18 using testing::make_tuple; |
17 | |
18 using webrtc::rtcp::App; | 19 using webrtc::rtcp::App; |
19 using webrtc::RTCPUtility::RtcpCommonHeader; | |
20 using webrtc::RTCPUtility::RtcpParseCommonHeader; | |
21 | 20 |
22 namespace webrtc { | 21 namespace webrtc { |
23 namespace { | 22 namespace { |
23 constexpr uint32_t kName = ((uint32_t)'n' << 24) | ((uint32_t)'a' << 16) | | |
24 ((uint32_t)'m' << 8) | (uint32_t)'e'; | |
25 constexpr uint8_t kSubtype = 0x1e; | |
26 constexpr uint32_t kSenderSsrc = 0x12345678; | |
27 constexpr uint8_t kData[] = {'t', 'e', 's', 't', 'd', 'a', 't', 'a'}; | |
28 constexpr uint8_t kPacketWithoutData[] = {0x9e, 204, 0x00, 0x02, | |
åsapersson
2016/07/26 09:58:15
App::kPacketType
danilchap
2016/07/26 11:37:02
Done.
First byte expanded too.
| |
29 0x12, 0x34, 0x56, 0x78, | |
30 'n', 'a', 'm', 'e'}; | |
31 constexpr uint8_t kPacketWithData[] = {0x9e, 204, 0x00, 0x04, | |
åsapersson
2016/07/26 09:58:15
App::kPacketType
danilchap
2016/07/26 11:37:02
Done.
| |
32 0x12, 0x34, 0x56, 0x78, | |
33 'n', 'a', 'm', 'e', | |
34 't', 'e', 's', 't', | |
35 'd', 'a', 't', 'a'}; | |
36 } // namespace | |
24 | 37 |
25 const uint32_t kName = ((uint32_t)'n' << 24) | ((uint32_t)'a' << 16) | | 38 TEST(RtcpPacketAppTest, CreateWithoutData) { |
26 ((uint32_t)'m' << 8) | (uint32_t)'e'; | |
27 const uint32_t kSenderSsrc = 0x12345678; | |
28 | |
29 class RtcpPacketAppTest : public ::testing::Test { | |
30 protected: | |
31 void BuildPacket() { packet = app.Build(); } | |
32 void ParsePacket() { | |
33 RtcpCommonHeader header; | |
34 EXPECT_TRUE(RtcpParseCommonHeader(packet.data(), packet.size(), &header)); | |
35 // Check there is exactly one RTCP packet in the buffer. | |
36 EXPECT_EQ(header.BlockSize(), packet.size()); | |
37 EXPECT_TRUE(parsed_.Parse( | |
38 header, packet.data() + RtcpCommonHeader::kHeaderSizeBytes)); | |
39 } | |
40 | |
41 App app; | 39 App app; |
42 rtc::Buffer packet; | 40 app.From(kSenderSsrc); |
43 const App& parsed() { return parsed_; } | 41 app.WithSubType(kSubtype); |
44 | |
45 private: | |
46 App parsed_; | |
47 }; | |
48 | |
49 TEST_F(RtcpPacketAppTest, WithNoData) { | |
50 app.WithSubType(30); | |
51 app.WithName(kName); | 42 app.WithName(kName); |
52 | 43 |
53 BuildPacket(); | 44 rtc::Buffer raw = app.Build(); |
54 ParsePacket(); | |
55 | 45 |
56 EXPECT_EQ(30U, parsed().sub_type()); | 46 EXPECT_THAT(make_tuple(raw.data(), raw.size()), |
57 EXPECT_EQ(kName, parsed().name()); | 47 ElementsAreArray(kPacketWithoutData)); |
58 EXPECT_EQ(0u, parsed().data_size()); | |
59 } | 48 } |
60 | 49 |
61 TEST_F(RtcpPacketAppTest, WithData) { | 50 TEST(RtcpPacketAppTest, ParseWithoutData) { |
62 app.From(kSenderSsrc); | 51 App parsed; |
63 app.WithSubType(30); | 52 EXPECT_TRUE(test::ParseSinglePacket(kPacketWithoutData, &parsed)); |
64 app.WithName(kName); | |
65 const uint8_t kData[] = {'t', 'e', 's', 't', 'd', 'a', 't', 'a'}; | |
66 const size_t kDataLength = sizeof(kData) / sizeof(kData[0]); | |
67 app.WithData(kData, kDataLength); | |
68 | 53 |
69 BuildPacket(); | 54 EXPECT_EQ(kSenderSsrc, parsed.ssrc()); |
70 ParsePacket(); | 55 EXPECT_EQ(kSubtype, parsed.sub_type()); |
71 | 56 EXPECT_EQ(kName, parsed.name()); |
72 EXPECT_EQ(30U, parsed().sub_type()); | 57 EXPECT_EQ(0u, parsed.data_size()); |
73 EXPECT_EQ(kName, parsed().name()); | |
74 EXPECT_EQ(kDataLength, parsed().data_size()); | |
75 EXPECT_EQ(0, memcmp(kData, parsed().data(), kDataLength)); | |
76 } | 58 } |
77 | 59 |
78 } // namespace | 60 TEST(RtcpPacketAppTest, CreateWithData) { |
61 App app; | |
62 app.From(kSenderSsrc); | |
63 app.WithSubType(kSubtype); | |
64 app.WithName(kName); | |
65 app.WithData(kData, sizeof(kData)); | |
66 | |
67 rtc::Buffer raw = app.Build(); | |
68 | |
69 EXPECT_THAT(make_tuple(raw.data(), raw.size()), | |
70 ElementsAreArray(kPacketWithData)); | |
71 } | |
72 | |
73 TEST(RtcpPacketAppTest, ParseWithData) { | |
74 App parsed; | |
75 EXPECT_TRUE(test::ParseSinglePacket(kPacketWithData, &parsed)); | |
76 | |
åsapersson
2016/07/26 09:58:15
maybe check kSenderSsrc
danilchap
2016/07/26 11:37:02
Done.
| |
77 EXPECT_EQ(kSubtype, parsed.sub_type()); | |
78 EXPECT_EQ(kName, parsed.name()); | |
79 EXPECT_THAT(make_tuple(parsed.data(), parsed.data_size()), | |
80 ElementsAreArray(kData)); | |
81 } | |
82 | |
83 TEST(RtcpPacketAppTest, ParseFailsOnTooSmallPacket) { | |
84 const uint8_t kTooSmallPacket[] = { | |
85 0x9e, App::kPacketType, 0x00, 0x01, 0x12, 0x34, 0x56, 0x78}; | |
86 App parsed; | |
87 EXPECT_FALSE(test::ParseSinglePacket(kTooSmallPacket, &parsed)); | |
88 } | |
89 | |
79 } // namespace webrtc | 90 } // namespace webrtc |
OLD | NEW |