src/chrome/test/base/chrome_unit_test_suite.cc

   1 // Copyright 2013 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 "chrome/test/base/chrome_unit_test_suite.h"
   6
   7 #include "base/path_service.h"
   8 #include "base/process/process_handle.h"
   9 #include "base/strings/stringprintf.h"
  10 #include "chrome/browser/chrome_content_browser_client.h"
  11 #include "chrome/browser/omaha_query_params/chrome_omaha_query_params_delegate.h"
  12 #include "chrome/common/chrome_content_client.h"
  13 #include "chrome/common/chrome_paths.h"
  14 #include "chrome/test/base/testing_browser_process.h"
  15 #include "chrome/utility/chrome_content_utility_client.h"
  16 #include "components/component_updater/component_updater_paths.h"
  17 #include "components/omaha_query_params/omaha_query_params.h"
  18 #include "content/public/common/content_paths.h"
  19 #include "testing/gtest/include/gtest/gtest.h"
  20 #include "ui/base/resource/resource_bundle.h"
  21 #include "ui/base/resource/resource_handle.h"
  22 #include "ui/base/ui_base_paths.h"
  23
  24 #if defined(OS_CHROMEOS)
  25 #include "chromeos/chromeos_paths.h"
  26 #endif
  27
  28 #if !defined(OS_IOS)
  29 #include "chrome/browser/ui/webui/chrome_web_ui_controller_factory.h"
  30 #include "ui/gl/gl_surface.h"
  31 #endif
  32
  33 #if defined(OS_POSIX)
  34 #include "base/memory/shared_memory.h"
  35 #endif
  36
  37 #if defined(ENABLE_EXTENSIONS)
  38 #include "chrome/common/extensions/chrome_extensions_client.h"
  39 #include "extensions/common/extension_paths.h"
  40 #endif
  41
  42 namespace {
  43
  44 // Creates a TestingBrowserProcess for each test.
  45 class ChromeUnitTestSuiteInitializer : public testing::EmptyTestEventListener {
  46  public:
  47   ChromeUnitTestSuiteInitializer() {}
  48   virtual ~ChromeUnitTestSuiteInitializer() {}
  49
  50   virtual void OnTestStart(const testing::TestInfo& test_info) override {
  51     content_client_.reset(new ChromeContentClient);
  52     content::SetContentClient(content_client_.get());
  53     // TODO(ios): Bring this back once ChromeContentBrowserClient is building.
  54 #if !defined(OS_IOS)
  55     browser_content_client_.reset(new chrome::ChromeContentBrowserClient());
  56     content::SetBrowserClientForTesting(browser_content_client_.get());
  57     utility_content_client_.reset(new ChromeContentUtilityClient());
  58     content::SetUtilityClientForTesting(utility_content_client_.get());
  59 #endif
  60
  61     TestingBrowserProcess::CreateInstance();
  62   }
  63
  64   virtual void OnTestEnd(const testing::TestInfo& test_info) override {
  65     // TODO(ios): Bring this back once ChromeContentBrowserClient is building.
  66 #if !defined(OS_IOS)
  67     browser_content_client_.reset();
  68     utility_content_client_.reset();
  69 #endif
  70     content_client_.reset();
  71     content::SetContentClient(NULL);
  72
  73     TestingBrowserProcess::DeleteInstance();
  74   }
  75
  76  private:
  77   // Client implementations for the content module.
  78   scoped_ptr<ChromeContentClient> content_client_;
  79   // TODO(ios): Bring this back once ChromeContentBrowserClient is building.
  80 #if !defined(OS_IOS)
  81   scoped_ptr<chrome::ChromeContentBrowserClient> browser_content_client_;
  82   scoped_ptr<ChromeContentUtilityClient> utility_content_client_;
  83 #endif
  84
  85   DISALLOW_COPY_AND_ASSIGN(ChromeUnitTestSuiteInitializer);
  86 };
  87
  88 }  // namespace
  89
  90 ChromeUnitTestSuite::ChromeUnitTestSuite(int argc, char** argv)
  91     : ChromeTestSuite(argc, argv) {}
  92
  93 ChromeUnitTestSuite::~ChromeUnitTestSuite() {}
  94
  95 void ChromeUnitTestSuite::Initialize() {
  96   // Add an additional listener to do the extra initialization for unit tests.
  97   // It will be started before the base class listeners and ended after the
  98   // base class listeners.
  99   testing::TestEventListeners& listeners =
 100       testing::UnitTest::GetInstance()->listeners();
 101   listeners.Append(new ChromeUnitTestSuiteInitializer);
 102
 103   InitializeProviders();
 104   RegisterInProcessThreads();
 105
 106   ChromeTestSuite::Initialize();
 107
 108   // This needs to run after ChromeTestSuite::Initialize which calls content's
 109   // intialization which calls base's which initializes ICU.
 110   InitializeResourceBundle();
 111 }
 112
 113 void ChromeUnitTestSuite::Shutdown() {
 114   ResourceBundle::CleanupSharedInstance();
 115   ChromeTestSuite::Shutdown();
 116 }
 117
 118 void ChromeUnitTestSuite::InitializeProviders() {
 119   {
 120     ChromeContentClient content_client;
 121     RegisterContentSchemes(&content_client);
 122   }
 123
 124   chrome::RegisterPathProvider();
 125   content::RegisterPathProvider();
 126   ui::RegisterPathProvider();
 127   component_updater::RegisterPathProvider(chrome::DIR_USER_DATA);
 128
 129 #if defined(OS_CHROMEOS)
 130   chromeos::RegisterPathProvider();
 131 #endif
 132
 133 #if defined(ENABLE_EXTENSIONS)
 134   extensions::RegisterPathProvider();
 135
 136   extensions::ExtensionsClient::Set(
 137       extensions::ChromeExtensionsClient::GetInstance());
 138 #endif
 139
 140 #if !defined(OS_IOS)
 141   content::WebUIControllerFactory::RegisterFactory(
 142       ChromeWebUIControllerFactory::GetInstance());
 143
 144   gfx::GLSurface::InitializeOneOffForTests();
 145
 146   omaha_query_params::OmahaQueryParams::SetDelegate(
 147       ChromeOmahaQueryParamsDelegate::GetInstance());
 148 #endif
 149 }
 150
 151 void ChromeUnitTestSuite::InitializeResourceBundle() {
 152   // Force unittests to run using en-US so if we test against string
 153   // output, it'll pass regardless of the system language.
 154   ui::ResourceBundle::InitSharedInstanceWithLocale(
 155       "en-US", NULL, ui::ResourceBundle::LOAD_COMMON_RESOURCES);
 156   base::FilePath resources_pack_path;
 157 #if defined(OS_MACOSX) && !defined(OS_IOS)
 158   PathService::Get(base::DIR_MODULE, &resources_pack_path);
 159   resources_pack_path =
 160       resources_pack_path.Append(FILE_PATH_LITERAL("resources.pak"));
 161 #else
 162   PathService::Get(chrome::FILE_RESOURCES_PACK, &resources_pack_path);
 163 #endif
 164   ResourceBundle::GetSharedInstance().AddDataPackFromPath(
 165       resources_pack_path, ui::SCALE_FACTOR_NONE);
 166 }