Initial chrome.mdns API.

chrome/browser/extensions/api/mdns/mdns_apitest.cc

+// Copyright (c) 2012 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/command_line.h"
+#include "chrome/browser/extensions/api/mdns/mdns_api.h"
+#include "chrome/browser/extensions/extension_apitest.h"
+#include "chrome/browser/extensions/extension_service.h"
+#include "chrome/common/chrome_switches.h"
+#include "chrome/common/extensions/api/mdns.h"
+#include "testing/gmock/include/gmock/gmock.h"
+
+using extensions::NsdRegistry;
+using ::testing::A;
+using ::testing::_;
+
+namespace api = extensions::api;
+
+namespace {
+
+class MockNsdRegistry : public NsdRegistry {
+ public:
+  explicit MockNsdRegistry(extensions::MDnsAPI* api) : api_(api) {}
+  virtual ~MockNsdRegistry() {}
+
+  MOCK_METHOD1(AddObserver, void(NsdObserver* observer));
+  MOCK_METHOD1(RemoveObserver, void(NsdObserver* observer));
+  MOCK_METHOD1(RegisterDnsSdListener, void(std::string service_type));
+  MOCK_METHOD1(UnregisterDnsSdListener, void(std::string service_type));
+
+  void DispatchMDnsEvent(const std::string& service_type,
+                         const NsdServiceList& services) {
+    api_->OnMDnsEvent(service_type, services);
+  }
+
+ private:
+  extensions::NsdRegistry::NsdObserver* api_;
+};
+
+class MDnsAPITest : public ExtensionApiTest {
+ public:
+  MDnsAPITest() {}
+
+  virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
+    ExtensionApiTest::SetUpCommandLine(command_line);
+    command_line->AppendSwitchASCII(
+        switches::kWhitelistedExtensionID, "ddchlicdkolnonkihahngkmmmjnjlkkf");
+  }
+
+  void SetUpTestNsdRegistry() {
+    extensions::MDnsAPI* api = extensions::MDnsAPI::Get(profile());
+    nsd_registry_ = new MockNsdRegistry(api);
+    // Transfers ownership of the registry instance.
+    api->SetNsdRegistryForTesting(
+        make_scoped_ptr<NsdRegistry>(nsd_registry_).Pass());
+  }
+
+ protected:
+  MockNsdRegistry* nsd_registry_;
+};
+
+}  // namespace
+
+// Test loading extension, registering an MDNS listener and dispatching events.
+IN_PROC_BROWSER_TEST_F(MDnsAPITest, RegisterListener) {
+  const std::string& service_type = "_googlecast._tcp.local";
+  SetUpTestNsdRegistry();
+
+  EXPECT_CALL(*nsd_registry_, RegisterDnsSdListener(service_type))
+      .Times(1);
+  EXPECT_CALL(*nsd_registry_, UnregisterDnsSdListener(service_type))
+      .Times(1);
+  EXPECT_CALL(*nsd_registry_,
+              RemoveObserver(A<extensions::NsdRegistry::NsdObserver*>()))
+      .Times(1);
+
+  EXPECT_TRUE(RunExtensionSubtest("mdns/api", "register_listener.html"))
+      << message_;
+
+  ResultCatcher catcher;
+  // Dispatch 3 events, one of which should not be sent to the test extension.
+  std::vector<std::string> services;
+  services.push_back("test");
+
+  nsd_registry_->DispatchMDnsEvent(service_type, services);
+  nsd_registry_->DispatchMDnsEvent("_uninteresting._tcp.local", services);
+  nsd_registry_->DispatchMDnsEvent(service_type, services);
+  EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
+}
+
+// Test loading extension and registering multiple listeners.
+IN_PROC_BROWSER_TEST_F(MDnsAPITest, RegisterMultipleListeners) {
+  const std::string& service_type = "_googlecast._tcp.local";
+  const std::string& test_service_type = "_testing._tcp.local";
+  SetUpTestNsdRegistry();
+
+  EXPECT_CALL(*nsd_registry_, RegisterDnsSdListener(service_type))
+      .Times(1);
+  EXPECT_CALL(*nsd_registry_, UnregisterDnsSdListener(service_type))
+      .Times(1);
+  EXPECT_CALL(*nsd_registry_, RegisterDnsSdListener(test_service_type))
+      .Times(1);
+  EXPECT_CALL(*nsd_registry_, UnregisterDnsSdListener(test_service_type))
+      .Times(1);
+  EXPECT_CALL(*nsd_registry_,
+              RemoveObserver(A<extensions::NsdRegistry::NsdObserver*>()))
+      .Times(1);
+
+  EXPECT_TRUE(RunExtensionSubtest("mdns/api",
+                                  "register_multiple_listeners.html"))
+      << message_;
+
+  ResultCatcher catcher;
+  std::vector<std::string> services;
+  services.push_back("test");
+
+  nsd_registry_->DispatchMDnsEvent(service_type, services);
+  nsd_registry_->DispatchMDnsEvent(test_service_type, services);
+  EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
+}