Refactored to make MDnsSocketFactory return array of sockets.

net/dns/mock_mdns_socket_factory.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <algorithm>
 
 #include "net/base/net_errors.h"
 #include "net/dns/mock_mdns_socket_factory.h"
 
 using testing::_;
 using testing::Invoke;
 
 namespace net {
 
 MockMDnsDatagramServerSocket::MockMDnsDatagramServerSocket() {
 }
 
 MockMDnsDatagramServerSocket::~MockMDnsDatagramServerSocket() {
 }
 
 int MockMDnsDatagramServerSocket::SendTo(IOBuffer* buf, int buf_len,
-                                     const IPEndPoint& address,
-                                     const CompletionCallback& callback) {
-    return SendToInternal(std::string(buf->data(), buf_len), address.ToString(),
-                          callback);
-}
-
-int MockMDnsDatagramServerSocket::Listen(const IPEndPoint& address) {
-    return ListenInternal(address.ToString());
-}
-
-int MockMDnsDatagramServerSocket::JoinGroup(
-    const IPAddressNumber& group_address) const {
-  return JoinGroupInternal(IPAddressToString(group_address));
-}
-
-int MockMDnsDatagramServerSocket::LeaveGroup(
-    const IPAddressNumber& group_address) const {
-  return LeaveGroupInternal(IPAddressToString(group_address));
+                                         const CompletionCallback& callback) {
+  return SendToInternal(std::string(buf->data(), buf_len), callback);
 }
 
 void MockMDnsDatagramServerSocket::SetResponsePacket(
     std::string response_packet) {
   response_packet_ = response_packet;
 }
 
 int MockMDnsDatagramServerSocket::HandleRecvNow(
     IOBuffer* buffer, int size, IPEndPoint* address,
     const CompletionCallback& callback) {
@@ skipping 10 matching lines @@
   base::MessageLoop::current()->PostTask(FROM_HERE, base::Bind(callback, rv));
   return ERR_IO_PENDING;
 }
 
 MockMDnsSocketFactory::MockMDnsSocketFactory() {
 }
 
 MockMDnsSocketFactory::~MockMDnsSocketFactory() {
 }
 
-scoped_ptr<DatagramServerSocket> MockMDnsSocketFactory::CreateSocket() {
-  scoped_ptr<MockMDnsDatagramServerSocket> new_socket(
-      new testing:: NiceMock<MockMDnsDatagramServerSocket>);
+void MockMDnsSocketFactory::CreateSockets(
+    ScopedVector<MDnsConnection::Socket>* sockets) {
+  for (size_t i = 0; i < 2; ++i) {
+    scoped_ptr<MockMDnsDatagramServerSocket> new_socket(
+        new testing::NiceMock<MockMDnsDatagramServerSocket>());
 
-  ON_CALL(*new_socket, SendToInternal(_, _, _))
-      .WillByDefault(Invoke(
-          this,
-          &MockMDnsSocketFactory::SendToInternal));
+    ON_CALL(*new_socket, SendToInternal(_, _))
+        .WillByDefault(Invoke(
+            this,
+            &MockMDnsSocketFactory::SendToInternal));
 
-  ON_CALL(*new_socket, RecvFrom(_, _, _, _))
-      .WillByDefault(Invoke(
-          this,
-          &MockMDnsSocketFactory::RecvFromInternal));
+    ON_CALL(*new_socket, RecvFrom(_, _, _, _))
+        .WillByDefault(Invoke(
+            this,
+            &MockMDnsSocketFactory::RecvFromInternal));
 
-  return new_socket.PassAs<DatagramServerSocket>();
+    sockets->push_back(new_socket.release());
+  }
 }
 
 void MockMDnsSocketFactory::SimulateReceive(const uint8* packet, int size) {
   DCHECK(recv_buffer_size_ >= size);
   DCHECK(recv_buffer_.get());
   DCHECK(!recv_callback_.is_null());
 
   memcpy(recv_buffer_->data(), packet, size);
   CompletionCallback recv_callback = recv_callback_;
   recv_callback_.Reset();
   recv_callback.Run(size);
 }
 
 int MockMDnsSocketFactory::RecvFromInternal(
     IOBuffer* buffer, int size,
     IPEndPoint* address,
     const CompletionCallback& callback) {
     recv_buffer_ = buffer;
     recv_buffer_size_ = size;
     recv_callback_ = callback;
     return ERR_IO_PENDING;
 }
 
-int MockMDnsSocketFactory::SendToInternal(
-    const std::string& packet, const std::string& address,
-    const CompletionCallback& callback) {
+int MockMDnsSocketFactory::SendToInternal(const std::string& packet,
+                                          const CompletionCallback& callback) {
   OnSendTo(packet);
   return packet.size();
 }
 
 }  // namespace net