blob: e1f841eaded2eac258d7bed88354bf2d2891a773 [file] [log] [blame]
[email protected]4d2451652012-02-14 23:54:261// Copyright (c) 2012 The Chromium Authors. All rights reserved.
[email protected]038d52e12009-10-14 16:53:412// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
[email protected]f0488f2f2009-07-01 05:25:225#include <vector>
6
7#include "base/command_line.h"
[email protected]f0488f2f2009-07-01 05:25:228#include "base/file_util.h"
[email protected]57999812013-02-24 05:40:529#include "base/files/file_path.h"
[email protected]f0488f2f2009-07-01 05:25:2210#include "base/path_service.h"
[email protected]bfc93562011-12-14 19:40:0411#include "base/string_util.h"
[email protected]eaa7dd182010-12-14 11:09:0012#include "chrome/browser/extensions/extension_service.h"
[email protected]5d272a42012-10-29 13:28:2213#include "chrome/browser/extensions/extension_system.h"
[email protected]8cb5d5b2010-02-09 11:36:1614#include "chrome/browser/extensions/user_script_master.h"
[email protected]8ecad5e2010-12-02 21:18:3315#include "chrome/browser/profiles/profile.h"
[email protected]7b5dc002010-11-16 23:08:1016#include "chrome/browser/ui/browser.h"
[email protected]47ae23372013-01-29 01:50:4817#include "chrome/browser/ui/tabs/tab_strip_model.h"
[email protected]432115822011-07-10 15:52:2718#include "chrome/common/chrome_notification_types.h"
[email protected]f0488f2f2009-07-01 05:25:2219#include "chrome/common/chrome_paths.h"
20#include "chrome/common/chrome_switches.h"
[email protected]00b5d0a52012-10-30 13:13:5321#include "chrome/common/extensions/feature_switch.h"
[email protected]af44e7fb2011-07-29 18:32:3222#include "chrome/test/base/in_process_browser_test.h"
[email protected]89dbb1772012-07-17 13:47:2523#include "chrome/test/base/testing_profile.h"
[email protected]af44e7fb2011-07-29 18:32:3224#include "chrome/test/base/ui_test_utils.h"
[email protected]6c2381d2011-10-19 02:52:5325#include "content/public/browser/notification_details.h"
[email protected]ad50def52011-10-19 23:17:0726#include "content/public/browser/notification_service.h"
[email protected]4ca15302012-01-03 05:53:2027#include "content/public/browser/web_contents.h"
[email protected]7d478cb2012-07-24 17:19:4228#include "content/public/test/browser_test_utils.h"
[email protected]f0488f2f2009-07-01 05:25:2229#include "net/base/net_util.h"
30
[email protected]00b5d0a52012-10-30 13:13:5331using extensions::FeatureSwitch;
32
[email protected]17c4f3c2009-07-04 16:36:2533// This file contains high-level startup tests for the extensions system. We've
34// had many silly bugs where command line flags did not get propagated correctly
35// into the services, so we didn't start correctly.
[email protected]f0488f2f2009-07-01 05:25:2236
[email protected]3e59bac2010-04-08 16:16:5537class ExtensionStartupTestBase : public InProcessBrowserTest {
[email protected]f0488f2f2009-07-01 05:25:2238 public:
[email protected]00b5d0a52012-10-30 13:13:5339 ExtensionStartupTestBase() :
[email protected]90c780ac2013-04-16 09:26:2840 enable_extensions_(false) {
[email protected]fe13bf62010-08-26 14:33:1941 num_expected_extensions_ = 3;
[email protected]f0488f2f2009-07-01 05:25:2242 }
43
44 protected:
45 // InProcessBrowserTest
[email protected]49aeab62013-02-07 02:53:1146 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
[email protected]10f6b142012-04-14 19:22:4347 if (!enable_extensions_)
48 command_line->AppendSwitch(switches::kDisableExtensions);
49
50 if (!load_extensions_.empty()) {
[email protected]dd6d36672013-05-17 22:50:0251 base::FilePath::StringType paths = JoinString(load_extensions_, ',');
[email protected]10f6b142012-04-14 19:22:4352 command_line->AppendSwitchNative(switches::kLoadExtension,
53 paths);
54 command_line->AppendSwitch(switches::kDisableExtensionsFileAccessCheck);
55 }
56 }
57
[email protected]49aeab62013-02-07 02:53:1158 virtual bool SetUpUserDataDirectory() OVERRIDE {
[email protected]650b2d52013-02-10 03:41:4559 base::FilePath profile_dir;
[email protected]f0488f2f2009-07-01 05:25:2260 PathService::Get(chrome::DIR_USER_DATA, &profile_dir);
[email protected]89dbb1772012-07-17 13:47:2561 profile_dir = profile_dir.AppendASCII(TestingProfile::kTestUserProfileDir);
[email protected]f0488f2f2009-07-01 05:25:2262 file_util::CreateDirectory(profile_dir);
63
64 preferences_file_ = profile_dir.AppendASCII("Preferences");
65 user_scripts_dir_ = profile_dir.AppendASCII("User Scripts");
66 extensions_dir_ = profile_dir.AppendASCII("Extensions");
67
[email protected]10f6b142012-04-14 19:22:4368 if (enable_extensions_ && load_extensions_.empty()) {
[email protected]650b2d52013-02-10 03:41:4569 base::FilePath src_dir;
[email protected]10f6b142012-04-14 19:22:4370 PathService::Get(chrome::DIR_TEST_DATA, &src_dir);
71 src_dir = src_dir.AppendASCII("extensions").AppendASCII("good");
[email protected]f0488f2f2009-07-01 05:25:2272
[email protected]10f6b142012-04-14 19:22:4373 file_util::CopyFile(src_dir.AppendASCII("Preferences"),
74 preferences_file_);
75 file_util::CopyDirectory(src_dir.AppendASCII("Extensions"),
76 profile_dir, true); // recursive
[email protected]f0488f2f2009-07-01 05:25:2277 }
[email protected]10f6b142012-04-14 19:22:4378 return true;
[email protected]f0488f2f2009-07-01 05:25:2279 }
80
[email protected]f0488f2f2009-07-01 05:25:2281 virtual void TearDown() {
[email protected]3e59bac2010-04-08 16:16:5582 EXPECT_TRUE(file_util::Delete(preferences_file_, false));
[email protected]020f49e2010-04-08 19:51:1283
84 // TODO(phajdan.jr): Check return values of the functions below, carefully.
85 file_util::Delete(user_scripts_dir_, true);
[email protected]9eaa021d2010-04-08 19:55:3486 file_util::Delete(extensions_dir_, true);
[email protected]36fb2c7c2011-04-04 15:49:0887
88 InProcessBrowserTest::TearDown();
[email protected]f0488f2f2009-07-01 05:25:2289 }
90
[email protected]919ddc82009-07-15 04:30:1291 void WaitForServicesToStart(int num_expected_extensions,
92 bool expect_extensions_enabled) {
[email protected]06bdd2b2012-11-30 18:47:1393 ExtensionService* service = extensions::ExtensionSystem::Get(
94 browser()->profile())->extension_service();
[email protected]919ddc82009-07-15 04:30:1295
[email protected]fe13bf62010-08-26 14:33:1996 // Count the number of non-component extensions.
97 int found_extensions = 0;
[email protected]84df8332011-12-06 18:22:4698 for (ExtensionSet::const_iterator it = service->extensions()->begin();
99 it != service->extensions()->end(); ++it)
[email protected]1d5e58b2013-01-31 08:41:40100 if ((*it)->location() != extensions::Manifest::COMPONENT)
[email protected]fe13bf62010-08-26 14:33:19101 found_extensions++;
102
[email protected]919ddc82009-07-15 04:30:12103 ASSERT_EQ(static_cast<uint32>(num_expected_extensions),
[email protected]fe13bf62010-08-26 14:33:19104 static_cast<uint32>(found_extensions));
[email protected]919ddc82009-07-15 04:30:12105 ASSERT_EQ(expect_extensions_enabled, service->extensions_enabled());
106
[email protected]a7fe9112012-07-20 02:34:45107 content::WindowedNotificationObserver user_scripts_observer(
[email protected]120abf132011-09-27 21:38:06108 chrome::NOTIFICATION_USER_SCRIPTS_UPDATED,
[email protected]ad50def52011-10-19 23:17:07109 content::NotificationService::AllSources());
[email protected]20f97c92012-07-13 23:12:37110 extensions::UserScriptMaster* master =
[email protected]5d272a42012-10-29 13:28:22111 extensions::ExtensionSystem::Get(browser()->profile())->
112 user_script_master();
[email protected]120abf132011-09-27 21:38:06113 if (!master->ScriptsReady())
114 user_scripts_observer.Wait();
[email protected]919ddc82009-07-15 04:30:12115 ASSERT_TRUE(master->ScriptsReady());
116 }
117
118 void TestInjection(bool expect_css, bool expect_script) {
119 // Load a page affected by the content script and test to see the effect.
[email protected]650b2d52013-02-10 03:41:45120 base::FilePath test_file;
[email protected]62771442009-11-22 02:25:04121 PathService::Get(chrome::DIR_TEST_DATA, &test_file);
122 test_file = test_file.AppendASCII("extensions")
123 .AppendASCII("test_file.html");
124
125 ui_test_utils::NavigateToURL(browser(), net::FilePathToFileURL(test_file));
[email protected]919ddc82009-07-15 04:30:12126
127 bool result = false;
[email protected]b6987e02013-01-04 18:30:43128 ASSERT_TRUE(content::ExecuteScriptAndExtractBool(
[email protected]47ae23372013-01-29 01:50:48129 browser()->tab_strip_model()->GetActiveWebContents(),
[email protected]06bc5d92013-01-02 22:44:13130 "window.domAutomationController.send("
[email protected]b6987e02013-01-04 18:30:43131 " document.defaultView.getComputedStyle(document.body, null)."
132 " getPropertyValue('background-color') == 'rgb(245, 245, 220)')",
[email protected]9fabbf72010-09-30 21:50:05133 &result));
[email protected]919ddc82009-07-15 04:30:12134 EXPECT_EQ(expect_css, result);
135
136 result = false;
[email protected]b6987e02013-01-04 18:30:43137 ASSERT_TRUE(content::ExecuteScriptAndExtractBool(
[email protected]47ae23372013-01-29 01:50:48138 browser()->tab_strip_model()->GetActiveWebContents(),
[email protected]06bc5d92013-01-02 22:44:13139 "window.domAutomationController.send(document.title == 'Modified')",
[email protected]9fabbf72010-09-30 21:50:05140 &result));
[email protected]919ddc82009-07-15 04:30:12141 EXPECT_EQ(expect_script, result);
142 }
143
[email protected]650b2d52013-02-10 03:41:45144 base::FilePath preferences_file_;
145 base::FilePath extensions_dir_;
146 base::FilePath user_scripts_dir_;
[email protected]f0488f2f2009-07-01 05:25:22147 bool enable_extensions_;
[email protected]bfc93562011-12-14 19:40:04148 // Extensions to load from the command line.
[email protected]650b2d52013-02-10 03:41:45149 std::vector<base::FilePath::StringType> load_extensions_;
[email protected]e50013c32010-08-18 21:05:24150
151 int num_expected_extensions_;
[email protected]f0488f2f2009-07-01 05:25:22152};
153
154
155// ExtensionsStartupTest
156// Ensures that we can startup the browser with --enable-extensions and some
157// extensions installed and see them run and do basic things.
158
159class ExtensionsStartupTest : public ExtensionStartupTestBase {
160 public:
161 ExtensionsStartupTest() {
162 enable_extensions_ = true;
163 }
164};
165
[email protected]c5b2c6112010-06-11 00:02:14166IN_PROC_BROWSER_TEST_F(ExtensionsStartupTest, Test) {
[email protected]e50013c32010-08-18 21:05:24167 WaitForServicesToStart(num_expected_extensions_, true);
[email protected]919ddc82009-07-15 04:30:12168 TestInjection(true, true);
169}
[email protected]919ddc82009-07-15 04:30:12170
[email protected]b2f665f52010-07-02 01:32:09171// Sometimes times out on Mac. http://crbug.com/48151
172#if defined(OS_MACOSX)
173#define MAYBE_NoFileAccess DISABLED_NoFileAccess
174#else
175#define MAYBE_NoFileAccess NoFileAccess
176#endif
[email protected]05c82182010-06-24 17:49:08177// Tests that disallowing file access on an extension prevents it from injecting
178// script into a page with a file URL.
[email protected]b2f665f52010-07-02 01:32:09179IN_PROC_BROWSER_TEST_F(ExtensionsStartupTest, MAYBE_NoFileAccess) {
[email protected]e50013c32010-08-18 21:05:24180 WaitForServicesToStart(num_expected_extensions_, true);
[email protected]05c82182010-06-24 17:49:08181
[email protected]84df8332011-12-06 18:22:46182 // Keep a separate list of extensions for which to disable file access, since
183 // doing so reloads them.
[email protected]1c321ee2012-05-21 03:02:34184 std::vector<const extensions::Extension*> extension_list;
[email protected]84df8332011-12-06 18:22:46185
[email protected]06bdd2b2012-11-30 18:47:13186 ExtensionService* service = extensions::ExtensionSystem::Get(
187 browser()->profile())->extension_service();
[email protected]84df8332011-12-06 18:22:46188 for (ExtensionSet::const_iterator it = service->extensions()->begin();
189 it != service->extensions()->end(); ++it) {
[email protected]1d5e58b2013-01-31 08:41:40190 if ((*it)->location() == extensions::Manifest::COMPONENT)
[email protected]fe13bf62010-08-26 14:33:19191 continue;
[email protected]84df8332011-12-06 18:22:46192 if (service->AllowFileAccess(*it))
193 extension_list.push_back(*it);
194 }
195
196 for (size_t i = 0; i < extension_list.size(); ++i) {
[email protected]a7fe9112012-07-20 02:34:45197 content::WindowedNotificationObserver user_scripts_observer(
[email protected]84df8332011-12-06 18:22:46198 chrome::NOTIFICATION_USER_SCRIPTS_UPDATED,
199 content::NotificationService::AllSources());
200 service->SetAllowFileAccess(extension_list[i], false);
201 user_scripts_observer.Wait();
[email protected]05c82182010-06-24 17:49:08202 }
203
204 TestInjection(false, false);
205}
206
[email protected]919ddc82009-07-15 04:30:12207// ExtensionsLoadTest
208// Ensures that we can startup the browser with --load-extension and see them
209// run.
210
211class ExtensionsLoadTest : public ExtensionStartupTestBase {
212 public:
213 ExtensionsLoadTest() {
[email protected]d728e002010-12-08 04:46:23214 enable_extensions_ = true;
[email protected]650b2d52013-02-10 03:41:45215 base::FilePath one_extension_path;
[email protected]bfc93562011-12-14 19:40:04216 PathService::Get(chrome::DIR_TEST_DATA, &one_extension_path);
217 one_extension_path = one_extension_path
[email protected]919ddc82009-07-15 04:30:12218 .AppendASCII("extensions")
219 .AppendASCII("good")
220 .AppendASCII("Extensions")
221 .AppendASCII("behllobkkfkfnphdnhnkndlbkcpglgmj")
222 .AppendASCII("1.0.0.0");
[email protected]bfc93562011-12-14 19:40:04223 load_extensions_.push_back(one_extension_path.value());
[email protected]f0488f2f2009-07-01 05:25:22224 }
[email protected]919ddc82009-07-15 04:30:12225};
[email protected]f0488f2f2009-07-01 05:25:22226
[email protected]4b109052011-04-29 12:53:24227// Fails inconsistently on Linux x64. http://crbug.com/80961
[email protected]dc128022012-02-17 23:14:58228// TODO(dpapad): Has not failed since October 2011, let's reenable, monitor
229// and act accordingly.
230IN_PROC_BROWSER_TEST_F(ExtensionsLoadTest, Test) {
[email protected]d728e002010-12-08 04:46:23231 WaitForServicesToStart(1, true);
[email protected]919ddc82009-07-15 04:30:12232 TestInjection(true, true);
[email protected]f0488f2f2009-07-01 05:25:22233}
[email protected]bfc93562011-12-14 19:40:04234
235// ExtensionsLoadMultipleTest
236// Ensures that we can startup the browser with multiple extensions
237// via --load-extension=X1,X2,X3.
238class ExtensionsLoadMultipleTest : public ExtensionStartupTestBase {
239 public:
240 ExtensionsLoadMultipleTest() {
241 enable_extensions_ = true;
[email protected]650b2d52013-02-10 03:41:45242 base::FilePath one_extension_path;
[email protected]bfc93562011-12-14 19:40:04243 PathService::Get(chrome::DIR_TEST_DATA, &one_extension_path);
244 one_extension_path = one_extension_path
245 .AppendASCII("extensions")
246 .AppendASCII("good")
247 .AppendASCII("Extensions")
248 .AppendASCII("behllobkkfkfnphdnhnkndlbkcpglgmj")
249 .AppendASCII("1.0.0.0");
250 load_extensions_.push_back(one_extension_path.value());
251
[email protected]650b2d52013-02-10 03:41:45252 base::FilePath second_extension_path;
[email protected]bfc93562011-12-14 19:40:04253 PathService::Get(chrome::DIR_TEST_DATA, &second_extension_path);
254 second_extension_path = second_extension_path
255 .AppendASCII("extensions")
256 .AppendASCII("app");
257 load_extensions_.push_back(second_extension_path.value());
258
[email protected]650b2d52013-02-10 03:41:45259 base::FilePath third_extension_path;
[email protected]bfc93562011-12-14 19:40:04260 PathService::Get(chrome::DIR_TEST_DATA, &third_extension_path);
261 third_extension_path = third_extension_path
262 .AppendASCII("extensions")
263 .AppendASCII("app1");
264 load_extensions_.push_back(third_extension_path.value());
265
[email protected]650b2d52013-02-10 03:41:45266 base::FilePath fourth_extension_path;
[email protected]bfc93562011-12-14 19:40:04267 PathService::Get(chrome::DIR_TEST_DATA, &fourth_extension_path);
268 fourth_extension_path = fourth_extension_path
269 .AppendASCII("extensions")
270 .AppendASCII("app2");
271 load_extensions_.push_back(fourth_extension_path.value());
272 }
273};
274
275IN_PROC_BROWSER_TEST_F(ExtensionsLoadMultipleTest, Test) {
276 WaitForServicesToStart(4, true);
277 TestInjection(true, true);
278}