OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 <stdint.h> | 5 #include <stdint.h> |
6 | 6 |
7 #include "base/big_endian.h" | 7 #include "base/big_endian.h" |
8 #include "base/test/simple_test_tick_clock.h" | 8 #include "base/test/simple_test_tick_clock.h" |
9 #include "media/cast/net/pacing/paced_sender.h" | 9 #include "media/cast/net/pacing/paced_sender.h" |
10 #include "media/cast/test/fake_single_thread_task_runner.h" | 10 #include "media/cast/test/fake_single_thread_task_runner.h" |
(...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
50 | 50 |
51 public: | 51 public: |
52 std::list<int> expected_packet_size_; | 52 std::list<int> expected_packet_size_; |
53 int64 bytes_sent_; | 53 int64 bytes_sent_; |
54 | 54 |
55 DISALLOW_COPY_AND_ASSIGN(TestPacketSender); | 55 DISALLOW_COPY_AND_ASSIGN(TestPacketSender); |
56 }; | 56 }; |
57 | 57 |
58 class PacedSenderTest : public ::testing::Test { | 58 class PacedSenderTest : public ::testing::Test { |
59 protected: | 59 protected: |
60 PacedSenderTest() { | 60 PacedSenderTest() : frame_id_(0) { |
61 testing_clock_.Advance( | 61 testing_clock_.Advance( |
62 base::TimeDelta::FromMilliseconds(kStartMillisecond)); | 62 base::TimeDelta::FromMilliseconds(kStartMillisecond)); |
63 task_runner_ = new test::FakeSingleThreadTaskRunner(&testing_clock_); | 63 task_runner_ = new test::FakeSingleThreadTaskRunner(&testing_clock_); |
64 paced_sender_.reset(new PacedSender(kTargetBurstSize, kMaxBurstSize, | 64 paced_sender_.reset(new PacedSender(kTargetBurstSize, kMaxBurstSize, |
65 &testing_clock_, &packet_events_, | 65 &testing_clock_, &packet_events_, |
66 &mock_transport_, task_runner_)); | 66 &mock_transport_, task_runner_)); |
67 paced_sender_->RegisterAudioSsrc(kAudioSsrc); | 67 paced_sender_->RegisterAudioSsrc(kAudioSsrc); |
68 paced_sender_->RegisterVideoSsrc(kVideoSsrc); | 68 paced_sender_->RegisterVideoSsrc(kVideoSsrc); |
69 } | 69 } |
70 | 70 |
71 static void UpdateCastTransportStatus(CastTransportStatus status) { | 71 static void UpdateCastTransportStatus(CastTransportStatus status) { |
72 NOTREACHED(); | 72 NOTREACHED(); |
73 } | 73 } |
74 | 74 |
75 SendPacketVector CreateSendPacketVector(size_t packet_size, | 75 SendPacketVector CreateSendPacketVector(size_t packet_size, |
76 int num_of_packets_in_frame, | 76 int num_of_packets_in_frame, |
77 bool audio) { | 77 bool audio) { |
78 DCHECK_GE(packet_size, 12u); | 78 DCHECK_GE(packet_size, 12u); |
79 SendPacketVector packets; | 79 SendPacketVector packets; |
80 base::TimeTicks frame_tick = testing_clock_.NowTicks(); | |
81 // Advance the clock so that we don't get the same frame_tick | |
82 // next time this function is called. | |
83 testing_clock_.Advance(base::TimeDelta::FromMilliseconds(1)); | |
84 for (int i = 0; i < num_of_packets_in_frame; ++i) { | 80 for (int i = 0; i < num_of_packets_in_frame; ++i) { |
85 PacketKey key = PacedPacketSender::MakePacketKey( | 81 PacketKey key = PacedPacketSender::MakePacketKey( |
86 frame_tick, | 82 PacketKey::RTP, frame_id_, |
87 audio ? kAudioSsrc : kVideoSsrc, // ssrc | 83 audio ? kAudioSsrc : kVideoSsrc, // ssrc |
88 i); | 84 i); |
89 | 85 |
90 PacketRef packet(new base::RefCountedData<Packet>); | 86 PacketRef packet(new base::RefCountedData<Packet>); |
91 packet->data.resize(packet_size, kValue); | 87 packet->data.resize(packet_size, kValue); |
92 // Fill-in packet header fields to test the header parsing (for populating | 88 // Fill-in packet header fields to test the header parsing (for populating |
93 // the logging events). | 89 // the logging events). |
94 base::BigEndianWriter writer(reinterpret_cast<char*>(&packet->data[0]), | 90 base::BigEndianWriter writer(reinterpret_cast<char*>(&packet->data[0]), |
95 packet_size); | 91 packet_size); |
96 bool success = writer.Skip(4); | 92 bool success = writer.Skip(4); |
97 success &= writer.WriteU32(kFrameRtpTimestamp); | 93 success &= writer.WriteU32(kFrameRtpTimestamp); |
98 success &= writer.WriteU32(audio ? kAudioSsrc : kVideoSsrc); | 94 success &= writer.WriteU32(audio ? kAudioSsrc : kVideoSsrc); |
99 success &= writer.Skip(2); | 95 success &= writer.Skip(2); |
100 success &= writer.WriteU16(i); | 96 success &= writer.WriteU16(i); |
101 success &= writer.WriteU16(num_of_packets_in_frame - 1); | 97 success &= writer.WriteU16(num_of_packets_in_frame - 1); |
102 CHECK(success); | 98 CHECK(success); |
103 packets.push_back(std::make_pair(key, packet)); | 99 packets.push_back(std::make_pair(key, packet)); |
104 } | 100 } |
| 101 // Increase |frame_id_| so that we don't get the same key next time this |
| 102 // function is called. |
| 103 ++frame_id_; |
105 return packets; | 104 return packets; |
106 } | 105 } |
107 | 106 |
108 // Use this function to drain the packet list in PacedSender without having | 107 // Use this function to drain the packet list in PacedSender without having |
109 // to test the pacing implementation details. | 108 // to test the pacing implementation details. |
110 bool RunUntilEmpty(int max_tries) { | 109 bool RunUntilEmpty(int max_tries) { |
111 for (int i = 0; i < max_tries; i++) { | 110 for (int i = 0; i < max_tries; i++) { |
112 testing_clock_.Advance(base::TimeDelta::FromMilliseconds(10)); | 111 testing_clock_.Advance(base::TimeDelta::FromMilliseconds(10)); |
113 task_runner_->RunTasks(); | 112 task_runner_->RunTasks(); |
114 if (mock_transport_.expected_packet_size_.empty()) | 113 if (mock_transport_.expected_packet_size_.empty()) |
115 return true; | 114 return true; |
116 i++; | 115 i++; |
117 } | 116 } |
118 | 117 |
119 return mock_transport_.expected_packet_size_.empty(); | 118 return mock_transport_.expected_packet_size_.empty(); |
120 } | 119 } |
121 | 120 |
122 std::vector<PacketEvent> packet_events_; | 121 std::vector<PacketEvent> packet_events_; |
123 base::SimpleTestTickClock testing_clock_; | 122 base::SimpleTestTickClock testing_clock_; |
124 TestPacketSender mock_transport_; | 123 TestPacketSender mock_transport_; |
125 scoped_refptr<test::FakeSingleThreadTaskRunner> task_runner_; | 124 scoped_refptr<test::FakeSingleThreadTaskRunner> task_runner_; |
126 scoped_ptr<PacedSender> paced_sender_; | 125 scoped_ptr<PacedSender> paced_sender_; |
| 126 uint32 frame_id_; |
127 | 127 |
128 DISALLOW_COPY_AND_ASSIGN(PacedSenderTest); | 128 DISALLOW_COPY_AND_ASSIGN(PacedSenderTest); |
129 }; | 129 }; |
130 | 130 |
131 } // namespace | 131 } // namespace |
132 | 132 |
133 TEST_F(PacedSenderTest, PassThroughRtcp) { | 133 TEST_F(PacedSenderTest, PassThroughRtcp) { |
134 mock_transport_.AddExpectedSize(kSize1, 2); | 134 mock_transport_.AddExpectedSize(kSize1, 2); |
135 SendPacketVector packets = CreateSendPacketVector(kSize1, 1, true); | 135 SendPacketVector packets = CreateSendPacketVector(kSize1, 1, true); |
136 | 136 |
(...skipping 315 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
452 EXPECT_TRUE(paced_sender_->ResendPackets(packets, dedup_info)); | 452 EXPECT_TRUE(paced_sender_->ResendPackets(packets, dedup_info)); |
453 EXPECT_EQ(static_cast<int64>(kSize1), mock_transport_.GetBytesSent()); | 453 EXPECT_EQ(static_cast<int64>(kSize1), mock_transport_.GetBytesSent()); |
454 | 454 |
455 dedup_info.resend_interval = base::TimeDelta::FromMilliseconds(5); | 455 dedup_info.resend_interval = base::TimeDelta::FromMilliseconds(5); |
456 EXPECT_TRUE(paced_sender_->ResendPackets(packets, dedup_info)); | 456 EXPECT_TRUE(paced_sender_->ResendPackets(packets, dedup_info)); |
457 EXPECT_EQ(static_cast<int64>(2 * kSize1), mock_transport_.GetBytesSent()); | 457 EXPECT_EQ(static_cast<int64>(2 * kSize1), mock_transport_.GetBytesSent()); |
458 } | 458 } |
459 | 459 |
460 } // namespace cast | 460 } // namespace cast |
461 } // namespace media | 461 } // namespace media |
OLD | NEW |