1 files changed, 112 insertions, 0 deletions
diff --git a/chrome/browser/extensions/all_urls_apitest.cc b/chrome/browser/extensions/all_urls_apitest.cc
new file mode 100644
index 0000000..067eef0
--- /dev/null
+++ b/chrome/browser/extensions/all_urls_apitest.cc
@@ -0,0 +1,112 @@
+// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+
+#include "chrome/browser/browser.h"
+#include "chrome/browser/extensions/extension_apitest.h"
+#include "chrome/browser/extensions/extension_test_message_listener.h"
+#include "chrome/browser/extensions/extensions_service.h"
+#include "chrome/browser/profile.h"
+#include "chrome/common/extensions/extension.h"
+#include "chrome/test/ui_test_utils.h"
+
+const std::string kAllUrlsTarget =
+    "files/extensions/uitests/component_all_urls/index.html";
+
+typedef ExtensionApiTest AllUrlsApiTest;
+
+#if !defined(OS_WIN)
+// Disabled, see http://crbug.com/57694.
+#define WhitelistedExtension DISABLED_WhitelistedExtension
+#endif
+
+IN_PROC_BROWSER_TEST_F(AllUrlsApiTest, WhitelistedExtension) {
+  // First add the two extensions we are going to load to the whitelist.
+  const char* kCanExecuteScriptsEverywhere[] = {
+    "fekpfaahmgnelcjpkefdnpiofglcgmgo",
+    "bpkfbiacjfimfmglhncgmibnddpnhmoj",
+  };
+  Extension::SetScriptingWhitelist(kCanExecuteScriptsEverywhere,
+                                   arraysize(kCanExecuteScriptsEverywhere));
+
+  // Then load the two extension.
+  FilePath extension_dir1 = test_data_dir_.AppendASCII("all_urls")
+                                          .AppendASCII("content_script");
+  FilePath extension_dir2 = test_data_dir_.AppendASCII("all_urls")
+                                          .AppendASCII("execute_script");
+
+  ExtensionsService* service = browser()->profile()->GetExtensionsService();
+  const size_t size_before = service->extensions()->size();
+  ASSERT_TRUE(LoadExtension(extension_dir1));
+  ASSERT_TRUE(LoadExtension(extension_dir2));
+  EXPECT_EQ(size_before + 2, service->extensions()->size());
+
+  std::string url;
+
+  // Now verify we run content scripts on chrome://newtab/.
+  url = "chrome://newtab/";
+  ExtensionTestMessageListener listener1a("content script: " + url);
+  ExtensionTestMessageListener listener1b("execute: " + url);
+  ui_test_utils::NavigateToURL(browser(), GURL(url));
+  ASSERT_TRUE(listener1a.WaitUntilSatisfied());
+  ASSERT_TRUE(listener1b.WaitUntilSatisfied());
+
+  // Now verify data: urls.
+  url = "data:text/html;charset=utf-8,<html>asdf</html>";
+  ExtensionTestMessageListener listener2a("content script: " + url);
+  ExtensionTestMessageListener listener2b("execute: " + url);
+  ui_test_utils::NavigateToURL(browser(), GURL(url));
+  ASSERT_TRUE(listener2a.WaitUntilSatisfied());
+  ASSERT_TRUE(listener2b.WaitUntilSatisfied());
+
+  // Now verify about:version.
+  url = "about:version";
+  ExtensionTestMessageListener listener3a("content script: " + url);
+  ExtensionTestMessageListener listener3b("execute: " + url);
+  ui_test_utils::NavigateToURL(browser(), GURL(url));
+  ASSERT_TRUE(listener3a.WaitUntilSatisfied());
+  ASSERT_TRUE(listener3b.WaitUntilSatisfied());
+
+  // Now verify about:blank.
+  url = "about:blank";
+  ExtensionTestMessageListener listener4a("content script: " + url);
+  ExtensionTestMessageListener listener4b("execute: " + url);
+  ui_test_utils::NavigateToURL(browser(), GURL(url));
+  ASSERT_TRUE(listener4a.WaitUntilSatisfied());
+  ASSERT_TRUE(listener4b.WaitUntilSatisfied());
+
+  // Now verify we can script a regular http page.
+  ASSERT_TRUE(test_server()->Start());
+  GURL page_url = test_server()->GetURL(kAllUrlsTarget);
+  ExtensionTestMessageListener listener5a("content script: " + page_url.spec());
+  ExtensionTestMessageListener listener5b("execute: " + page_url.spec());
+  ui_test_utils::NavigateToURL(browser(), page_url);
+  ASSERT_TRUE(listener5a.WaitUntilSatisfied());
+  ASSERT_TRUE(listener5b.WaitUntilSatisfied());
+}
+
+// Test that an extension NOT whitelisted for scripting can ask for <all_urls>
+// and run scripts on non-restricted all pages.
+IN_PROC_BROWSER_TEST_F(AllUrlsApiTest, RegularExtensions) {
+  // First load the two extension.
+  FilePath extension_dir1 = test_data_dir_.AppendASCII("all_urls")
+                                          .AppendASCII("content_script");
+  FilePath extension_dir2 = test_data_dir_.AppendASCII("all_urls")
+                                          .AppendASCII("execute_script");
+
+  ExtensionsService* service = browser()->profile()->GetExtensionsService();
+  const size_t size_before = service->extensions()->size();
+  ASSERT_TRUE(LoadExtension(extension_dir1));
+  ASSERT_TRUE(LoadExtension(extension_dir2));
+  EXPECT_EQ(size_before + 2, service->extensions()->size());
+
+  // Now verify we can script a regular http page.
+  ASSERT_TRUE(test_server()->Start());
+  GURL page_url = test_server()->GetURL(kAllUrlsTarget);
+  ExtensionTestMessageListener listener1a("content script: " + page_url.spec());
+  ExtensionTestMessageListener listener1b("execute: " + page_url.spec());
+  ui_test_utils::NavigateToURL(browser(), page_url);
+  ASSERT_TRUE(listener1a.WaitUntilSatisfied());
+  ASSERT_TRUE(listener1b.WaitUntilSatisfied());
+}