Index: trunk/src/chrome/browser/extensions/api/mdns/mdns_apitest.cc |
=================================================================== |
--- trunk/src/chrome/browser/extensions/api/mdns/mdns_apitest.cc (revision 221667) |
+++ trunk/src/chrome/browser/extensions/api/mdns/mdns_apitest.cc (working copy) |
@@ -1,120 +0,0 @@ |
-// 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 "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::DnsSdRegistry; |
-using ::testing::A; |
-using ::testing::_; |
- |
-namespace api = extensions::api; |
- |
-namespace { |
- |
-class MockDnsSdRegistry : public DnsSdRegistry { |
- public: |
- explicit MockDnsSdRegistry(extensions::MDnsAPI* api) : api_(api) {} |
- virtual ~MockDnsSdRegistry() {} |
- |
- MOCK_METHOD1(AddObserver, void(DnsSdObserver* observer)); |
- MOCK_METHOD1(RemoveObserver, void(DnsSdObserver* 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 DnsSdServiceList& services) { |
- api_->OnDnsSdEvent(service_type, services); |
- } |
- |
- private: |
- extensions::DnsSdRegistry::DnsSdObserver* 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 SetUpTestDnsSdRegistry() { |
- extensions::MDnsAPI* api = extensions::MDnsAPI::Get(profile()); |
- dns_sd_registry_ = new MockDnsSdRegistry(api); |
- // Transfers ownership of the registry instance. |
- api->SetDnsSdRegistryForTesting( |
- make_scoped_ptr<DnsSdRegistry>(dns_sd_registry_).Pass()); |
- } |
- |
- protected: |
- MockDnsSdRegistry* dns_sd_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"; |
- SetUpTestDnsSdRegistry(); |
- |
- EXPECT_CALL(*dns_sd_registry_, RegisterDnsSdListener(service_type)) |
- .Times(1); |
- EXPECT_CALL(*dns_sd_registry_, UnregisterDnsSdListener(service_type)) |
- .Times(1); |
- EXPECT_CALL(*dns_sd_registry_, |
- RemoveObserver(A<extensions::DnsSdRegistry::DnsSdObserver*>())) |
- .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"); |
- |
- dns_sd_registry_->DispatchMDnsEvent(service_type, services); |
- dns_sd_registry_->DispatchMDnsEvent("_uninteresting._tcp.local", services); |
- dns_sd_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"; |
- SetUpTestDnsSdRegistry(); |
- |
- EXPECT_CALL(*dns_sd_registry_, RegisterDnsSdListener(service_type)) |
- .Times(1); |
- EXPECT_CALL(*dns_sd_registry_, UnregisterDnsSdListener(service_type)) |
- .Times(1); |
- EXPECT_CALL(*dns_sd_registry_, RegisterDnsSdListener(test_service_type)) |
- .Times(1); |
- EXPECT_CALL(*dns_sd_registry_, UnregisterDnsSdListener(test_service_type)) |
- .Times(1); |
- EXPECT_CALL(*dns_sd_registry_, |
- RemoveObserver(A<extensions::DnsSdRegistry::DnsSdObserver*>())) |
- .Times(1); |
- |
- EXPECT_TRUE(RunExtensionSubtest("mdns/api", |
- "register_multiple_listeners.html")) |
- << message_; |
- |
- ResultCatcher catcher; |
- std::vector<std::string> services; |
- services.push_back("test"); |
- |
- dns_sd_registry_->DispatchMDnsEvent(service_type, services); |
- dns_sd_registry_->DispatchMDnsEvent(test_service_type, services); |
- EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); |
-} |