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 <gtest/gtest.h> | 5 #include <gtest/gtest.h> |
6 #include <stdint.h> | 6 #include <stdint.h> |
7 | 7 |
8 #include "base/bind.h" | 8 #include "base/bind.h" |
9 #include "base/bind_helpers.h" | 9 #include "base/bind_helpers.h" |
10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
11 #include "base/test/simple_test_tick_clock.h" | 11 #include "base/test/simple_test_tick_clock.h" |
12 #include "base/values.h" | 12 #include "base/values.h" |
13 #include "media/cast/net/cast_transport_config.h" | 13 #include "media/cast/net/cast_transport_config.h" |
14 #include "media/cast/net/cast_transport_sender_impl.h" | 14 #include "media/cast/net/cast_transport_sender_impl.h" |
15 #include "media/cast/net/rtcp/rtcp.h" | 15 #include "media/cast/net/rtcp/rtcp.h" |
16 #include "media/cast/test/fake_single_thread_task_runner.h" | 16 #include "media/cast/test/fake_single_thread_task_runner.h" |
17 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
18 | 18 |
19 namespace media { | 19 namespace media { |
20 namespace cast { | 20 namespace cast { |
21 | 21 |
22 namespace { | 22 namespace { |
23 const int64 kStartMillisecond = INT64_C(12345678900000); | 23 const int64_t kStartMillisecond = INT64_C(12345678900000); |
24 const uint32 kVideoSsrc = 1; | 24 const uint32_t kVideoSsrc = 1; |
25 const uint32 kAudioSsrc = 2; | 25 const uint32_t kAudioSsrc = 2; |
26 } // namespace | 26 } // namespace |
27 | 27 |
28 class FakePacketSender : public PacketSender { | 28 class FakePacketSender : public PacketSender { |
29 public: | 29 public: |
30 FakePacketSender() | 30 FakePacketSender() |
31 : paused_(false), packets_sent_(0), bytes_sent_(0) {} | 31 : paused_(false), packets_sent_(0), bytes_sent_(0) {} |
32 | 32 |
33 bool SendPacket(PacketRef packet, const base::Closure& cb) final { | 33 bool SendPacket(PacketRef packet, const base::Closure& cb) final { |
34 if (paused_) { | 34 if (paused_) { |
35 stored_packet_ = packet; | 35 stored_packet_ = packet; |
36 callback_ = cb; | 36 callback_ = cb; |
37 return false; | 37 return false; |
38 } | 38 } |
39 ++packets_sent_; | 39 ++packets_sent_; |
40 bytes_sent_ += packet->data.size(); | 40 bytes_sent_ += packet->data.size(); |
41 return true; | 41 return true; |
42 } | 42 } |
43 | 43 |
44 int64 GetBytesSent() final { return bytes_sent_; } | 44 int64_t GetBytesSent() final { return bytes_sent_; } |
45 | 45 |
46 void SetPaused(bool paused) { | 46 void SetPaused(bool paused) { |
47 paused_ = paused; | 47 paused_ = paused; |
48 if (!paused && stored_packet_.get()) { | 48 if (!paused && stored_packet_.get()) { |
49 SendPacket(stored_packet_, callback_); | 49 SendPacket(stored_packet_, callback_); |
50 callback_.Run(); | 50 callback_.Run(); |
51 } | 51 } |
52 } | 52 } |
53 | 53 |
54 int packets_sent() const { return packets_sent_; } | 54 int packets_sent() const { return packets_sent_; } |
55 | 55 |
56 private: | 56 private: |
57 bool paused_; | 57 bool paused_; |
58 base::Closure callback_; | 58 base::Closure callback_; |
59 PacketRef stored_packet_; | 59 PacketRef stored_packet_; |
60 int packets_sent_; | 60 int packets_sent_; |
61 int64 bytes_sent_; | 61 int64_t bytes_sent_; |
62 | 62 |
63 DISALLOW_COPY_AND_ASSIGN(FakePacketSender); | 63 DISALLOW_COPY_AND_ASSIGN(FakePacketSender); |
64 }; | 64 }; |
65 | 65 |
66 class CastTransportSenderImplTest : public ::testing::Test { | 66 class CastTransportSenderImplTest : public ::testing::Test { |
67 protected: | 67 protected: |
68 CastTransportSenderImplTest() : num_times_logging_callback_called_(0) { | 68 CastTransportSenderImplTest() : num_times_logging_callback_called_(0) { |
69 testing_clock_.Advance( | 69 testing_clock_.Advance( |
70 base::TimeDelta::FromMilliseconds(kStartMillisecond)); | 70 base::TimeDelta::FromMilliseconds(kStartMillisecond)); |
71 task_runner_ = new test::FakeSingleThreadTaskRunner(&testing_clock_); | 71 task_runner_ = new test::FakeSingleThreadTaskRunner(&testing_clock_); |
(...skipping 177 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
249 | 249 |
250 transport_.SetPaused(true); | 250 transport_.SetPaused(true); |
251 DedupInfo dedup_info; | 251 DedupInfo dedup_info; |
252 dedup_info.resend_interval = base::TimeDelta::FromMilliseconds(10); | 252 dedup_info.resend_interval = base::TimeDelta::FromMilliseconds(10); |
253 transport_sender_->ResendPackets( | 253 transport_sender_->ResendPackets( |
254 kVideoSsrc, missing_packets, true, dedup_info); | 254 kVideoSsrc, missing_packets, true, dedup_info); |
255 | 255 |
256 task_runner_->Sleep(base::TimeDelta::FromMilliseconds(10)); | 256 task_runner_->Sleep(base::TimeDelta::FromMilliseconds(10)); |
257 EXPECT_EQ(2, num_times_logging_callback_called_); | 257 EXPECT_EQ(2, num_times_logging_callback_called_); |
258 | 258 |
259 std::vector<uint32> cancel_sending_frames; | 259 std::vector<uint32_t> cancel_sending_frames; |
260 cancel_sending_frames.push_back(1); | 260 cancel_sending_frames.push_back(1); |
261 transport_sender_->CancelSendingFrames(kVideoSsrc, | 261 transport_sender_->CancelSendingFrames(kVideoSsrc, |
262 cancel_sending_frames); | 262 cancel_sending_frames); |
263 transport_.SetPaused(false); | 263 transport_.SetPaused(false); |
264 task_runner_->Sleep(base::TimeDelta::FromMilliseconds(10)); | 264 task_runner_->Sleep(base::TimeDelta::FromMilliseconds(10)); |
265 EXPECT_EQ(2, num_times_logging_callback_called_); | 265 EXPECT_EQ(2, num_times_logging_callback_called_); |
266 | 266 |
267 // Resend one packet in the socket when unpaused. | 267 // Resend one packet in the socket when unpaused. |
268 EXPECT_EQ(5, transport_.packets_sent()); | 268 EXPECT_EQ(5, transport_.packets_sent()); |
269 } | 269 } |
(...skipping 116 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
386 task_runner_->RunTasks(); | 386 task_runner_->RunTasks(); |
387 EXPECT_EQ(7, transport_.packets_sent()); | 387 EXPECT_EQ(7, transport_.packets_sent()); |
388 EXPECT_EQ(1, num_times_logging_callback_called_); // Only 8 ms since last. | 388 EXPECT_EQ(1, num_times_logging_callback_called_); // Only 8 ms since last. |
389 | 389 |
390 task_runner_->Sleep(base::TimeDelta::FromMilliseconds(2)); | 390 task_runner_->Sleep(base::TimeDelta::FromMilliseconds(2)); |
391 EXPECT_EQ(2, num_times_logging_callback_called_); | 391 EXPECT_EQ(2, num_times_logging_callback_called_); |
392 } | 392 } |
393 | 393 |
394 } // namespace cast | 394 } // namespace cast |
395 } // namespace media | 395 } // namespace media |
OLD | NEW |