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 |