1 // Copyright (c) 2012 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/dial/dial_api.h"
7 #include "chrome/browser/extensions/api/dial/dial_api_factory.h"
8 #include "chrome/browser/extensions/api/dial/dial_registry.h"
9 #include "chrome/browser/extensions/extension_apitest.h"
10 #include "chrome/browser/extensions/extension_service.h"
11 #include "extensions/common/switches.h"
12 #include "extensions/test/extension_test_message_listener.h"
13 #include "extensions/test/result_catcher.h"
14 #include "testing/gmock/include/gmock/gmock.h"
15 #include "url/gurl.h"
16
17 using extensions::DialDeviceData;
18 using extensions::Extension;
19
20 namespace api = extensions::api;
21
22 namespace {
23
24 class DialAPITest : public ExtensionApiTest {
25 public:
DialAPITest()26 DialAPITest() {}
27
SetUpCommandLine(CommandLine * command_line)28 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
29 ExtensionApiTest::SetUpCommandLine(command_line);
30 command_line->AppendSwitchASCII(
31 extensions::switches::kWhitelistedExtensionID,
32 "ddchlicdkolnonkihahngkmmmjnjlkkf");
33 }
34 };
35
36 } // namespace
37
38 // http://crbug.com/177163
39 #if defined(OS_WIN) && !defined(NDEBUG)
40 #define MAYBE_DeviceListEvents DISABLED_DeviceListEvents
41 #else
42 #define MAYBE_DeviceListEvents DeviceListEvents
43 #endif
44 // Test receiving DIAL API events.
IN_PROC_BROWSER_TEST_F(DialAPITest,MAYBE_DeviceListEvents)45 IN_PROC_BROWSER_TEST_F(DialAPITest, MAYBE_DeviceListEvents) {
46 // Setup the test.
47 ASSERT_TRUE(RunExtensionSubtest("dial/experimental", "device_list.html"));
48
49 // Send three device list updates.
50 scoped_refptr<extensions::DialAPI> api =
51 extensions::DialAPIFactory::GetInstance()->GetForBrowserContext(
52 profile());
53 ASSERT_TRUE(api.get());
54 extensions::DialRegistry::DeviceList devices;
55
56 extensions::ResultCatcher catcher;
57
58 DialDeviceData device1;
59 device1.set_device_id("1");
60 device1.set_label("1");
61 device1.set_device_description_url(GURL("http://127.0.0.1/dd.xml"));
62
63 devices.push_back(device1);
64 api->SendEventOnUIThread(devices);
65
66 DialDeviceData device2;
67 device2.set_device_id("2");
68 device2.set_label("2");
69 device2.set_device_description_url(GURL("http://127.0.0.2/dd.xml"));
70
71 devices.push_back(device2);
72 api->SendEventOnUIThread(devices);
73
74 DialDeviceData device3;
75 device3.set_device_id("3");
76 device3.set_label("3");
77 device3.set_device_description_url(GURL("http://127.0.0.3/dd.xml"));
78
79 devices.push_back(device3);
80 api->SendEventOnUIThread(devices);
81
82 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
83 }
84
IN_PROC_BROWSER_TEST_F(DialAPITest,Discovery)85 IN_PROC_BROWSER_TEST_F(DialAPITest, Discovery) {
86 ASSERT_TRUE(RunExtensionSubtest("dial/experimental", "discovery.html"));
87 }
88
89 // discoverNow does not do discovery when there are no listeners; in that case
90 // the DIAL service will not be active.
IN_PROC_BROWSER_TEST_F(DialAPITest,DiscoveryNoListeners)91 IN_PROC_BROWSER_TEST_F(DialAPITest, DiscoveryNoListeners) {
92 ASSERT_TRUE(RunExtensionSubtest("dial/experimental",
93 "discovery_no_listeners.html"));
94 }
95
96 // Make sure this API is only accessible to whitelisted extensions.
IN_PROC_BROWSER_TEST_F(DialAPITest,NonWhitelistedExtension)97 IN_PROC_BROWSER_TEST_F(DialAPITest, NonWhitelistedExtension) {
98 extensions::ResultCatcher catcher;
99 catcher.RestrictToBrowserContext(browser()->profile());
100
101 ExtensionTestMessageListener listener("ready", true);
102 const extensions::Extension* extension = LoadExtensionWithFlags(
103 test_data_dir_.AppendASCII("dial/whitelist"),
104 ExtensionBrowserTest::kFlagIgnoreManifestWarnings);
105 // We should have a DIAL API not available warning.
106 ASSERT_FALSE(extension->install_warnings().empty());
107
108 EXPECT_TRUE(listener.WaitUntilSatisfied());
109
110 listener.Reply("go");
111 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
112 }
113
IN_PROC_BROWSER_TEST_F(DialAPITest,OnError)114 IN_PROC_BROWSER_TEST_F(DialAPITest, OnError) {
115 ASSERT_TRUE(RunExtensionSubtest("dial/experimental", "on_error.html"));
116 }
117