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 "media/cast/transport/rtp_sender/rtp_packetizer/rtp_packetizer.h" | 5 #include "media/cast/transport/rtp_sender/rtp_packetizer/rtp_packetizer.h" |
6 | 6 |
7 #include <stdint.h> | 7 #include <stdint.h> |
8 | 8 |
9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
10 #include "base/test/simple_test_tick_clock.h" | 10 #include "base/test/simple_test_tick_clock.h" |
(...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
105 config_.sequence_number = kSeqNum; | 105 config_.sequence_number = kSeqNum; |
106 config_.ssrc = kSsrc; | 106 config_.ssrc = kSsrc; |
107 config_.payload_type = kPayload; | 107 config_.payload_type = kPayload; |
108 config_.max_payload_length = kMaxPacketLength; | 108 config_.max_payload_length = kMaxPacketLength; |
109 transport_.reset(new TestRtpPacketTransport(config_)); | 109 transport_.reset(new TestRtpPacketTransport(config_)); |
110 pacer_.reset(new PacedSender( | 110 pacer_.reset(new PacedSender( |
111 &testing_clock_, &logging_, transport_.get(), task_runner_)); | 111 &testing_clock_, &logging_, transport_.get(), task_runner_)); |
112 pacer_->RegisterVideoSsrc(config_.ssrc); | 112 pacer_->RegisterVideoSsrc(config_.ssrc); |
113 rtp_packetizer_.reset(new RtpPacketizer( | 113 rtp_packetizer_.reset(new RtpPacketizer( |
114 pacer_.get(), &packet_storage_, config_)); | 114 pacer_.get(), &packet_storage_, config_)); |
115 video_frame_.key_frame = false; | 115 video_frame_.dependency = EncodedFrame::DEPENDENT; |
116 video_frame_.frame_id = 0; | 116 video_frame_.frame_id = 0; |
117 video_frame_.last_referenced_frame_id = kStartFrameId; | 117 video_frame_.referenced_frame_id = kStartFrameId; |
118 video_frame_.data.assign(kFrameSize, 123); | 118 video_frame_.data.assign(kFrameSize, 123); |
119 video_frame_.rtp_timestamp = | 119 video_frame_.rtp_timestamp = |
120 GetVideoRtpTimestamp(testing_clock_.NowTicks()); | 120 GetVideoRtpTimestamp(testing_clock_.NowTicks()); |
121 } | 121 } |
122 | 122 |
123 void RunTasks(int during_ms) { | 123 void RunTasks(int during_ms) { |
124 for (int i = 0; i < during_ms; ++i) { | 124 for (int i = 0; i < during_ms; ++i) { |
125 // Call process the timers every 1 ms. | 125 // Call process the timers every 1 ms. |
126 testing_clock_.Advance(base::TimeDelta::FromMilliseconds(1)); | 126 testing_clock_.Advance(base::TimeDelta::FromMilliseconds(1)); |
127 task_runner_->RunTasks(); | 127 task_runner_->RunTasks(); |
128 } | 128 } |
129 } | 129 } |
130 | 130 |
131 base::SimpleTestTickClock testing_clock_; | 131 base::SimpleTestTickClock testing_clock_; |
132 scoped_refptr<test::FakeSingleThreadTaskRunner> task_runner_; | 132 scoped_refptr<test::FakeSingleThreadTaskRunner> task_runner_; |
133 EncodedVideoFrame video_frame_; | 133 EncodedFrame video_frame_; |
134 PacketStorage packet_storage_; | 134 PacketStorage packet_storage_; |
135 RtpPacketizerConfig config_; | 135 RtpPacketizerConfig config_; |
136 scoped_ptr<TestRtpPacketTransport> transport_; | 136 scoped_ptr<TestRtpPacketTransport> transport_; |
137 LoggingImpl logging_; | 137 LoggingImpl logging_; |
138 scoped_ptr<PacedSender> pacer_; | 138 scoped_ptr<PacedSender> pacer_; |
139 scoped_ptr<RtpPacketizer> rtp_packetizer_; | 139 scoped_ptr<RtpPacketizer> rtp_packetizer_; |
140 | 140 |
141 DISALLOW_COPY_AND_ASSIGN(RtpPacketizerTest); | 141 DISALLOW_COPY_AND_ASSIGN(RtpPacketizerTest); |
142 }; | 142 }; |
143 | 143 |
144 TEST_F(RtpPacketizerTest, SendStandardPackets) { | 144 TEST_F(RtpPacketizerTest, SendStandardPackets) { |
145 size_t expected_num_of_packets = kFrameSize / kMaxPacketLength + 1; | 145 size_t expected_num_of_packets = kFrameSize / kMaxPacketLength + 1; |
146 transport_->set_expected_number_of_packets(expected_num_of_packets); | 146 transport_->set_expected_number_of_packets(expected_num_of_packets); |
147 transport_->set_rtp_timestamp(video_frame_.rtp_timestamp); | 147 transport_->set_rtp_timestamp(video_frame_.rtp_timestamp); |
148 | 148 |
149 base::TimeTicks time; | 149 testing_clock_.Advance(base::TimeDelta::FromMilliseconds(kTimestampMs)); |
150 time += base::TimeDelta::FromMilliseconds(kTimestampMs); | 150 video_frame_.reference_time = testing_clock_.NowTicks(); |
151 rtp_packetizer_->IncomingEncodedVideoFrame(&video_frame_, time); | 151 rtp_packetizer_->SendFrameAsPackets(video_frame_); |
152 RunTasks(33 + 1); | 152 RunTasks(33 + 1); |
153 EXPECT_EQ(expected_num_of_packets, transport_->number_of_packets_received()); | 153 EXPECT_EQ(expected_num_of_packets, transport_->number_of_packets_received()); |
154 } | 154 } |
155 | 155 |
156 TEST_F(RtpPacketizerTest, Stats) { | 156 TEST_F(RtpPacketizerTest, Stats) { |
157 EXPECT_FALSE(rtp_packetizer_->send_packet_count()); | 157 EXPECT_FALSE(rtp_packetizer_->send_packet_count()); |
158 EXPECT_FALSE(rtp_packetizer_->send_octet_count()); | 158 EXPECT_FALSE(rtp_packetizer_->send_octet_count()); |
159 // Insert packets at varying lengths. | 159 // Insert packets at varying lengths. |
160 size_t expected_num_of_packets = kFrameSize / kMaxPacketLength + 1; | 160 size_t expected_num_of_packets = kFrameSize / kMaxPacketLength + 1; |
161 transport_->set_expected_number_of_packets(expected_num_of_packets); | 161 transport_->set_expected_number_of_packets(expected_num_of_packets); |
162 transport_->set_rtp_timestamp(video_frame_.rtp_timestamp); | 162 transport_->set_rtp_timestamp(video_frame_.rtp_timestamp); |
163 | 163 |
164 testing_clock_.Advance(base::TimeDelta::FromMilliseconds(kTimestampMs)); | 164 testing_clock_.Advance(base::TimeDelta::FromMilliseconds(kTimestampMs)); |
165 rtp_packetizer_->IncomingEncodedVideoFrame(&video_frame_, | 165 video_frame_.reference_time = testing_clock_.NowTicks(); |
166 testing_clock_.NowTicks()); | 166 rtp_packetizer_->SendFrameAsPackets(video_frame_); |
167 RunTasks(33 + 1); | 167 RunTasks(33 + 1); |
168 EXPECT_EQ(expected_num_of_packets, rtp_packetizer_->send_packet_count()); | 168 EXPECT_EQ(expected_num_of_packets, rtp_packetizer_->send_packet_count()); |
169 EXPECT_EQ(kFrameSize, rtp_packetizer_->send_octet_count()); | 169 EXPECT_EQ(kFrameSize, rtp_packetizer_->send_octet_count()); |
170 EXPECT_EQ(expected_num_of_packets, transport_->number_of_packets_received()); | 170 EXPECT_EQ(expected_num_of_packets, transport_->number_of_packets_received()); |
171 } | 171 } |
172 | 172 |
173 } // namespace transport | 173 } // namespace transport |
174 } // namespace cast | 174 } // namespace cast |
175 } // namespace media | 175 } // namespace media |
OLD | NEW |