Prevent duplicate content script injection defined in manifest.json (reland)

chrome/browser/extensions/content_script_apitest.cc

 // Copyright (c) 2012 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 <stddef.h>
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
@@ skipping 316 matching lines @@
     // could lead to spoofing). Unfortunately, host_resolver seems to have
     // problems with redirecting "chrome.google.com" to the test server, so we
     // can't use the real Webstore's URL. If this changes, we could clean this
     // up.
     command_line->AppendSwitchASCII(
         switches::kAppsGalleryURL,
         base::StringPrintf("http://%s", kWebstoreDomain));
   }
 };
 
+IN_PROC_BROWSER_TEST_F(ExtensionApiTest,
+                       ContentScriptDuplicateScriptInjection) {
+  host_resolver()->AddRule("maps.google.com", "127.0.0.1");
+  ASSERT_TRUE(StartEmbeddedTestServer());
+
+  GURL url(
+      base::StringPrintf("http://maps.google.com:%i/extensions/test_file.html",
+                         embedded_test_server()->port()));
+
+  ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII(
+      "content_scripts/duplicate_script_injection")));
+
+  ui_test_utils::NavigateToURL(browser(), url);
+
+  // Test that a script that matches two separate, yet overlapping match
+  // patterns is only injected once.
+  bool scripts_injected_once = false;
+  ASSERT_TRUE(content::ExecuteScriptAndExtractBool(
+      browser()->tab_strip_model()->GetActiveWebContents(),
+      "window.domAutomationController.send("
+      "document.getElementsByClassName('injected-once')"
+      ".length == 1)",
+      &scripts_injected_once));
+  ASSERT_TRUE(scripts_injected_once);
+
+  // Test that a script injected at two different load process times, document
+  // idle and document end, is injected exactly twice.
+  bool scripts_injected_twice = false;
+  ASSERT_TRUE(content::ExecuteScriptAndExtractBool(
+      browser()->tab_strip_model()->GetActiveWebContents(),
+      "window.domAutomationController.send("
+      "document.getElementsByClassName('injected-twice')"
+      ".length == 2)",
+      &scripts_injected_twice));
+  ASSERT_TRUE(scripts_injected_twice);
+}
+
 IN_PROC_BROWSER_TEST_F(ContentScriptCssInjectionTest,
                        ContentScriptInjectsStyles) {
   ASSERT_TRUE(StartEmbeddedTestServer());
   host_resolver()->AddRule(kWebstoreDomain, "127.0.0.1");
 
   ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("content_scripts")
                                           .AppendASCII("css_injection")));
 
   // CSS injection should be allowed on an aribitrary web page.
   GURL url =
@@ skipping 276 matching lines @@
   ExtensionTestMessageListener iframe_loaded_listener("iframe loaded", false);
   ExtensionTestMessageListener content_script_listener("script injected",
                                                        false);
   LoadExtension(data_dir.AppendASCII("script_a_com"));
   LoadExtension(data_dir.AppendASCII("background_page_iframe"));
   iframe_loaded_listener.WaitUntilSatisfied();
   EXPECT_FALSE(content_script_listener.was_satisfied());
 }
 
 }  // namespace extensions