| OLD | NEW |
|---|
| (Empty) |
| 1 // Copyright 2013 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. | |
| 4 | |
| 5 #include "base/command_line.h" | |
| 6 #include "chrome/browser/extensions/api/mdns/mdns_api.h" | |
| 7 #include "chrome/browser/extensions/extension_apitest.h" | |
| 8 #include "chrome/browser/extensions/extension_service.h" | |
| 9 #include "chrome/common/chrome_switches.h" | |
| 10 #include "chrome/common/extensions/api/mdns.h" | |
| 11 #include "testing/gmock/include/gmock/gmock.h" | |
| 12 | |
| 13 using extensions::DnsSdRegistry; | |
| 14 using ::testing::A; | |
| 15 using ::testing::_; | |
| 16 | |
| 17 namespace api = extensions::api; | |
| 18 | |
| 19 namespace { | |
| 20 | |
| 21 class MockDnsSdRegistry : public DnsSdRegistry { | |
| 22 public: | |
| 23 explicit MockDnsSdRegistry(extensions::MDnsAPI* api) : api_(api) {} | |
| 24 virtual ~MockDnsSdRegistry() {} | |
| 25 | |
| 26 MOCK_METHOD1(AddObserver, void(DnsSdObserver* observer)); | |
| 27 MOCK_METHOD1(RemoveObserver, void(DnsSdObserver* observer)); | |
| 28 MOCK_METHOD1(RegisterDnsSdListener, void(std::string service_type)); | |
| 29 MOCK_METHOD1(UnregisterDnsSdListener, void(std::string service_type)); | |
| 30 | |
| 31 void DispatchMDnsEvent(const std::string& service_type, | |
| 32 const DnsSdServiceList& services) { | |
| 33 api_->OnDnsSdEvent(service_type, services); | |
| 34 } | |
| 35 | |
| 36 private: | |
| 37 extensions::DnsSdRegistry::DnsSdObserver* api_; | |
| 38 }; | |
| 39 | |
| 40 class MDnsAPITest : public ExtensionApiTest { | |
| 41 public: | |
| 42 MDnsAPITest() {} | |
| 43 | |
| 44 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { | |
| 45 ExtensionApiTest::SetUpCommandLine(command_line); | |
| 46 command_line->AppendSwitchASCII( | |
| 47 switches::kWhitelistedExtensionID, "ddchlicdkolnonkihahngkmmmjnjlkkf"); | |
| 48 } | |
| 49 | |
| 50 void SetUpTestDnsSdRegistry() { | |
| 51 extensions::MDnsAPI* api = extensions::MDnsAPI::Get(profile()); | |
| 52 dns_sd_registry_ = new MockDnsSdRegistry(api); | |
| 53 // Transfers ownership of the registry instance. | |
| 54 api->SetDnsSdRegistryForTesting( | |
| 55 make_scoped_ptr<DnsSdRegistry>(dns_sd_registry_).Pass()); | |
| 56 } | |
| 57 | |
| 58 protected: | |
| 59 MockDnsSdRegistry* dns_sd_registry_; | |
| 60 }; | |
| 61 | |
| 62 } // namespace | |
| 63 | |
| 64 // Test loading extension, registering an MDNS listener and dispatching events. | |
| 65 IN_PROC_BROWSER_TEST_F(MDnsAPITest, RegisterListener) { | |
| 66 const std::string& service_type = "_googlecast._tcp.local"; | |
| 67 SetUpTestDnsSdRegistry(); | |
| 68 | |
| 69 EXPECT_CALL(*dns_sd_registry_, RegisterDnsSdListener(service_type)) | |
| 70 .Times(1); | |
| 71 EXPECT_CALL(*dns_sd_registry_, UnregisterDnsSdListener(service_type)) | |
| 72 .Times(1); | |
| 73 EXPECT_CALL(*dns_sd_registry_, | |
| 74 RemoveObserver(A<extensions::DnsSdRegistry::DnsSdObserver*>())) | |
| 75 .Times(1); | |
| 76 | |
| 77 EXPECT_TRUE(RunExtensionSubtest("mdns/api", "register_listener.html")) | |
| 78 << message_; | |
| 79 | |
| 80 ResultCatcher catcher; | |
| 81 // Dispatch 3 events, one of which should not be sent to the test extension. | |
| 82 std::vector<std::string> services; | |
| 83 services.push_back("test"); | |
| 84 | |
| 85 dns_sd_registry_->DispatchMDnsEvent(service_type, services); | |
| 86 dns_sd_registry_->DispatchMDnsEvent("_uninteresting._tcp.local", services); | |
| 87 dns_sd_registry_->DispatchMDnsEvent(service_type, services); | |
| 88 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); | |
| 89 } | |
| 90 | |
| 91 // Test loading extension and registering multiple listeners. | |
| 92 IN_PROC_BROWSER_TEST_F(MDnsAPITest, RegisterMultipleListeners) { | |
| 93 const std::string& service_type = "_googlecast._tcp.local"; | |
| 94 const std::string& test_service_type = "_testing._tcp.local"; | |
| 95 SetUpTestDnsSdRegistry(); | |
| 96 | |
| 97 EXPECT_CALL(*dns_sd_registry_, RegisterDnsSdListener(service_type)) | |
| 98 .Times(1); | |
| 99 EXPECT_CALL(*dns_sd_registry_, UnregisterDnsSdListener(service_type)) | |
| 100 .Times(1); | |
| 101 EXPECT_CALL(*dns_sd_registry_, RegisterDnsSdListener(test_service_type)) | |
| 102 .Times(1); | |
| 103 EXPECT_CALL(*dns_sd_registry_, UnregisterDnsSdListener(test_service_type)) | |
| 104 .Times(1); | |
| 105 EXPECT_CALL(*dns_sd_registry_, | |
| 106 RemoveObserver(A<extensions::DnsSdRegistry::DnsSdObserver*>())) | |
| 107 .Times(1); | |
| 108 | |
| 109 EXPECT_TRUE(RunExtensionSubtest("mdns/api", | |
| 110 "register_multiple_listeners.html")) | |
| 111 << message_; | |
| 112 | |
| 113 ResultCatcher catcher; | |
| 114 std::vector<std::string> services; | |
| 115 services.push_back("test"); | |
| 116 | |
| 117 dns_sd_registry_->DispatchMDnsEvent(service_type, services); | |
| 118 dns_sd_registry_->DispatchMDnsEvent(test_service_type, services); | |
| 119 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); | |
| 120 } | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 23945003:
Revert 221619 "Initial chrome.mdns API." (Closed)
Base URL: svn://svn.chromium.org/chrome/