| 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 #include <vector> |
| 9 | 10 |
| 11 #include "base/memory/scoped_ptr.h" |
| 10 #include "net/dns/mdns_client_impl.h" | 12 #include "net/dns/mdns_client_impl.h" |
| 11 #include "testing/gmock/include/gmock/gmock.h" | 13 #include "testing/gmock/include/gmock/gmock.h" |
| 12 | 14 |
| 13 namespace net { | 15 namespace net { |
| 14 | 16 |
| 15 class MockMDnsDatagramServerSocket : public DatagramServerSocket { | 17 class MockMDnsDatagramServerSocket : public DatagramServerSocket { |
| 16 public: | 18 public: |
| 17 explicit MockMDnsDatagramServerSocket(AddressFamily address_family); | 19 explicit MockMDnsDatagramServerSocket(AddressFamily address_family); |
| 18 ~MockMDnsDatagramServerSocket() override; | 20 ~MockMDnsDatagramServerSocket() override; |
| 19 | 21 |
| (...skipping 49 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 69 private: | 71 private: |
| 70 std::string response_packet_; | 72 std::string response_packet_; |
| 71 IPEndPoint local_address_; | 73 IPEndPoint local_address_; |
| 72 }; | 74 }; |
| 73 | 75 |
| 74 class MockMDnsSocketFactory : public MDnsSocketFactory { | 76 class MockMDnsSocketFactory : public MDnsSocketFactory { |
| 75 public: | 77 public: |
| 76 MockMDnsSocketFactory(); | 78 MockMDnsSocketFactory(); |
| 77 ~MockMDnsSocketFactory() override; | 79 ~MockMDnsSocketFactory() override; |
| 78 | 80 |
| 79 void CreateSockets(ScopedVector<DatagramServerSocket>* sockets) override; | 81 void CreateSockets( |
| 82 std::vector<scoped_ptr<DatagramServerSocket>>* sockets) override; |
| 80 | 83 |
| 81 void SimulateReceive(const uint8* packet, int size); | 84 void SimulateReceive(const uint8* packet, int size); |
| 82 | 85 |
| 83 MOCK_METHOD1(OnSendTo, void(const std::string&)); | 86 MOCK_METHOD1(OnSendTo, void(const std::string&)); |
| 84 | 87 |
| 85 private: | 88 private: |
| 86 int SendToInternal(const std::string& packet, const std::string& address, | 89 int SendToInternal(const std::string& packet, const std::string& address, |
| 87 const CompletionCallback& callback); | 90 const CompletionCallback& callback); |
| 88 | 91 |
| 89 // The latest receive callback is always saved, since the MDnsConnection | 92 // The latest receive callback is always saved, since the MDnsConnection |
| 90 // does not care which socket a packet is received on. | 93 // does not care which socket a packet is received on. |
| 91 int RecvFromInternal(IOBuffer* buffer, int size, | 94 int RecvFromInternal(IOBuffer* buffer, int size, |
| 92 IPEndPoint* address, | 95 IPEndPoint* address, |
| 93 const CompletionCallback& callback); | 96 const CompletionCallback& callback); |
| 94 | 97 |
| 95 void CreateSocket(AddressFamily address_family, | 98 void CreateSocket(AddressFamily address_family, |
| 96 ScopedVector<DatagramServerSocket>* sockets); | 99 std::vector<scoped_ptr<DatagramServerSocket>>* sockets); |
| 97 | 100 |
| 98 scoped_refptr<IOBuffer> recv_buffer_; | 101 scoped_refptr<IOBuffer> recv_buffer_; |
| 99 int recv_buffer_size_; | 102 int recv_buffer_size_; |
| 100 CompletionCallback recv_callback_; | 103 CompletionCallback recv_callback_; |
| 101 }; | 104 }; |
| 102 | 105 |
| 103 } // namespace net | 106 } // namespace net |
| 104 | 107 |
| 105 #endif // NET_DNS_MOCK_MDNS_SOCKET_FACTORY_H_ | 108 #endif // NET_DNS_MOCK_MDNS_SOCKET_FACTORY_H_ |
| OLD | NEW |