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 #ifndef NET_DNS_MOCK_MDNS_SOCKET_FACTORY_H_ | 5 #ifndef NET_DNS_MOCK_MDNS_SOCKET_FACTORY_H_ |
6 #define NET_DNS_MOCK_MDNS_SOCKET_FACTORY_H_ | 6 #define NET_DNS_MOCK_MDNS_SOCKET_FACTORY_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "net/dns/mdns_client_impl.h" | 10 #include "net/dns/mdns_client_impl.h" |
11 #include "testing/gmock/include/gmock/gmock.h" | 11 #include "testing/gmock/include/gmock/gmock.h" |
12 | 12 |
13 namespace net { | 13 namespace net { |
14 | 14 |
15 class MockMDnsDatagramServerSocket : public DatagramServerSocket { | 15 class MockMDnsDatagramServerSocket : public DatagramServerSocket { |
16 public: | 16 public: |
17 explicit MockMDnsDatagramServerSocket(AddressFamily address_family); | 17 explicit MockMDnsDatagramServerSocket(AddressFamily address_family); |
18 ~MockMDnsDatagramServerSocket(); | 18 ~MockMDnsDatagramServerSocket(); |
19 | 19 |
20 // DatagramServerSocket implementation: | 20 // DatagramServerSocket implementation: |
21 MOCK_METHOD1(Listen, int(const IPEndPoint& address)); | 21 MOCK_METHOD1(Listen, int(const IPEndPoint& address)); |
22 | 22 |
23 MOCK_METHOD1(ListenInternal, int(const std::string& address)); | 23 MOCK_METHOD1(ListenInternal, int(const std::string& address)); |
24 | 24 |
25 MOCK_METHOD4(RecvFrom, int(IOBuffer* buffer, int size, | 25 MOCK_METHOD4(RecvFrom, int(IOBuffer* buffer, int size, |
26 IPEndPoint* address, | 26 IPEndPoint* address, |
27 const CompletionCallback& callback)); | 27 const CompletionCallback& callback)); |
28 | 28 |
29 virtual int SendTo(IOBuffer* buf, int buf_len, const IPEndPoint& address, | 29 virtual int SendTo(IOBuffer* buf, int buf_len, const IPEndPoint& address, |
30 const CompletionCallback& callback) OVERRIDE; | 30 const CompletionCallback& callback) override; |
31 | 31 |
32 MOCK_METHOD3(SendToInternal, int(const std::string& packet, | 32 MOCK_METHOD3(SendToInternal, int(const std::string& packet, |
33 const std::string address, | 33 const std::string address, |
34 const CompletionCallback& callback)); | 34 const CompletionCallback& callback)); |
35 | 35 |
36 MOCK_METHOD1(SetReceiveBufferSize, int(int32 size)); | 36 MOCK_METHOD1(SetReceiveBufferSize, int(int32 size)); |
37 MOCK_METHOD1(SetSendBufferSize, int(int32 size)); | 37 MOCK_METHOD1(SetSendBufferSize, int(int32 size)); |
38 | 38 |
39 MOCK_METHOD0(Close, void()); | 39 MOCK_METHOD0(Close, void()); |
40 | 40 |
41 MOCK_CONST_METHOD1(GetPeerAddress, int(IPEndPoint* address)); | 41 MOCK_CONST_METHOD1(GetPeerAddress, int(IPEndPoint* address)); |
42 virtual int GetLocalAddress(IPEndPoint* address) const OVERRIDE; | 42 virtual int GetLocalAddress(IPEndPoint* address) const override; |
43 MOCK_CONST_METHOD0(NetLog, const BoundNetLog&()); | 43 MOCK_CONST_METHOD0(NetLog, const BoundNetLog&()); |
44 | 44 |
45 MOCK_METHOD0(AllowAddressReuse, void()); | 45 MOCK_METHOD0(AllowAddressReuse, void()); |
46 MOCK_METHOD0(AllowBroadcast, void()); | 46 MOCK_METHOD0(AllowBroadcast, void()); |
47 | 47 |
48 MOCK_CONST_METHOD1(JoinGroup, int(const IPAddressNumber& group_address)); | 48 MOCK_CONST_METHOD1(JoinGroup, int(const IPAddressNumber& group_address)); |
49 MOCK_CONST_METHOD1(LeaveGroup, int(const IPAddressNumber& address)); | 49 MOCK_CONST_METHOD1(LeaveGroup, int(const IPAddressNumber& address)); |
50 | 50 |
51 MOCK_METHOD1(SetMulticastInterface, int(uint32 interface_index)); | 51 MOCK_METHOD1(SetMulticastInterface, int(uint32 interface_index)); |
52 MOCK_METHOD1(SetMulticastTimeToLive, int(int ttl)); | 52 MOCK_METHOD1(SetMulticastTimeToLive, int(int ttl)); |
(...skipping 15 matching lines...) Expand all Loading... |
68 std::string response_packet_; | 68 std::string response_packet_; |
69 IPEndPoint local_address_; | 69 IPEndPoint local_address_; |
70 }; | 70 }; |
71 | 71 |
72 class MockMDnsSocketFactory : public MDnsSocketFactory { | 72 class MockMDnsSocketFactory : public MDnsSocketFactory { |
73 public: | 73 public: |
74 MockMDnsSocketFactory(); | 74 MockMDnsSocketFactory(); |
75 virtual ~MockMDnsSocketFactory(); | 75 virtual ~MockMDnsSocketFactory(); |
76 | 76 |
77 virtual void CreateSockets( | 77 virtual void CreateSockets( |
78 ScopedVector<DatagramServerSocket>* sockets) OVERRIDE; | 78 ScopedVector<DatagramServerSocket>* sockets) override; |
79 | 79 |
80 void SimulateReceive(const uint8* packet, int size); | 80 void SimulateReceive(const uint8* packet, int size); |
81 | 81 |
82 MOCK_METHOD1(OnSendTo, void(const std::string&)); | 82 MOCK_METHOD1(OnSendTo, void(const std::string&)); |
83 | 83 |
84 private: | 84 private: |
85 int SendToInternal(const std::string& packet, const std::string& address, | 85 int SendToInternal(const std::string& packet, const std::string& address, |
86 const CompletionCallback& callback); | 86 const CompletionCallback& callback); |
87 | 87 |
88 // The latest receive callback is always saved, since the MDnsConnection | 88 // The latest receive callback is always saved, since the MDnsConnection |
89 // does not care which socket a packet is received on. | 89 // does not care which socket a packet is received on. |
90 int RecvFromInternal(IOBuffer* buffer, int size, | 90 int RecvFromInternal(IOBuffer* buffer, int size, |
91 IPEndPoint* address, | 91 IPEndPoint* address, |
92 const CompletionCallback& callback); | 92 const CompletionCallback& callback); |
93 | 93 |
94 void CreateSocket(AddressFamily address_family, | 94 void CreateSocket(AddressFamily address_family, |
95 ScopedVector<DatagramServerSocket>* sockets); | 95 ScopedVector<DatagramServerSocket>* sockets); |
96 | 96 |
97 scoped_refptr<IOBuffer> recv_buffer_; | 97 scoped_refptr<IOBuffer> recv_buffer_; |
98 int recv_buffer_size_; | 98 int recv_buffer_size_; |
99 CompletionCallback recv_callback_; | 99 CompletionCallback recv_callback_; |
100 }; | 100 }; |
101 | 101 |
102 } // namespace net | 102 } // namespace net |
103 | 103 |
104 #endif // NET_DNS_MOCK_MDNS_SOCKET_FACTORY_H_ | 104 #endif // NET_DNS_MOCK_MDNS_SOCKET_FACTORY_H_ |
OLD | NEW |