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 "media/cast/net/udp_transport.h" | 5 #include "media/cast/net/udp_transport.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 #include <string> | 8 #include <string> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
11 #include "base/bind.h" | 11 #include "base/bind.h" |
12 #include "base/callback.h" | 12 #include "base/callback.h" |
13 #include "base/message_loop/message_loop.h" | 13 #include "base/message_loop/message_loop.h" |
14 #include "base/run_loop.h" | 14 #include "base/run_loop.h" |
15 #include "media/cast/net/cast_transport_config.h" | 15 #include "media/cast/net/cast_transport_config.h" |
16 #include "media/cast/test/utility/net_utility.h" | 16 #include "media/cast/test/utility/net_utility.h" |
17 #include "net/base/net_util.h" | 17 #include "net/base/net_util.h" |
18 #include "testing/gtest/include/gtest/gtest.h" | 18 #include "testing/gtest/include/gtest/gtest.h" |
19 | 19 |
20 namespace media { | 20 namespace media { |
21 namespace cast { | 21 namespace cast { |
22 | 22 |
23 class MockPacketReceiver { | 23 class MockPacketReceiver { |
24 public: | 24 public: |
25 MockPacketReceiver(const base::Closure& callback) | 25 MockPacketReceiver(const base::Closure& callback) |
26 : packet_callback_(callback) {} | 26 : packet_callback_(callback) {} |
27 | 27 |
28 void ReceivedPacket(scoped_ptr<Packet> packet) { | 28 bool ReceivedPacket(scoped_ptr<Packet> packet) { |
29 packet_ = std::string(packet->size(), '\0'); | 29 packet_ = std::string(packet->size(), '\0'); |
30 std::copy(packet->begin(), packet->end(), packet_.begin()); | 30 std::copy(packet->begin(), packet->end(), packet_.begin()); |
31 packet_callback_.Run(); | 31 packet_callback_.Run(); |
| 32 return true; |
32 } | 33 } |
33 | 34 |
34 std::string packet() const { return packet_; } | 35 std::string packet() const { return packet_; } |
35 PacketReceiverCallback packet_receiver() { | 36 PacketReceiverCallbackWithStatus packet_receiver() { |
36 return base::Bind(&MockPacketReceiver::ReceivedPacket, | 37 return base::Bind(&MockPacketReceiver::ReceivedPacket, |
37 base::Unretained(this)); | 38 base::Unretained(this)); |
38 } | 39 } |
39 | 40 |
40 private: | 41 private: |
41 std::string packet_; | 42 std::string packet_; |
42 base::Closure packet_callback_; | 43 base::Closure packet_callback_; |
43 | 44 |
44 DISALLOW_COPY_AND_ASSIGN(MockPacketReceiver); | 45 DISALLOW_COPY_AND_ASSIGN(MockPacketReceiver); |
45 }; | 46 }; |
(...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
91 send_transport.SendPacket(new base::RefCountedData<Packet>(packet), cb); | 92 send_transport.SendPacket(new base::RefCountedData<Packet>(packet), cb); |
92 run_loop.Run(); | 93 run_loop.Run(); |
93 EXPECT_TRUE( | 94 EXPECT_TRUE( |
94 std::equal(packet.begin(), packet.end(), receiver1.packet().begin())); | 95 std::equal(packet.begin(), packet.end(), receiver1.packet().begin())); |
95 EXPECT_TRUE( | 96 EXPECT_TRUE( |
96 std::equal(packet.begin(), packet.end(), receiver2.packet().begin())); | 97 std::equal(packet.begin(), packet.end(), receiver2.packet().begin())); |
97 } | 98 } |
98 | 99 |
99 } // namespace cast | 100 } // namespace cast |
100 } // namespace media | 101 } // namespace media |
OLD | NEW |