OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <gtest/gtest.h> | 5 #include <gtest/gtest.h> |
6 | 6 |
7 #include "base/memory/scoped_ptr.h" | 7 #include "base/memory/scoped_ptr.h" |
8 #include "media/cast/rtp_receiver/rtp_parser/rtp_parser.h" | 8 #include "media/cast/rtp_receiver/rtp_parser/rtp_parser.h" |
9 #include "media/cast/rtp_receiver/rtp_parser/test/rtp_packet_builder.h" | 9 #include "media/cast/rtp_receiver/rtp_parser/test/rtp_packet_builder.h" |
10 #include "media/cast/rtp_receiver/rtp_receiver.h" | 10 #include "media/cast/rtp_receiver/rtp_receiver.h" |
11 #include "media/cast/rtp_receiver/rtp_receiver_defines.h" | 11 #include "media/cast/rtp_receiver/rtp_receiver_defines.h" |
12 | 12 |
13 namespace media { | 13 namespace media { |
14 namespace cast { | 14 namespace cast { |
15 | 15 |
16 static const size_t kPacketLength = 1500; | 16 static const size_t kPacketLength = 1500; |
17 static const int kTestPayloadType = 127; | 17 static const int kTestPayloadType = 127; |
18 static const uint32 kTestSsrc = 1234; | 18 static const uint32 kTestSsrc = 1234; |
19 static const uint32 kTestTimestamp = 111111; | 19 static const uint32 kTestTimestamp = 111111; |
20 static const uint16 kTestSeqNum = 4321; | 20 static const uint16 kTestSeqNum = 4321; |
21 static const uint8 kRefFrameId = 17; | 21 static const uint8 kRefFrameId = 17; |
22 | 22 |
23 class RtpDataTest : public RtpData { | 23 class RtpDataTest : public RtpData { |
24 public: | 24 public: |
25 RtpDataTest() { | 25 RtpDataTest() { expected_header_.reset(new RtpCastHeader()); } |
26 expected_header_.reset(new RtpCastHeader()); | |
27 } | |
28 | 26 |
29 virtual ~RtpDataTest() {} | 27 virtual ~RtpDataTest() {} |
30 | 28 |
31 void SetExpectedHeader(const RtpCastHeader& cast_header) { | 29 void SetExpectedHeader(const RtpCastHeader& cast_header) { |
32 memcpy(expected_header_.get(), &cast_header, sizeof(RtpCastHeader)); | 30 memcpy(expected_header_.get(), &cast_header, sizeof(RtpCastHeader)); |
33 } | 31 } |
34 | 32 |
35 virtual void OnReceivedPayloadData(const uint8* payloadData, | 33 virtual void OnReceivedPayloadData(const uint8* payloadData, |
36 size_t payloadSize, | 34 size_t payloadSize, |
37 const RtpCastHeader* rtpHeader) OVERRIDE { | 35 const RtpCastHeader* rtpHeader) OVERRIDE { |
38 VerifyCommonHeader(*rtpHeader); | 36 VerifyCommonHeader(*rtpHeader); |
39 VerifyCastHeader(*rtpHeader); | 37 VerifyCastHeader(*rtpHeader); |
40 } | 38 } |
41 | 39 |
42 void VerifyCommonHeader(const RtpCastHeader& parsed_header) { | 40 void VerifyCommonHeader(const RtpCastHeader& parsed_header) { |
43 EXPECT_EQ(expected_header_->packet_id == expected_header_->max_packet_id, | 41 EXPECT_EQ(expected_header_->packet_id == expected_header_->max_packet_id, |
44 parsed_header.webrtc.header.markerBit); | 42 parsed_header.webrtc.header.markerBit); |
45 EXPECT_EQ(kTestPayloadType, parsed_header.webrtc.header.payloadType); | 43 EXPECT_EQ(kTestPayloadType, parsed_header.webrtc.header.payloadType); |
46 EXPECT_EQ(kTestSsrc, parsed_header.webrtc.header.ssrc); | 44 EXPECT_EQ(kTestSsrc, parsed_header.webrtc.header.ssrc); |
47 EXPECT_EQ(0, parsed_header.webrtc.header.numCSRCs); | 45 EXPECT_EQ(0, parsed_header.webrtc.header.numCSRCs); |
48 } | 46 } |
49 | 47 |
50 void VerifyCastHeader(const RtpCastHeader& parsed_header) { | 48 void VerifyCastHeader(const RtpCastHeader& parsed_header) { |
51 EXPECT_EQ(expected_header_->is_key_frame, parsed_header.is_key_frame); | 49 EXPECT_EQ(expected_header_->is_key_frame, parsed_header.is_key_frame); |
52 EXPECT_EQ(expected_header_->frame_id, parsed_header.frame_id); | 50 EXPECT_EQ(expected_header_->frame_id, parsed_header.frame_id); |
53 EXPECT_EQ(expected_header_->packet_id, parsed_header.packet_id); | 51 EXPECT_EQ(expected_header_->packet_id, parsed_header.packet_id); |
54 EXPECT_EQ(expected_header_->max_packet_id, parsed_header.max_packet_id); | 52 EXPECT_EQ(expected_header_->max_packet_id, parsed_header.max_packet_id); |
55 EXPECT_EQ(expected_header_->is_reference, parsed_header.is_reference); | 53 EXPECT_EQ(expected_header_->is_reference, parsed_header.is_reference); |
56 } | 54 } |
57 | 55 |
58 private: | 56 private: |
59 scoped_ptr<RtpCastHeader> expected_header_; | 57 scoped_ptr<RtpCastHeader> expected_header_; |
| 58 |
| 59 DISALLOW_COPY_AND_ASSIGN(RtpDataTest); |
60 }; | 60 }; |
61 | 61 |
62 class RtpParserTest : public ::testing::Test { | 62 class RtpParserTest : public ::testing::Test { |
63 protected: | 63 protected: |
64 RtpParserTest() { | 64 RtpParserTest() { |
65 PopulateConfig(); | 65 PopulateConfig(); |
66 rtp_data_.reset(new RtpDataTest()); | 66 rtp_data_.reset(new RtpDataTest()); |
67 rtp_parser_.reset(new RtpParser(rtp_data_.get(), config_)); | 67 rtp_parser_.reset(new RtpParser(rtp_data_.get(), config_)); |
68 } | |
69 | |
70 virtual ~RtpParserTest() {} | |
71 | |
72 virtual void SetUp() { | |
73 cast_header_.is_reference = true; | 68 cast_header_.is_reference = true; |
74 cast_header_.reference_frame_id = kRefFrameId; | 69 cast_header_.reference_frame_id = kRefFrameId; |
75 packet_builder_.SetSsrc(kTestSsrc); | 70 packet_builder_.SetSsrc(kTestSsrc); |
76 packet_builder_.SetReferenceFrameId(kRefFrameId, true); | 71 packet_builder_.SetReferenceFrameId(kRefFrameId, true); |
77 packet_builder_.SetSequenceNumber(kTestSeqNum); | 72 packet_builder_.SetSequenceNumber(kTestSeqNum); |
78 packet_builder_.SetTimestamp(kTestTimestamp); | 73 packet_builder_.SetTimestamp(kTestTimestamp); |
79 packet_builder_.SetPayloadType(kTestPayloadType); | 74 packet_builder_.SetPayloadType(kTestPayloadType); |
80 packet_builder_.SetMarkerBit(true); // Only one packet. | 75 packet_builder_.SetMarkerBit(true); // Only one packet. |
81 } | 76 } |
82 | 77 |
| 78 virtual ~RtpParserTest() {} |
| 79 |
83 void PopulateConfig() { | 80 void PopulateConfig() { |
84 config_.payload_type = kTestPayloadType; | 81 config_.payload_type = kTestPayloadType; |
85 config_.ssrc = kTestSsrc; | 82 config_.ssrc = kTestSsrc; |
86 } | 83 } |
87 | 84 |
88 scoped_ptr<RtpDataTest> rtp_data_; | 85 scoped_ptr<RtpDataTest> rtp_data_; |
89 RtpPacketBuilder packet_builder_; | 86 RtpPacketBuilder packet_builder_; |
90 scoped_ptr<RtpParser> rtp_parser_; | 87 scoped_ptr<RtpParser> rtp_parser_; |
91 RtpParserConfig config_; | 88 RtpParserConfig config_; |
92 RtpCastHeader cast_header_; | 89 RtpCastHeader cast_header_; |
(...skipping 96 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
189 uint8 packet[kPacketLength]; | 186 uint8 packet[kPacketLength]; |
190 packet_builder_.BuildHeader(packet, kPacketLength); | 187 packet_builder_.BuildHeader(packet, kPacketLength); |
191 // Parse packet as is. | 188 // Parse packet as is. |
192 RtpCastHeader rtp_header; | 189 RtpCastHeader rtp_header; |
193 rtp_data_->SetExpectedHeader(cast_header_); | 190 rtp_data_->SetExpectedHeader(cast_header_); |
194 EXPECT_TRUE(rtp_parser_->ParsePacket(packet, kPacketLength, &rtp_header)); | 191 EXPECT_TRUE(rtp_parser_->ParsePacket(packet, kPacketLength, &rtp_header)); |
195 } | 192 } |
196 | 193 |
197 } // namespace cast | 194 } // namespace cast |
198 } // namespace media | 195 } // namespace media |
OLD | NEW |