Add queryForNewLocalDevices to gcdPrivate

chrome/browser/extensions/api/gcd_private/gcd_private_apitest.cc

 // Copyright 2014 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 "base/bind.h"
 #include "base/command_line.h"
 #include "base/json/json_reader.h"
 #include "base/message_loop/message_loop_proxy.h"
 #include "chrome/browser/extensions/api/gcd_private/gcd_private_api.h"
 #include "chrome/browser/extensions/extension_apitest.h"
@@ skipping 120 matching lines @@
     0x09, 'm',  'y',  'S',  'e',  'r',  'v', 'i',  'c',  'e',  0xc0, 0x0c,
     0x00, 0x21,                          // Type is SRV
     0x00, 0x01,                          // CLASS is IN
     0x00, 0x00,                          // TTL (4 bytes) is 0 seconds.
     0x00, 0x00, 0x00, 0x17,              // RDLENGTH is 23
     0x00, 0x00, 0x00, 0x00, 0x22, 0xb8,  // port 8888
     0x09, 'm',  'y',  'S',  'e',  'r',  'v', 'i',  'c',  'e',  0x05, 'l',
     'o',  'c',  'a',  'l',  0x00,
 };
 
+const uint8 kQueryPacket[] = {
+    // Header
+    0x00, 0x00,  // ID is zeroed out
+    0x00, 0x00,  // No flags.
+    0x00, 0x01,  // One question.
+    0x00, 0x00,  // 0 RRs (answers)
+    0x00, 0x00,  // 0 authority RRs
+    0x00, 0x00,  // 0 additional RRs
+
+    // Question
+    // This part is echoed back from the respective query.
+    0x07, '_',  'p', 'r', 'i', 'v', 'e', 't',  0x04, '_',  't', 'c',
+    'p',  0x05, 'l', 'o', 'c', 'a', 'l', 0x00, 0x00, 0x0c,  // TYPE is PTR.
+    0x00, 0x01,                                             // CLASS is IN.
+};
+
 #endif  // ENABLE_MDNS
 
 // Sentinel value to signify the request should fail.
 const char kResponseValueFailure[] = "FAILURE";
 
 class FakeGCDApiFlowFactory
     : public extensions::GcdPrivateAPI::GCDApiFlowFactoryForTests {
  public:
   FakeGCDApiFlowFactory() {
     extensions::GcdPrivateAPI::SetGCDApiFlowFactoryForTests(this);
@@ skipping 150 matching lines @@
       FROM_HERE,
       base::Bind(&local_discovery::TestServiceDiscoveryClient::SimulateReceive,
                  test_service_discovery_client_,
                  kGoodbyePacket,
                  sizeof(kGoodbyePacket)),
       base::TimeDelta::FromSeconds(1));
 
   EXPECT_TRUE(RunExtensionSubtest("gcd_private/api", "remove_device.html"));
 }
 
+IN_PROC_BROWSER_TEST_F(GcdPrivateAPITest, SendQuery) {
+  EXPECT_CALL(*test_service_discovery_client_,
+              OnSendTo(std::string(reinterpret_cast<const char*>(kQueryPacket),
+                                   sizeof(kQueryPacket)))).Times(2);
+
+  EXPECT_TRUE(RunExtensionSubtest("gcd_private/api", "send_query.html"));
+}
+
 #endif  // ENABLE_MDNS
 
 #if defined(ENABLE_WIFI_BOOTSTRAPPING)
 IN_PROC_BROWSER_TEST_F(GcdPrivateAPITest, WifiAdd) {
   response_callback_ = base::Bind(&GcdPrivateAPITest::RespondWithFilledList,
                                   base::Unretained(this));
 
   EXPECT_CALL(wifi_manager_factory_, OnWifiManagerCreated(_)).WillOnce(
       Invoke(this, &GcdPrivateAPITest::RespondToSSIDListWithCallback));
 
   EXPECT_TRUE(RunExtensionSubtest("gcd_private/api", "add_wifi_device.html"));
 }
 
 IN_PROC_BROWSER_TEST_F(GcdPrivateAPITest, WifiAddRemove) {
   response_callback_ = base::Bind(&GcdPrivateAPITest::RespondWithAddRemove,
                                   base::Unretained(this));
 
   EXPECT_CALL(wifi_manager_factory_, OnWifiManagerCreated(_)).WillOnce(
       Invoke(this, &GcdPrivateAPITest::RespondToSSIDListWithCallback));
 
   EXPECT_TRUE(
       RunExtensionSubtest("gcd_private/api", "remove_wifi_device.html"));
 }
+
 #endif
 
 }  // namespace