1 // Copyright 2014 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.
5 #include "chrome/common/local_discovery/service_discovery_client_impl.h"
6 #include "net/dns/mdns_client_impl.h"
7 #include "net/dns/mock_mdns_socket_factory.h"
8 #include "testing/gmock/include/gmock/gmock.h"
9 #include "testing/gtest/include/gtest/gtest.h"
13 namespace local_discovery {
17 const uint8 kSamplePacketA[] = {
19 0x00, 0x00, // ID is zeroed out
20 0x81, 0x80, // Standard query response, RA, no error
21 0x00, 0x00, // No questions (for simplicity)
22 0x00, 0x01, // 1 RR (answers)
23 0x00, 0x00, // 0 authority RRs
24 0x00, 0x00, // 0 additional RRs
26 0x07, 'm', 'y', 'h', 'e', 'l', 'l', 'o',
27 0x05, 'l', 'o', 'c', 'a', 'l',
29 0x00, 0x01, // TYPE is A.
30 0x00, 0x01, // CLASS is IN.
31 0x00, 0x00, // TTL (4 bytes) is 16 seconds.
33 0x00, 0x04, // RDLENGTH is 4 bytes.
38 const uint8 kSamplePacketAAAA[] = {
40 0x00, 0x00, // ID is zeroed out
41 0x81, 0x80, // Standard query response, RA, no error
42 0x00, 0x00, // No questions (for simplicity)
43 0x00, 0x01, // 1 RR (answers)
44 0x00, 0x00, // 0 authority RRs
45 0x00, 0x00, // 0 additional RRs
47 0x07, 'm', 'y', 'h', 'e', 'l', 'l', 'o',
48 0x05, 'l', 'o', 'c', 'a', 'l',
50 0x00, 0x1C, // TYPE is AAAA.
51 0x00, 0x01, // CLASS is IN.
52 0x00, 0x00, // TTL (4 bytes) is 16 seconds.
54 0x00, 0x10, // RDLENGTH is 4 bytes.
55 0x00, 0x0A, 0x00, 0x00,
56 0x00, 0x00, 0x00, 0x00,
57 0x00, 0x01, 0x00, 0x02,
58 0x00, 0x03, 0x00, 0x04,
61 class LocalDomainResolverTest : public testing::Test {
63 virtual void SetUp() OVERRIDE {
64 mdns_client_.StartListening(&socket_factory_);
67 std::string IPAddressToStringWithEmpty(const net::IPAddressNumber& address) {
68 if (address.empty()) return "";
69 return net::IPAddressToString(address);
72 void AddressCallback(bool resolved,
73 const net::IPAddressNumber& address_ipv4,
74 const net::IPAddressNumber& address_ipv6) {
75 AddressCallbackInternal(resolved,
76 IPAddressToStringWithEmpty(address_ipv4),
77 IPAddressToStringWithEmpty(address_ipv6));
80 void RunFor(base::TimeDelta time_period) {
81 base::CancelableCallback<void()> callback(base::Bind(
82 &base::MessageLoop::Quit,
83 base::Unretained(base::MessageLoop::current())));
84 base::MessageLoop::current()->PostDelayedTask(
85 FROM_HERE, callback.callback(), time_period);
87 base::MessageLoop::current()->Run();
91 MOCK_METHOD3(AddressCallbackInternal,
93 std::string address_ipv4,
94 std::string address_ipv6));
96 net::MockMDnsSocketFactory socket_factory_;
97 net::MDnsClientImpl mdns_client_;
98 base::MessageLoop message_loop_;
101 TEST_F(LocalDomainResolverTest, ResolveDomainA) {
102 LocalDomainResolverImpl resolver(
103 "myhello.local", net::ADDRESS_FAMILY_IPV4,
104 base::Bind(&LocalDomainResolverTest::AddressCallback,
105 base::Unretained(this)), &mdns_client_);
107 EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(2); // Twice per query
111 EXPECT_CALL(*this, AddressCallbackInternal(true, "1.2.3.4", ""));
113 socket_factory_.SimulateReceive(kSamplePacketA, sizeof(kSamplePacketA));
116 TEST_F(LocalDomainResolverTest, ResolveDomainAAAA) {
117 LocalDomainResolverImpl resolver(
118 "myhello.local", net::ADDRESS_FAMILY_IPV6,
119 base::Bind(&LocalDomainResolverTest::AddressCallback,
120 base::Unretained(this)), &mdns_client_);
122 EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(2); // Twice per query
126 EXPECT_CALL(*this, AddressCallbackInternal(true, "", "a::1:2:3:4"));
128 socket_factory_.SimulateReceive(kSamplePacketAAAA, sizeof(kSamplePacketAAAA));
131 TEST_F(LocalDomainResolverTest, ResolveDomainAnyOneAvailable) {
132 LocalDomainResolverImpl resolver(
133 "myhello.local", net::ADDRESS_FAMILY_UNSPECIFIED,
134 base::Bind(&LocalDomainResolverTest::AddressCallback,
135 base::Unretained(this)), &mdns_client_);
137 EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(4); // Twice per query
141 socket_factory_.SimulateReceive(kSamplePacketAAAA, sizeof(kSamplePacketAAAA));
143 EXPECT_CALL(*this, AddressCallbackInternal(true, "", "a::1:2:3:4"));
145 RunFor(base::TimeDelta::FromMilliseconds(150));
149 TEST_F(LocalDomainResolverTest, ResolveDomainAnyBothAvailable) {
150 LocalDomainResolverImpl resolver(
151 "myhello.local", net::ADDRESS_FAMILY_UNSPECIFIED,
152 base::Bind(&LocalDomainResolverTest::AddressCallback,
153 base::Unretained(this)), &mdns_client_);
155 EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(4); // Twice per query
159 EXPECT_CALL(*this, AddressCallbackInternal(true, "1.2.3.4", "a::1:2:3:4"));
161 socket_factory_.SimulateReceive(kSamplePacketAAAA, sizeof(kSamplePacketAAAA));
163 socket_factory_.SimulateReceive(kSamplePacketA, sizeof(kSamplePacketA));
166 TEST_F(LocalDomainResolverTest, ResolveDomainNone) {
167 LocalDomainResolverImpl resolver(
168 "myhello.local", net::ADDRESS_FAMILY_UNSPECIFIED,
169 base::Bind(&LocalDomainResolverTest::AddressCallback,
170 base::Unretained(this)), &mdns_client_);
172 EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(4); // Twice per query
176 EXPECT_CALL(*this, AddressCallbackInternal(false, "", ""));
178 RunFor(base::TimeDelta::FromSeconds(4));
183 } // namespace local_discovery