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