Revert of Move service_discovery_client_impl.* into chrome/common.

chrome/utility/local_discovery/local_domain_resolver_unittest.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 "chrome/utility/local_discovery/service_discovery_client_impl.h"
+#include "net/dns/mdns_client_impl.h"
+#include "net/dns/mock_mdns_socket_factory.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using ::testing::_;
+
+namespace local_discovery {
+
+namespace {
+
+const uint8 kSamplePacketA[] = {
+  // Header
+  0x00, 0x00,               // ID is zeroed out
+  0x81, 0x80,               // Standard query response, RA, no error
+  0x00, 0x00,               // No questions (for simplicity)
+  0x00, 0x01,               // 1 RR (answers)
+  0x00, 0x00,               // 0 authority RRs
+  0x00, 0x00,               // 0 additional RRs
+
+  0x07, 'm', 'y', 'h', 'e', 'l', 'l', 'o',
+  0x05, 'l', 'o', 'c', 'a', 'l',
+  0x00,
+  0x00, 0x01,        // TYPE is A.
+  0x00, 0x01,        // CLASS is IN.
+  0x00, 0x00,        // TTL (4 bytes) is 16 seconds.
+  0x00, 0x10,
+  0x00, 0x04,        // RDLENGTH is 4 bytes.
+  0x01, 0x02,
+  0x03, 0x04,
+};
+
+const uint8 kSamplePacketAAAA[] = {
+  // Header
+  0x00, 0x00,               // ID is zeroed out
+  0x81, 0x80,               // Standard query response, RA, no error
+  0x00, 0x00,               // No questions (for simplicity)
+  0x00, 0x01,               // 1 RR (answers)
+  0x00, 0x00,               // 0 authority RRs
+  0x00, 0x00,               // 0 additional RRs
+
+  0x07, 'm', 'y', 'h', 'e', 'l', 'l', 'o',
+  0x05, 'l', 'o', 'c', 'a', 'l',
+  0x00,
+  0x00, 0x1C,        // TYPE is AAAA.
+  0x00, 0x01,        // CLASS is IN.
+  0x00, 0x00,        // TTL (4 bytes) is 16 seconds.
+  0x00, 0x10,
+  0x00, 0x10,        // RDLENGTH is 4 bytes.
+  0x00, 0x0A, 0x00, 0x00,
+  0x00, 0x00, 0x00, 0x00,
+  0x00, 0x01, 0x00, 0x02,
+  0x00, 0x03, 0x00, 0x04,
+};
+
+class LocalDomainResolverTest : public testing::Test {
+ public:
+  virtual void SetUp() OVERRIDE {
+    mdns_client_.StartListening(&socket_factory_);
+  }
+
+  std::string IPAddressToStringWithEmpty(const net::IPAddressNumber& address) {
+    if (address.empty()) return "";
+    return net::IPAddressToString(address);
+  }
+
+  void AddressCallback(bool resolved,
+                       const net::IPAddressNumber& address_ipv4,
+                       const net::IPAddressNumber& address_ipv6) {
+      AddressCallbackInternal(resolved,
+                              IPAddressToStringWithEmpty(address_ipv4),
+                              IPAddressToStringWithEmpty(address_ipv6));
+  }
+
+  void RunFor(base::TimeDelta time_period) {
+    base::CancelableCallback<void()> callback(base::Bind(
+        &base::MessageLoop::Quit,
+        base::Unretained(base::MessageLoop::current())));
+    base::MessageLoop::current()->PostDelayedTask(
+        FROM_HERE, callback.callback(), time_period);
+
+    base::MessageLoop::current()->Run();
+    callback.Cancel();
+  }
+
+  MOCK_METHOD3(AddressCallbackInternal,
+               void(bool resolved,
+                    std::string address_ipv4,
+                    std::string address_ipv6));
+
+  net::MockMDnsSocketFactory socket_factory_;
+  net::MDnsClientImpl mdns_client_;
+  base::MessageLoop message_loop_;
+};
+
+TEST_F(LocalDomainResolverTest, ResolveDomainA) {
+  LocalDomainResolverImpl resolver(
+      "myhello.local", net::ADDRESS_FAMILY_IPV4,
+      base::Bind(&LocalDomainResolverTest::AddressCallback,
+                 base::Unretained(this)), &mdns_client_);
+
+  EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(2);  // Twice per query
+
+  resolver.Start();
+
+  EXPECT_CALL(*this, AddressCallbackInternal(true, "1.2.3.4", ""));
+
+  socket_factory_.SimulateReceive(kSamplePacketA, sizeof(kSamplePacketA));
+}
+
+TEST_F(LocalDomainResolverTest, ResolveDomainAAAA) {
+  LocalDomainResolverImpl resolver(
+      "myhello.local", net::ADDRESS_FAMILY_IPV6,
+      base::Bind(&LocalDomainResolverTest::AddressCallback,
+                 base::Unretained(this)), &mdns_client_);
+
+  EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(2);  // Twice per query
+
+  resolver.Start();
+
+  EXPECT_CALL(*this, AddressCallbackInternal(true, "", "a::1:2:3:4"));
+
+  socket_factory_.SimulateReceive(kSamplePacketAAAA, sizeof(kSamplePacketAAAA));
+}
+
+TEST_F(LocalDomainResolverTest, ResolveDomainAnyOneAvailable) {
+  LocalDomainResolverImpl resolver(
+      "myhello.local", net::ADDRESS_FAMILY_UNSPECIFIED,
+      base::Bind(&LocalDomainResolverTest::AddressCallback,
+                 base::Unretained(this)), &mdns_client_);
+
+  EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(4);  // Twice per query
+
+  resolver.Start();
+
+  socket_factory_.SimulateReceive(kSamplePacketAAAA, sizeof(kSamplePacketAAAA));
+
+  EXPECT_CALL(*this, AddressCallbackInternal(true, "", "a::1:2:3:4"));
+
+  RunFor(base::TimeDelta::FromMilliseconds(150));
+}
+
+
+TEST_F(LocalDomainResolverTest, ResolveDomainAnyBothAvailable) {
+  LocalDomainResolverImpl resolver(
+      "myhello.local", net::ADDRESS_FAMILY_UNSPECIFIED,
+      base::Bind(&LocalDomainResolverTest::AddressCallback,
+                 base::Unretained(this)), &mdns_client_);
+
+  EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(4);  // Twice per query
+
+  resolver.Start();
+
+  EXPECT_CALL(*this, AddressCallbackInternal(true, "1.2.3.4", "a::1:2:3:4"));
+
+  socket_factory_.SimulateReceive(kSamplePacketAAAA, sizeof(kSamplePacketAAAA));
+
+  socket_factory_.SimulateReceive(kSamplePacketA, sizeof(kSamplePacketA));
+}
+
+TEST_F(LocalDomainResolverTest, ResolveDomainNone) {
+  LocalDomainResolverImpl resolver(
+      "myhello.local", net::ADDRESS_FAMILY_UNSPECIFIED,
+      base::Bind(&LocalDomainResolverTest::AddressCallback,
+                 base::Unretained(this)), &mdns_client_);
+
+  EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(4);  // Twice per query
+
+  resolver.Start();
+
+  EXPECT_CALL(*this, AddressCallbackInternal(false, "", ""));
+
+  RunFor(base::TimeDelta::FromSeconds(4));
+}
+
+}  // namespace
+
+}  // namespace local_discovery