| 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 |