1 // Copyright (c) 2011 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/extension_apitest.h" 7 #include "chrome/browser/extensions/extension_service.h" 8 #include "chrome/browser/extensions/extension_system.h" 9 #include "chrome/browser/extensions/extension_test_message_listener.h" 10 #include "chrome/browser/profiles/profile.h" 11 #include "chrome/browser/ui/browser.h" 12 #include "chrome/common/extensions/extension.h" 13 #include "chrome/test/base/test_switches.h" 14 #include "chrome/test/base/ui_test_utils.h" 15 #include "extensions/common/id_util.h" 16 17 const std::string kAllUrlsTarget = 18 "files/extensions/api_test/all_urls/index.html"; 19 20 typedef ExtensionApiTest AllUrlsApiTest; 21 22 #if defined(OS_WIN) && !defined(NDEBUG) 23 // http://crbug.com/174341 24 #define MAYBE_WhitelistedExtension DISABLED_WhitelistedExtension 25 #else 26 #define MAYBE_WhitelistedExtension WhitelistedExtension 27 #endif 28 IN_PROC_BROWSER_TEST_F(AllUrlsApiTest, MAYBE_WhitelistedExtension) { 29 #if defined(OS_WIN) && defined(USE_ASH) 30 // Disable this test in Metro+Ash for now (http://crbug.com/262796). 31 if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kAshBrowserTests)) 32 return; 33 #endif 34 35 // First setup the two extensions. 36 base::FilePath extension_dir1 = test_data_dir_.AppendASCII("all_urls") 37 .AppendASCII("content_script"); 38 base::FilePath extension_dir2 = test_data_dir_.AppendASCII("all_urls") 39 .AppendASCII("execute_script"); 40 41 // Then add the two extensions to the whitelist. 42 extensions::Extension::ScriptingWhitelist whitelist; 43 whitelist.push_back(extensions::id_util::GenerateIdForPath(extension_dir1)); 44 whitelist.push_back(extensions::id_util::GenerateIdForPath(extension_dir2)); 45 extensions::Extension::SetScriptingWhitelist(whitelist); 46 47 // Then load extensions. 48 ExtensionService* service = extensions::ExtensionSystem::Get( 49 browser()->profile())->extension_service(); 50 const size_t size_before = service->extensions()->size(); 51 ASSERT_TRUE(LoadExtension(extension_dir1)); 52 ASSERT_TRUE(LoadExtension(extension_dir2)); 53 EXPECT_EQ(size_before + 2, service->extensions()->size()); 54 55 std::string url; 56 57 // Now verify we run content scripts on chrome://newtab/. 58 url = "chrome://newtab/"; 59 ExtensionTestMessageListener listener1a("content script: " + url, false); 60 ExtensionTestMessageListener listener1b("execute: " + url, false); 61 ui_test_utils::NavigateToURL(browser(), GURL(url)); 62 ASSERT_TRUE(listener1a.WaitUntilSatisfied()); 63 ASSERT_TRUE(listener1b.WaitUntilSatisfied()); 64 65 // Now verify data: urls. 66 url = "data:text/html;charset=utf-8,<html>asdf</html>"; 67 ExtensionTestMessageListener listener2a("content script: " + url, false); 68 ExtensionTestMessageListener listener2b("execute: " + url, false); 69 ui_test_utils::NavigateToURL(browser(), GURL(url)); 70 ASSERT_TRUE(listener2a.WaitUntilSatisfied()); 71 ASSERT_TRUE(listener2b.WaitUntilSatisfied()); 72 73 // Now verify chrome://version/. 74 url = "chrome://version/"; 75 ExtensionTestMessageListener listener3a("content script: " + url, false); 76 ExtensionTestMessageListener listener3b("execute: " + url, false); 77 ui_test_utils::NavigateToURL(browser(), GURL(url)); 78 ASSERT_TRUE(listener3a.WaitUntilSatisfied()); 79 ASSERT_TRUE(listener3b.WaitUntilSatisfied()); 80 81 // Now verify about:blank. 82 url = "about:blank"; 83 ExtensionTestMessageListener listener4a("content script: " + url, false); 84 ExtensionTestMessageListener listener4b("execute: " + url, false); 85 ui_test_utils::NavigateToURL(browser(), GURL(url)); 86 ASSERT_TRUE(listener4a.WaitUntilSatisfied()); 87 ASSERT_TRUE(listener4b.WaitUntilSatisfied()); 88 89 // Now verify we can script a regular http page. 90 ASSERT_TRUE(test_server()->Start()); 91 GURL page_url = test_server()->GetURL(kAllUrlsTarget); 92 ExtensionTestMessageListener listener5a("content script: " + page_url.spec(), 93 false); 94 ExtensionTestMessageListener listener5b("execute: " + page_url.spec(), false); 95 ui_test_utils::NavigateToURL(browser(), page_url); 96 ASSERT_TRUE(listener5a.WaitUntilSatisfied()); 97 ASSERT_TRUE(listener5b.WaitUntilSatisfied()); 98 } 99 100 // Test that an extension NOT whitelisted for scripting can ask for <all_urls> 101 // and run scripts on non-restricted all pages. 102 IN_PROC_BROWSER_TEST_F(AllUrlsApiTest, RegularExtensions) { 103 // First load the two extensions. 104 base::FilePath extension_dir1 = test_data_dir_.AppendASCII("all_urls") 105 .AppendASCII("content_script"); 106 base::FilePath extension_dir2 = test_data_dir_.AppendASCII("all_urls") 107 .AppendASCII("execute_script"); 108 109 ExtensionService* service = extensions::ExtensionSystem::Get( 110 browser()->profile())->extension_service(); 111 const size_t size_before = service->extensions()->size(); 112 ASSERT_TRUE(LoadExtension(extension_dir1)); 113 ASSERT_TRUE(LoadExtension(extension_dir2)); 114 EXPECT_EQ(size_before + 2, service->extensions()->size()); 115 116 // Now verify we can script a regular http page. 117 ASSERT_TRUE(test_server()->Start()); 118 GURL page_url = test_server()->GetURL(kAllUrlsTarget); 119 ExtensionTestMessageListener listener1a("content script: " + page_url.spec(), 120 false); 121 ExtensionTestMessageListener listener1b("execute: " + page_url.spec(), false); 122 ui_test_utils::NavigateToURL(browser(), page_url); 123 ASSERT_TRUE(listener1a.WaitUntilSatisfied()); 124 ASSERT_TRUE(listener1b.WaitUntilSatisfied()); 125 } 126
