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
3 // found in the LICENSE file.
5 #include "media/cast/transport/rtp_sender/rtp_packetizer/rtp_packetizer.h"
9 #include "base/memory/scoped_ptr.h"
10 #include "base/test/simple_test_tick_clock.h"
11 #include "media/cast/logging/simple_event_subscriber.h"
12 #include "media/cast/test/fake_single_thread_task_runner.h"
13 #include "media/cast/transport/pacing/paced_sender.h"
14 #include "media/cast/transport/rtp_sender/packet_storage/packet_storage.h"
15 #include "media/cast/transport/rtp_sender/rtp_packetizer/test/rtp_header_parser.h"
16 #include "testing/gmock/include/gmock/gmock.h"
23 static const int kPayload = 127;
24 static const uint32 kTimestampMs = 10;
25 static const uint16 kSeqNum = 33;
26 static const int kMaxPacketLength = 1500;
27 static const int kSsrc = 0x12345;
28 static const unsigned int kFrameSize = 5000;
29 static const uint32 kStartFrameId = UINT32_C(0xffffffff);
32 class TestRtpPacketTransport : public PacketSender {
34 explicit TestRtpPacketTransport(RtpPacketizerConfig config)
36 sequence_number_(kSeqNum),
38 expected_number_of_packets_(0),
39 expected_packet_id_(0),
40 expected_frame_id_(0),
41 expectd_rtp_timestamp_(0) {}
43 void VerifyRtpHeader(const RtpCastTestHeader& rtp_header) {
44 VerifyCommonRtpHeader(rtp_header);
45 VerifyCastRtpHeader(rtp_header);
48 void VerifyCommonRtpHeader(const RtpCastTestHeader& rtp_header) {
49 EXPECT_EQ(kPayload, rtp_header.payload_type);
50 EXPECT_EQ(sequence_number_, rtp_header.sequence_number);
51 EXPECT_EQ(expectd_rtp_timestamp_, rtp_header.rtp_timestamp);
52 EXPECT_EQ(config_.ssrc, rtp_header.ssrc);
53 EXPECT_EQ(0, rtp_header.num_csrcs);
56 void VerifyCastRtpHeader(const RtpCastTestHeader& rtp_header) {
57 EXPECT_FALSE(rtp_header.is_key_frame);
58 EXPECT_EQ(expected_frame_id_, rtp_header.frame_id);
59 EXPECT_EQ(expected_packet_id_, rtp_header.packet_id);
60 EXPECT_EQ(expected_number_of_packets_ - 1, rtp_header.max_packet_id);
61 EXPECT_TRUE(rtp_header.is_reference);
62 EXPECT_EQ(expected_frame_id_ - 1u, rtp_header.reference_frame_id);
65 virtual bool SendPacket(PacketRef packet, const base::Closure& cb) OVERRIDE {
67 RtpHeaderParser parser(packet->data.data(), packet->data.size());
68 RtpCastTestHeader rtp_header;
69 parser.Parse(&rtp_header);
70 VerifyRtpHeader(rtp_header);
72 ++expected_packet_id_;
76 int number_of_packets_received() const { return packets_sent_; }
78 void set_expected_number_of_packets(int expected_number_of_packets) {
79 expected_number_of_packets_ = expected_number_of_packets;
82 void set_rtp_timestamp(uint32 rtp_timestamp) {
83 expectd_rtp_timestamp_ = rtp_timestamp;
86 RtpPacketizerConfig config_;
87 uint32 sequence_number_;
89 int number_of_packets_;
90 int expected_number_of_packets_;
91 // Assuming packets arrive in sequence.
92 int expected_packet_id_;
93 uint32 expected_frame_id_;
94 uint32 expectd_rtp_timestamp_;
96 DISALLOW_COPY_AND_ASSIGN(TestRtpPacketTransport);
99 class RtpPacketizerTest : public ::testing::Test {
102 : task_runner_(new test::FakeSingleThreadTaskRunner(&testing_clock_)),
104 packet_storage_(200) {
105 config_.sequence_number = kSeqNum;
106 config_.ssrc = kSsrc;
107 config_.payload_type = kPayload;
108 config_.max_payload_length = kMaxPacketLength;
109 transport_.reset(new TestRtpPacketTransport(config_));
110 pacer_.reset(new PacedSender(
111 &testing_clock_, &logging_, transport_.get(), task_runner_));
112 pacer_->RegisterVideoSsrc(config_.ssrc);
113 rtp_packetizer_.reset(new RtpPacketizer(
114 pacer_.get(), &packet_storage_, config_));
115 video_frame_.key_frame = false;
116 video_frame_.frame_id = 0;
117 video_frame_.last_referenced_frame_id = kStartFrameId;
118 video_frame_.data.assign(kFrameSize, 123);
119 video_frame_.rtp_timestamp =
120 GetVideoRtpTimestamp(testing_clock_.NowTicks());
123 void RunTasks(int during_ms) {
124 for (int i = 0; i < during_ms; ++i) {
125 // Call process the timers every 1 ms.
126 testing_clock_.Advance(base::TimeDelta::FromMilliseconds(1));
127 task_runner_->RunTasks();
131 base::SimpleTestTickClock testing_clock_;
132 scoped_refptr<test::FakeSingleThreadTaskRunner> task_runner_;
133 EncodedVideoFrame video_frame_;
134 PacketStorage packet_storage_;
135 RtpPacketizerConfig config_;
136 scoped_ptr<TestRtpPacketTransport> transport_;
137 LoggingImpl logging_;
138 scoped_ptr<PacedSender> pacer_;
139 scoped_ptr<RtpPacketizer> rtp_packetizer_;
141 DISALLOW_COPY_AND_ASSIGN(RtpPacketizerTest);
144 TEST_F(RtpPacketizerTest, SendStandardPackets) {
145 int expected_num_of_packets = kFrameSize / kMaxPacketLength + 1;
146 transport_->set_expected_number_of_packets(expected_num_of_packets);
147 transport_->set_rtp_timestamp(video_frame_.rtp_timestamp);
149 base::TimeTicks time;
150 time += base::TimeDelta::FromMilliseconds(kTimestampMs);
151 rtp_packetizer_->IncomingEncodedVideoFrame(&video_frame_, time);
153 EXPECT_EQ(expected_num_of_packets, transport_->number_of_packets_received());
156 TEST_F(RtpPacketizerTest, Stats) {
157 EXPECT_FALSE(rtp_packetizer_->send_packets_count());
158 EXPECT_FALSE(rtp_packetizer_->send_octet_count());
159 // Insert packets at varying lengths.
160 int expected_num_of_packets = kFrameSize / kMaxPacketLength + 1;
161 transport_->set_expected_number_of_packets(expected_num_of_packets);
162 transport_->set_rtp_timestamp(video_frame_.rtp_timestamp);
164 testing_clock_.Advance(base::TimeDelta::FromMilliseconds(kTimestampMs));
165 rtp_packetizer_->IncomingEncodedVideoFrame(&video_frame_,
166 testing_clock_.NowTicks());
168 EXPECT_EQ(expected_num_of_packets, rtp_packetizer_->send_packets_count());
169 EXPECT_EQ(kFrameSize, rtp_packetizer_->send_octet_count());
170 EXPECT_EQ(expected_num_of_packets, transport_->number_of_packets_received());
173 } // namespace transport