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.
7 #include "chrome/browser/chrome_notification_types.h"
8 #include "chrome/browser/extensions/api/management/management_api.h"
9 #include "chrome/browser/extensions/extension_apitest.h"
10 #include "chrome/browser/extensions/extension_service.h"
11 #include "chrome/browser/extensions/extension_system.h"
12 #include "chrome/browser/extensions/extension_test_message_listener.h"
13 #include "chrome/browser/extensions/launch_util.h"
14 #include "chrome/browser/profiles/profile.h"
15 #include "chrome/browser/ui/browser.h"
16 #include "chrome/browser/ui/browser_commands.h"
17 #include "chrome/browser/ui/browser_finder.h"
18 #include "chrome/browser/ui/browser_iterator.h"
19 #include "chrome/browser/ui/tabs/tab_strip_model.h"
20 #include "chrome/common/chrome_switches.h"
21 #include "chrome/common/extensions/extension_constants.h"
22 #include "content/public/test/test_utils.h"
23 #include "extensions/browser/test_management_policy.h"
24 #include "extensions/common/manifest.h"
26 using extensions::Extension;
27 using extensions::Manifest;
31 // Find a browser other than |browser|.
32 Browser* FindOtherBrowser(Browser* browser) {
33 Browser* found = NULL;
34 for (chrome::BrowserIterator it; !it.done(); it.Next()) {
44 class ExtensionManagementApiTest : public ExtensionApiTest {
46 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
47 ExtensionApiTest::SetUpCommandLine(command_line);
48 command_line->AppendSwitch(switches::kEnablePanels);
51 virtual void LoadExtensions() {
52 base::FilePath basedir = test_data_dir_.AppendASCII("management");
54 // Load 5 enabled items.
55 LoadNamedExtension(basedir, "enabled_extension");
56 LoadNamedExtension(basedir, "enabled_app");
57 LoadNamedExtension(basedir, "description");
58 LoadNamedExtension(basedir, "permissions");
59 LoadNamedExtension(basedir, "short_name");
61 // Load 2 disabled items.
62 LoadNamedExtension(basedir, "disabled_extension");
63 DisableExtension(extension_ids_["disabled_extension"]);
64 LoadNamedExtension(basedir, "disabled_app");
65 DisableExtension(extension_ids_["disabled_app"]);
68 // Load an app, and wait for a message from app "management/launch_on_install"
69 // indicating that the new app has been launched.
70 void LoadAndWaitForLaunch(const std::string& app_path,
71 std::string* out_app_id) {
72 ExtensionTestMessageListener launched_app("launched app", false);
73 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII(app_path)));
76 *out_app_id = last_loaded_extension_id();
78 ASSERT_TRUE(launched_app.WaitUntilSatisfied());
82 void LoadNamedExtension(const base::FilePath& path,
83 const std::string& name) {
84 const Extension* extension = LoadExtension(path.AppendASCII(name));
85 ASSERT_TRUE(extension);
86 extension_ids_[name] = extension->id();
89 void InstallNamedExtension(const base::FilePath& path,
90 const std::string& name,
91 Manifest::Location install_source) {
92 const Extension* extension = InstallExtension(path.AppendASCII(name), 1,
94 ASSERT_TRUE(extension);
95 extension_ids_[name] = extension->id();
98 // Maps installed extension names to their IDs.
99 std::map<std::string, std::string> extension_ids_;
102 IN_PROC_BROWSER_TEST_F(ExtensionManagementApiTest, Basics) {
105 base::FilePath basedir = test_data_dir_.AppendASCII("management");
106 InstallNamedExtension(basedir, "internal_extension", Manifest::INTERNAL);
107 InstallNamedExtension(basedir, "external_extension",
108 Manifest::EXTERNAL_PREF);
109 InstallNamedExtension(basedir, "admin_extension",
110 Manifest::EXTERNAL_POLICY_DOWNLOAD);
112 ASSERT_TRUE(RunExtensionSubtest("management/test", "basics.html"));
115 IN_PROC_BROWSER_TEST_F(ExtensionManagementApiTest, NoPermission) {
117 ASSERT_TRUE(RunExtensionSubtest("management/no_permission", "test.html"));
120 // Disabled: http://crbug.com/174411
122 #define MAYBE_Uninstall DISABLED_Uninstall
124 #define MAYBE_Uninstall Uninstall
127 IN_PROC_BROWSER_TEST_F(ExtensionManagementApiTest, MAYBE_Uninstall) {
129 // Confirmation dialog will be shown for uninstallations except for self.
130 extensions::ManagementUninstallFunction::SetAutoConfirmForTest(true);
131 ASSERT_TRUE(RunExtensionSubtest("management/test", "uninstall.html"));
134 // Fails often on Windows dbg bots. http://crbug.com/177163
136 #define MAYBE_ManagementPolicyAllowed DISABLED_ManagementPolicyAllowed
138 #define MAYBE_ManagementPolicyAllowed ManagementPolicyAllowed
139 #endif // defined(OS_WIN)
140 // Tests actions on extensions when no management policy is in place.
141 IN_PROC_BROWSER_TEST_F(ExtensionManagementApiTest,
142 MAYBE_ManagementPolicyAllowed) {
144 extensions::ManagementUninstallFunction::SetAutoConfirmForTest(true);
145 ExtensionService* service = extensions::ExtensionSystem::Get(
146 browser()->profile())->extension_service();
147 EXPECT_TRUE(service->GetExtensionById(extension_ids_["enabled_extension"],
150 // Ensure that all actions are allowed.
151 extensions::ExtensionSystem::Get(
152 browser()->profile())->management_policy()->UnregisterAllProviders();
154 ASSERT_TRUE(RunExtensionSubtest("management/management_policy",
156 // The last thing the test does is uninstall the "enabled_extension".
157 EXPECT_FALSE(service->GetExtensionById(extension_ids_["enabled_extension"],
161 // Fails often on Windows dbg bots. http://crbug.com/177163
163 #define MAYBE_ManagementPolicyProhibited DISABLED_ManagementPolicyProhibited
165 #define MAYBE_ManagementPolicyProhibited ManagementPolicyProhibited
166 #endif // defined(OS_WIN)
167 // Tests actions on extensions when management policy prohibits those actions.
168 IN_PROC_BROWSER_TEST_F(ExtensionManagementApiTest,
169 MAYBE_ManagementPolicyProhibited) {
171 ExtensionService* service = extensions::ExtensionSystem::Get(
172 browser()->profile())->extension_service();
173 EXPECT_TRUE(service->GetExtensionById(extension_ids_["enabled_extension"],
176 // Prohibit status changes.
177 extensions::ManagementPolicy* policy = extensions::ExtensionSystem::Get(
178 browser()->profile())->management_policy();
179 policy->UnregisterAllProviders();
180 extensions::TestManagementPolicyProvider provider(
181 extensions::TestManagementPolicyProvider::PROHIBIT_MODIFY_STATUS);
182 policy->RegisterProvider(&provider);
183 ASSERT_TRUE(RunExtensionSubtest("management/management_policy",
187 // Disabled. See http://crbug.com/176023
188 IN_PROC_BROWSER_TEST_F(ExtensionManagementApiTest, DISABLED_LaunchPanelApp) {
189 ExtensionService* service = extensions::ExtensionSystem::Get(
190 browser()->profile())->extension_service();
192 // Load an extension that calls launchApp() on any app that gets
194 ExtensionTestMessageListener launcher_loaded("launcher loaded", false);
195 ASSERT_TRUE(LoadExtension(
196 test_data_dir_.AppendASCII("management/launch_on_install")));
197 ASSERT_TRUE(launcher_loaded.WaitUntilSatisfied());
199 // Load an app with app.launch.container = "panel".
201 LoadAndWaitForLaunch("management/launch_app_panel", &app_id);
202 ASSERT_FALSE(HasFatalFailure()); // Stop the test if any ASSERT failed.
204 // Find the app's browser. Check that it is a popup.
205 ASSERT_EQ(2u, chrome::GetBrowserCount(browser()->profile(),
206 browser()->host_desktop_type()));
207 Browser* app_browser = FindOtherBrowser(browser());
208 ASSERT_TRUE(app_browser->is_type_popup());
209 ASSERT_TRUE(app_browser->is_app());
211 // Close the app panel.
212 content::WindowedNotificationObserver signal(
213 chrome::NOTIFICATION_BROWSER_CLOSED,
214 content::Source<Browser>(app_browser));
216 chrome::CloseWindow(app_browser);
219 // Unload the extension.
220 UninstallExtension(app_id);
221 ASSERT_EQ(1u, chrome::GetBrowserCount(browser()->profile(),
222 browser()->host_desktop_type()));
223 ASSERT_FALSE(service->GetExtensionById(app_id, true));
225 // Set a pref indicating that the user wants to launch in a regular tab.
226 // This should be ignored, because panel apps always load in a popup.
227 extensions::SetLaunchType(service, app_id, extensions::LAUNCH_TYPE_REGULAR);
229 // Load the extension again.
230 std::string app_id_new;
231 LoadAndWaitForLaunch("management/launch_app_panel", &app_id_new);
232 ASSERT_FALSE(HasFatalFailure());
234 // If the ID changed, then the pref will not apply to the app.
235 ASSERT_EQ(app_id, app_id_new);
237 // Find the app's browser. Apps that should load in a panel ignore
238 // prefs, so we should still see the launch in a popup.
239 ASSERT_EQ(2u, chrome::GetBrowserCount(browser()->profile(),
240 browser()->host_desktop_type()));
241 app_browser = FindOtherBrowser(browser());
242 ASSERT_TRUE(app_browser->is_type_popup());
243 ASSERT_TRUE(app_browser->is_app());
246 // Disabled: http://crbug.com/230165
248 #define MAYBE_LaunchTabApp DISABLED_LaunchTabApp
250 #define MAYBE_LaunchTabApp LaunchTabApp
252 IN_PROC_BROWSER_TEST_F(ExtensionManagementApiTest, MAYBE_LaunchTabApp) {
253 ExtensionService* service = extensions::ExtensionSystem::Get(
254 browser()->profile())->extension_service();
256 // Load an extension that calls launchApp() on any app that gets
258 ExtensionTestMessageListener launcher_loaded("launcher loaded", false);
259 ASSERT_TRUE(LoadExtension(
260 test_data_dir_.AppendASCII("management/launch_on_install")));
261 ASSERT_TRUE(launcher_loaded.WaitUntilSatisfied());
263 // Code below assumes that the test starts with a single browser window
265 ASSERT_EQ(1u, chrome::GetBrowserCount(browser()->profile(),
266 browser()->host_desktop_type()));
267 ASSERT_EQ(1, browser()->tab_strip_model()->count());
269 // Load an app with app.launch.container = "tab".
271 LoadAndWaitForLaunch("management/launch_app_tab", &app_id);
272 ASSERT_FALSE(HasFatalFailure());
274 // Check that the app opened in a new tab of the existing browser.
275 ASSERT_EQ(1u, chrome::GetBrowserCount(browser()->profile(),
276 browser()->host_desktop_type()));
277 ASSERT_EQ(2, browser()->tab_strip_model()->count());
279 // Unload the extension.
280 UninstallExtension(app_id);
281 ASSERT_EQ(1u, chrome::GetBrowserCount(browser()->profile(),
282 browser()->host_desktop_type()));
283 ASSERT_FALSE(service->GetExtensionById(app_id, true));
285 // Set a pref indicating that the user wants to launch in a window.
286 extensions::SetLaunchType(service, app_id, extensions::LAUNCH_TYPE_WINDOW);
288 std::string app_id_new;
289 LoadAndWaitForLaunch("management/launch_app_tab", &app_id_new);
290 ASSERT_FALSE(HasFatalFailure());
292 // If the ID changed, then the pref will not apply to the app.
293 ASSERT_EQ(app_id, app_id_new);
295 #if defined(OS_MACOSX)
296 // App windows are not yet implemented on mac os. We should fall back
298 ASSERT_EQ(1u, chrome::GetBrowserCount(browser()->profile(),
299 browser()->host_desktop_type()));
300 ASSERT_EQ(2, browser()->tab_strip_model()->count());
302 // Find the app's browser. Opening in a new window will create
304 ASSERT_EQ(2u, chrome::GetBrowserCount(browser()->profile(),
305 browser()->host_desktop_type()));
306 Browser* app_browser = FindOtherBrowser(browser());
307 ASSERT_TRUE(app_browser->is_app());