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:
MockDnsSdRegistry(extensions::MDnsAPI * api)23 explicit MockDnsSdRegistry(extensions::MDnsAPI* api) : api_(api) {}
~MockDnsSdRegistry()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
DispatchMDnsEvent(const std::string & service_type,const DnsSdServiceList & services)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:
MDnsAPITest()42 MDnsAPITest() {}
43
SetUpCommandLine(CommandLine * command_line)44 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
45 ExtensionApiTest::SetUpCommandLine(command_line);
46 command_line->AppendSwitchASCII(
47 switches::kWhitelistedExtensionID, "ddchlicdkolnonkihahngkmmmjnjlkkf");
48 }
49
SetUpTestDnsSdRegistry()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 // TODO(justinlin): Win Dbg has a workaround that makes RunExtensionSubtest
65 // always return true without actually running the test. Remove when fixed.
66 #if defined(OS_WIN) && !defined(NDEBUG)
67 #define MAYBE_RegisterListener DISABLED_RegisterListener
68 #else
69 #define MAYBE_RegisterListener RegisterListener
70 #endif
71 // Test loading extension, registering an MDNS listener and dispatching events.
IN_PROC_BROWSER_TEST_F(MDnsAPITest,MAYBE_RegisterListener)72 IN_PROC_BROWSER_TEST_F(MDnsAPITest, MAYBE_RegisterListener) {
73 const std::string& service_type = "_googlecast._tcp.local";
74 SetUpTestDnsSdRegistry();
75
76 EXPECT_CALL(*dns_sd_registry_, RegisterDnsSdListener(service_type))
77 .Times(1);
78 EXPECT_CALL(*dns_sd_registry_, UnregisterDnsSdListener(service_type))
79 .Times(1);
80 EXPECT_CALL(*dns_sd_registry_,
81 RemoveObserver(A<extensions::DnsSdRegistry::DnsSdObserver*>()))
82 .Times(1);
83
84 EXPECT_TRUE(RunExtensionSubtest("mdns/api", "register_listener.html"))
85 << message_;
86
87 ResultCatcher catcher;
88 // Dispatch 3 events, one of which should not be sent to the test extension.
89 DnsSdRegistry::DnsSdServiceList services;
90
91 extensions::DnsSdService service;
92 service.service_name = service_type;
93 services.push_back(service);
94
95 dns_sd_registry_->DispatchMDnsEvent(service_type, services);
96 dns_sd_registry_->DispatchMDnsEvent("_uninteresting._tcp.local", services);
97 dns_sd_registry_->DispatchMDnsEvent(service_type, services);
98 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
99 }
100
101 // TODO(justinlin): Win Dbg has a workaround that makes RunExtensionSubtest
102 // always return true without actually running the test. Remove when fixed.
103 #if defined(OS_WIN) && !defined(NDEBUG)
104 #define MAYBE_RegisterMultipleListeners DISABLED_RegisterMultipleListeners
105 #else
106 #define MAYBE_RegisterMultipleListeners RegisterMultipleListeners
107 #endif
108 // Test loading extension and registering multiple listeners.
IN_PROC_BROWSER_TEST_F(MDnsAPITest,MAYBE_RegisterMultipleListeners)109 IN_PROC_BROWSER_TEST_F(MDnsAPITest, MAYBE_RegisterMultipleListeners) {
110 const std::string& service_type = "_googlecast._tcp.local";
111 const std::string& test_service_type = "_testing._tcp.local";
112 SetUpTestDnsSdRegistry();
113
114 EXPECT_CALL(*dns_sd_registry_, RegisterDnsSdListener(service_type))
115 .Times(1);
116 EXPECT_CALL(*dns_sd_registry_, UnregisterDnsSdListener(service_type))
117 .Times(1);
118 EXPECT_CALL(*dns_sd_registry_, RegisterDnsSdListener(test_service_type))
119 .Times(1);
120 EXPECT_CALL(*dns_sd_registry_, UnregisterDnsSdListener(test_service_type))
121 .Times(1);
122 EXPECT_CALL(*dns_sd_registry_,
123 RemoveObserver(A<extensions::DnsSdRegistry::DnsSdObserver*>()))
124 .Times(1);
125
126 EXPECT_TRUE(RunExtensionSubtest("mdns/api",
127 "register_multiple_listeners.html"))
128 << message_;
129
130 ResultCatcher catcher;
131 DnsSdRegistry::DnsSdServiceList services;
132
133 extensions::DnsSdService service;
134 service.service_name = service_type;
135 services.push_back(service);
136
137 dns_sd_registry_->DispatchMDnsEvent(service_type, services);
138 dns_sd_registry_->DispatchMDnsEvent(test_service_type, services);
139 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
140 }
141