| 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 <stdint.h> |
| 9 |
| 8 #include <string> | 10 #include <string> |
| 9 #include <vector> | 11 #include <vector> |
| 10 | 12 |
| 11 #include "base/memory/scoped_ptr.h" | 13 #include "base/memory/scoped_ptr.h" |
| 12 #include "net/dns/mdns_client_impl.h" | 14 #include "net/dns/mdns_client_impl.h" |
| 13 #include "testing/gmock/include/gmock/gmock.h" | 15 #include "testing/gmock/include/gmock/gmock.h" |
| 14 | 16 |
| 15 namespace net { | 17 namespace net { |
| 16 | 18 |
| 17 class MockMDnsDatagramServerSocket : public DatagramServerSocket { | 19 class MockMDnsDatagramServerSocket : public DatagramServerSocket { |
| (...skipping 12 matching lines...) Expand all Loading... |
| 30 | 32 |
| 31 int SendTo(IOBuffer* buf, | 33 int SendTo(IOBuffer* buf, |
| 32 int buf_len, | 34 int buf_len, |
| 33 const IPEndPoint& address, | 35 const IPEndPoint& address, |
| 34 const CompletionCallback& callback) override; | 36 const CompletionCallback& callback) override; |
| 35 | 37 |
| 36 MOCK_METHOD3(SendToInternal, int(const std::string& packet, | 38 MOCK_METHOD3(SendToInternal, int(const std::string& packet, |
| 37 const std::string address, | 39 const std::string address, |
| 38 const CompletionCallback& callback)); | 40 const CompletionCallback& callback)); |
| 39 | 41 |
| 40 MOCK_METHOD1(SetReceiveBufferSize, int(int32 size)); | 42 MOCK_METHOD1(SetReceiveBufferSize, int(int32_t size)); |
| 41 MOCK_METHOD1(SetSendBufferSize, int(int32 size)); | 43 MOCK_METHOD1(SetSendBufferSize, int(int32_t size)); |
| 42 | 44 |
| 43 MOCK_METHOD0(Close, void()); | 45 MOCK_METHOD0(Close, void()); |
| 44 | 46 |
| 45 MOCK_CONST_METHOD1(GetPeerAddress, int(IPEndPoint* address)); | 47 MOCK_CONST_METHOD1(GetPeerAddress, int(IPEndPoint* address)); |
| 46 int GetLocalAddress(IPEndPoint* address) const override; | 48 int GetLocalAddress(IPEndPoint* address) const override; |
| 47 MOCK_CONST_METHOD0(NetLog, const BoundNetLog&()); | 49 MOCK_CONST_METHOD0(NetLog, const BoundNetLog&()); |
| 48 | 50 |
| 49 MOCK_METHOD0(AllowAddressReuse, void()); | 51 MOCK_METHOD0(AllowAddressReuse, void()); |
| 50 MOCK_METHOD0(AllowBroadcast, void()); | 52 MOCK_METHOD0(AllowBroadcast, void()); |
| 51 | 53 |
| 52 MOCK_CONST_METHOD1(JoinGroup, int(const IPAddressNumber& group_address)); | 54 MOCK_CONST_METHOD1(JoinGroup, int(const IPAddressNumber& group_address)); |
| 53 MOCK_CONST_METHOD1(LeaveGroup, int(const IPAddressNumber& address)); | 55 MOCK_CONST_METHOD1(LeaveGroup, int(const IPAddressNumber& address)); |
| 54 | 56 |
| 55 MOCK_METHOD1(SetMulticastInterface, int(uint32 interface_index)); | 57 MOCK_METHOD1(SetMulticastInterface, int(uint32_t interface_index)); |
| 56 MOCK_METHOD1(SetMulticastTimeToLive, int(int ttl)); | 58 MOCK_METHOD1(SetMulticastTimeToLive, int(int ttl)); |
| 57 MOCK_METHOD1(SetMulticastLoopbackMode, int(bool loopback)); | 59 MOCK_METHOD1(SetMulticastLoopbackMode, int(bool loopback)); |
| 58 | 60 |
| 59 MOCK_METHOD1(SetDiffServCodePoint, int(DiffServCodePoint dscp)); | 61 MOCK_METHOD1(SetDiffServCodePoint, int(DiffServCodePoint dscp)); |
| 60 | 62 |
| 61 MOCK_METHOD0(DetachFromThread, void()); | 63 MOCK_METHOD0(DetachFromThread, void()); |
| 62 | 64 |
| 63 void SetResponsePacket(const std::string& response_packet); | 65 void SetResponsePacket(const std::string& response_packet); |
| 64 | 66 |
| 65 int HandleRecvNow(IOBuffer* buffer, int size, IPEndPoint* address, | 67 int HandleRecvNow(IOBuffer* buffer, int size, IPEndPoint* address, |
| 66 const CompletionCallback& callback); | 68 const CompletionCallback& callback); |
| 67 | 69 |
| 68 int HandleRecvLater(IOBuffer* buffer, int size, IPEndPoint* address, | 70 int HandleRecvLater(IOBuffer* buffer, int size, IPEndPoint* address, |
| 69 const CompletionCallback& callback); | 71 const CompletionCallback& callback); |
| 70 | 72 |
| 71 private: | 73 private: |
| 72 std::string response_packet_; | 74 std::string response_packet_; |
| 73 IPEndPoint local_address_; | 75 IPEndPoint local_address_; |
| 74 }; | 76 }; |
| 75 | 77 |
| 76 class MockMDnsSocketFactory : public MDnsSocketFactory { | 78 class MockMDnsSocketFactory : public MDnsSocketFactory { |
| 77 public: | 79 public: |
| 78 MockMDnsSocketFactory(); | 80 MockMDnsSocketFactory(); |
| 79 ~MockMDnsSocketFactory() override; | 81 ~MockMDnsSocketFactory() override; |
| 80 | 82 |
| 81 void CreateSockets( | 83 void CreateSockets( |
| 82 std::vector<scoped_ptr<DatagramServerSocket>>* sockets) override; | 84 std::vector<scoped_ptr<DatagramServerSocket>>* sockets) override; |
| 83 | 85 |
| 84 void SimulateReceive(const uint8* packet, int size); | 86 void SimulateReceive(const uint8_t* packet, int size); |
| 85 | 87 |
| 86 MOCK_METHOD1(OnSendTo, void(const std::string&)); | 88 MOCK_METHOD1(OnSendTo, void(const std::string&)); |
| 87 | 89 |
| 88 private: | 90 private: |
| 89 int SendToInternal(const std::string& packet, const std::string& address, | 91 int SendToInternal(const std::string& packet, const std::string& address, |
| 90 const CompletionCallback& callback); | 92 const CompletionCallback& callback); |
| 91 | 93 |
| 92 // The latest receive callback is always saved, since the MDnsConnection | 94 // The latest receive callback is always saved, since the MDnsConnection |
| 93 // does not care which socket a packet is received on. | 95 // does not care which socket a packet is received on. |
| 94 int RecvFromInternal(IOBuffer* buffer, int size, | 96 int RecvFromInternal(IOBuffer* buffer, int size, |
| 95 IPEndPoint* address, | 97 IPEndPoint* address, |
| 96 const CompletionCallback& callback); | 98 const CompletionCallback& callback); |
| 97 | 99 |
| 98 void CreateSocket(AddressFamily address_family, | 100 void CreateSocket(AddressFamily address_family, |
| 99 std::vector<scoped_ptr<DatagramServerSocket>>* sockets); | 101 std::vector<scoped_ptr<DatagramServerSocket>>* sockets); |
| 100 | 102 |
| 101 scoped_refptr<IOBuffer> recv_buffer_; | 103 scoped_refptr<IOBuffer> recv_buffer_; |
| 102 int recv_buffer_size_; | 104 int recv_buffer_size_; |
| 103 CompletionCallback recv_callback_; | 105 CompletionCallback recv_callback_; |
| 104 }; | 106 }; |
| 105 | 107 |
| 106 } // namespace net | 108 } // namespace net |
| 107 | 109 |
| 108 #endif // NET_DNS_MOCK_MDNS_SOCKET_FACTORY_H_ | 110 #endif // NET_DNS_MOCK_MDNS_SOCKET_FACTORY_H_ |
| OLD | NEW |