Home | History | Annotate | Download | only in dns 
      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 
     13 namespace net {
     14 
     15 class MockMDnsDatagramServerSocket : public DatagramServerSocket {
     16  public:
     17   MockMDnsDatagramServerSocket();
     18   ~MockMDnsDatagramServerSocket();
     19 
     20   // DatagramServerSocket implementation:
     21   int Listen(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   int SendTo(IOBuffer* buf, int buf_len, const IPEndPoint& address,
     30              const CompletionCallback& callback);
     31 
     32   MOCK_METHOD3(SendToInternal, int(const std::string& packet,
     33                                    const std::string address,
     34                                    const CompletionCallback& callback));
     35 
     36   MOCK_METHOD1(SetReceiveBufferSize, bool(int32 size));
     37   MOCK_METHOD1(SetSendBufferSize, bool(int32 size));
     38 
     39   MOCK_METHOD0(Close, void());
     40 
     41   MOCK_CONST_METHOD1(GetPeerAddress, int(IPEndPoint* address));
     42   MOCK_CONST_METHOD1(GetLocalAddress, int(IPEndPoint* address));
     43   MOCK_CONST_METHOD0(NetLog, const BoundNetLog&());
     44 
     45   MOCK_METHOD0(AllowAddressReuse, void());
     46   MOCK_METHOD0(AllowBroadcast, void());
     47 
     48   int JoinGroup(const IPAddressNumber& group_address) const;
     49 
     50   MOCK_CONST_METHOD1(JoinGroupInternal, int(const std::string& group));
     51 
     52   int LeaveGroup(const IPAddressNumber& group_address) const;
     53 
     54   MOCK_CONST_METHOD1(LeaveGroupInternal, int(const std::string& group));
     55 
     56   MOCK_METHOD1(SetMulticastTimeToLive, int(int ttl));
     57 
     58   MOCK_METHOD1(SetMulticastLoopbackMode, int(bool loopback));
     59 
     60   void SetResponsePacket(std::string response_packet);
     61 
     62   int HandleRecvNow(IOBuffer* buffer, int size, IPEndPoint* address,
     63                     const CompletionCallback& callback);
     64 
     65   int HandleRecvLater(IOBuffer* buffer, int size, IPEndPoint* address,
     66                       const CompletionCallback& callback);
     67 
     68  private:
     69   std::string response_packet_;
     70 };
     71 
     72 class MockMDnsSocketFactory : public MDnsConnection::SocketFactory {
     73  public:
     74   MockMDnsSocketFactory();
     75 
     76   virtual ~MockMDnsSocketFactory();
     77 
     78   virtual scoped_ptr<DatagramServerSocket> CreateSocket() 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   scoped_refptr<IOBuffer> recv_buffer_;
     95   int recv_buffer_size_;
     96   CompletionCallback recv_callback_;
     97 };
     98 
     99 }  // namespace net
    100 
    101 #endif  // NET_DNS_MOCK_MDNS_SOCKET_FACTORY_H_
    102