[Extensions] Allow shared resources between tests via _test_resources

chrome/browser/extensions/extension_browsertest.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 "chrome/browser/extensions/extension_browsertest.h"
 
 #include <stddef.h>
 
 #include <utility>
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/path_service.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
+#include "base/threading/sequenced_worker_pool.h"
 #include "build/build_config.h"
 #include "chrome/browser/extensions/browsertest_util.h"
 #include "chrome/browser/extensions/chrome_test_extension_loader.h"
 #include "chrome/browser/extensions/component_loader.h"
 #include "chrome/browser/extensions/crx_installer.h"
 #include "chrome/browser/extensions/extension_creator.h"
 #include "chrome/browser/extensions/extension_error_reporter.h"
 #include "chrome/browser/extensions/extension_install_prompt.h"
 #include "chrome/browser/extensions/extension_install_prompt_show_params.h"
 #include "chrome/browser/extensions/extension_service.h"
@@ skipping 21 matching lines @@
 #include "content/public/test/test_utils.h"
 #include "extensions/browser/extension_dialog_auto_confirm.h"
 #include "extensions/browser/extension_host.h"
 #include "extensions/browser/extension_prefs.h"
 #include "extensions/browser/extension_registry.h"
 #include "extensions/browser/extension_system.h"
 #include "extensions/browser/notification_types.h"
 #include "extensions/browser/uninstall_reason.h"
 #include "extensions/common/constants.h"
 #include "extensions/common/extension_set.h"
+#include "net/url_request/url_request_file_job.h"
 
 #if defined(OS_CHROMEOS)
 #include "chromeos/chromeos_switches.h"
 #endif
 
 using extensions::Extension;
 using extensions::ExtensionCreator;
 using extensions::ExtensionRegistry;
 using extensions::FeatureSwitch;
 using extensions::Manifest;
 using extensions::ScopedTestDialogAutoConfirm;
 
+namespace {
+
+// Maps all chrome-extension://<id>/_test_resources/foo requests to
+// chrome/test/data/extensions/foo. This is what allows us to share code between
+// tests without needing to duplicate files in each extension.
+net::URLRequestJob* ExtensionProtocolTestHandler(
+    const base::FilePath& test_dir_root,
+    net::URLRequest* request,
+    net::NetworkDelegate* network_delegate,
+    const base::FilePath& relative_path) {
+  // Only map paths that begin with _test_resources.
+  if (!base::FilePath(FILE_PATH_LITERAL("_test_resources"))
+           .IsParent(relative_path)) {
+    return nullptr;
+  }
+
+  // Replace _test_resources/foo with chrome/test/data/foo.
+  std::vector<base::FilePath::StringType> components;
+  relative_path.GetComponents(&components);
+  DCHECK_GT(components.size(), 1u);
+  base::FilePath resource_path = test_dir_root;
+  for (size_t i = 1u; i < components.size(); ++i)
+    resource_path = resource_path.Append(components[i]);
+
+  return new net::URLRequestFileJob(
+      request, network_delegate, resource_path,
+      content::BrowserThread::GetBlockingPool()
+          ->GetTaskRunnerWithShutdownBehavior(
+              base::SequencedWorkerPool::SKIP_ON_SHUTDOWN));
+}
+
+}  // namespace
+
 ExtensionBrowserTest::ExtensionBrowserTest()
     : loaded_(false),
       installed_(false),
 #if defined(OS_CHROMEOS)
       set_chromeos_user_(true),
 #endif
       // Default channel is STABLE but override with UNKNOWN so that unlaunched
       // or incomplete APIs can write tests.
       current_channel_(version_info::Channel::UNKNOWN),
       override_prompt_for_external_extensions_(
@@ skipping 63 matching lines @@
 }
 
 void ExtensionBrowserTest::SetUpOnMainThread() {
   InProcessBrowserTest::SetUpOnMainThread();
   observer_.reset(
       new extensions::ChromeExtensionTestNotificationObserver(browser()));
   if (extension_service()->updater()) {
     extension_service()->updater()->SetExtensionCacheForTesting(
         test_extension_cache_.get());
   }
+
+  // We don't use test_data_dir_ here because we want this to point to
+  // chrome/test/data/extensions, and subclasses have a nasty habit of altering
+  // the data dir in SetUpCommandLine().
+  base::FilePath test_root_path;
+  PathService::Get(chrome::DIR_TEST_DATA, &test_root_path);
+  test_root_path = test_root_path.AppendASCII("extensions");
+  test_protocol_handler_ =
+      base::Bind(&ExtensionProtocolTestHandler, test_root_path);
+  extensions::SetExtensionProtocolTestHandler(&test_protocol_handler_);
 }
 
 void ExtensionBrowserTest::TearDownOnMainThread() {
   ExtensionMessageBubbleFactory::set_override_for_tests(
       ExtensionMessageBubbleFactory::NO_OVERRIDE);
+  extensions::SetExtensionProtocolTestHandler(nullptr);
   InProcessBrowserTest::TearDownOnMainThread();
 }
 
 const Extension* ExtensionBrowserTest::LoadExtension(
     const base::FilePath& path) {
   return LoadExtensionWithFlags(path, kFlagEnableFileAccess);
 }
 
 const Extension* ExtensionBrowserTest::LoadExtensionIncognito(
     const base::FilePath& path) {
@@ skipping 380 matching lines @@
   return extensions::browsertest_util::ExecuteScriptInBackgroundPage(
       profile(), extension_id, script);
 }
 
 bool ExtensionBrowserTest::ExecuteScriptInBackgroundPageNoWait(
     const std::string& extension_id,
     const std::string& script) {
   return extensions::browsertest_util::ExecuteScriptInBackgroundPageNoWait(
       profile(), extension_id, script);
 }