1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "media/cast/net/udp_transport.h"
#include <algorithm>
#include <string>
#include <vector>
#include "base/bind.h"
#include "base/callback.h"
#include "base/macros.h"
#include "base/message_loop/message_loop.h"
#include "base/run_loop.h"
#include "media/cast/net/cast_transport_config.h"
#include "media/cast/test/utility/net_utility.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace media {
namespace cast {
class MockPacketReceiver {
public:
MockPacketReceiver(const base::Closure& callback)
: packet_callback_(callback) {}
bool ReceivedPacket(scoped_ptr<Packet> packet) {
packet_ = std::string(packet->size(), '\0');
std::copy(packet->begin(), packet->end(), packet_.begin());
packet_callback_.Run();
return true;
}
std::string packet() const { return packet_; }
PacketReceiverCallbackWithStatus packet_receiver() {
return base::Bind(&MockPacketReceiver::ReceivedPacket,
base::Unretained(this));
}
private:
std::string packet_;
base::Closure packet_callback_;
DISALLOW_COPY_AND_ASSIGN(MockPacketReceiver);
};
void SendPacket(UdpTransport* transport, Packet packet) {
base::Closure cb;
transport->SendPacket(new base::RefCountedData<Packet>(packet), cb);
}
static void UpdateCastTransportStatus(CastTransportStatus status) {
NOTREACHED();
}
TEST(UdpTransport, SendAndReceive) {
base::MessageLoopForIO message_loop;
net::IPEndPoint free_local_port1 = test::GetFreeLocalPort();
net::IPEndPoint free_local_port2 = test::GetFreeLocalPort();
net::IPAddressNumber empty_addr_number;
net::ParseIPLiteralToNumber("0.0.0.0", &empty_addr_number);
UdpTransport send_transport(NULL,
message_loop.task_runner(),
free_local_port1,
free_local_port2,
base::Bind(&UpdateCastTransportStatus));
send_transport.SetSendBufferSize(65536);
UdpTransport recv_transport(NULL,
message_loop.task_runner(),
free_local_port2,
net::IPEndPoint(empty_addr_number, 0),
base::Bind(&UpdateCastTransportStatus));
recv_transport.SetSendBufferSize(65536);
Packet packet;
packet.push_back('t');
packet.push_back('e');
packet.push_back('s');
packet.push_back('t');
base::RunLoop run_loop;
MockPacketReceiver receiver1(run_loop.QuitClosure());
MockPacketReceiver receiver2(
base::Bind(&SendPacket, &recv_transport, packet));
send_transport.StartReceiving(receiver1.packet_receiver());
recv_transport.StartReceiving(receiver2.packet_receiver());
base::Closure cb;
send_transport.SendPacket(new base::RefCountedData<Packet>(packet), cb);
run_loop.Run();
EXPECT_TRUE(
std::equal(packet.begin(), packet.end(), receiver1.packet().begin()));
EXPECT_TRUE(
std::equal(packet.begin(), packet.end(), receiver2.packet().begin()));
}
} // namespace cast
} // namespace media
|
