chrome/test/base/chrome_test_suite.cc

   1 // Copyright (c) 2012 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_test_suite.h"
   6
   7 #if defined(OS_CHROMEOS)
   8 #include <stdio.h>
   9 #include <unistd.h>
  10 #endif
  11
  12 #include "base/command_line.h"
  13 #include "base/memory/ref_counted.h"
  14 #include "base/metrics/stats_table.h"
  15 #include "base/path_service.h"
  16 #include "base/strings/stringprintf.h"
  17 #include "base/strings/utf_string_conversions.h"
  18 #include "chrome/browser/browser_process.h"
  19 #include "chrome/browser/chrome_content_browser_client.h"
  20 #include "chrome/browser/ui/webui/chrome_web_ui_controller_factory.h"
  21 #include "chrome/common/chrome_constants.h"
  22 #include "chrome/common/chrome_content_client.h"
  23 #include "chrome/common/chrome_paths.h"
  24 #include "chrome/common/extensions/chrome_extensions_client.h"
  25 #include "chrome/common/url_constants.h"
  26 #include "chrome/utility/chrome_content_utility_client.h"
  27 #include "content/public/test/test_launcher.h"
  28 #include "extensions/common/extension_paths.h"
  29 #include "testing/gtest/include/gtest/gtest.h"
  30 #include "ui/base/resource/resource_bundle.h"
  31 #include "ui/base/resource/resource_handle.h"
  32
  33 #if defined(OS_ANDROID)
  34 #include "base/android/jni_android.h"
  35 #include "chrome/browser/android/chrome_jni_registrar.h"
  36 #include "net/android/net_jni_registrar.h"
  37 #include "ui/base/android/ui_base_jni_registrar.h"
  38 #include "ui/gfx/android/gfx_jni_registrar.h"
  39 #include "ui/gl/android/gl_jni_registrar.h"
  40 #endif
  41
  42 #if defined(OS_CHROMEOS)
  43 #include "base/process/process_metrics.h"
  44 #include "chromeos/chromeos_paths.h"
  45 #endif
  46
  47 #if defined(OS_MACOSX)
  48 #include "base/mac/bundle_locations.h"
  49 #include "base/mac/scoped_nsautorelease_pool.h"
  50 #if !defined(OS_IOS)
  51 #include "base/mac/mac_util.h"
  52 #include "chrome/browser/chrome_browser_application_mac.h"
  53 #endif  // !defined(OS_IOS)
  54 #endif
  55
  56 #if defined(OS_POSIX)
  57 #include "base/memory/shared_memory.h"
  58 #endif
  59
  60 namespace {
  61
  62 void RemoveSharedMemoryFile(const std::string& filename) {
  63   // Stats uses SharedMemory under the hood. On posix, this results in a file
  64   // on disk.
  65 #if defined(OS_POSIX)
  66   base::SharedMemory memory;
  67   memory.Delete(filename);
  68 #endif
  69 }
  70
  71 bool IsCrosPythonProcess() {
  72 #if defined(OS_CHROMEOS)
  73   char buf[80];
  74   int num_read = readlink(base::kProcSelfExe, buf, sizeof(buf) - 1);
  75   if (num_read == -1)
  76     return false;
  77   buf[num_read] = 0;
  78   const char kPythonPrefix[] = "/python";
  79   return !strncmp(strrchr(buf, '/'), kPythonPrefix, sizeof(kPythonPrefix) - 1);
  80 #else
  81   return false;
  82 #endif  // defined(OS_CHROMEOS)
  83 }
  84
  85 // Initializes services needed by both unit tests and browser tests.
  86 // See also ChromeUnitTestSuite for additional services created for unit tests.
  87 class ChromeTestSuiteInitializer : public testing::EmptyTestEventListener {
  88  public:
  89   ChromeTestSuiteInitializer() {
  90   }
  91
  92   virtual void OnTestStart(const testing::TestInfo& test_info) OVERRIDE {
  93     content_client_.reset(new ChromeContentClient);
  94     content::SetContentClient(content_client_.get());
  95     // TODO(ios): Bring this back once ChromeContentBrowserClient is building.
  96 #if !defined(OS_IOS)
  97     browser_content_client_.reset(new chrome::ChromeContentBrowserClient());
  98     content::SetBrowserClientForTesting(browser_content_client_.get());
  99     utility_content_client_.reset(new chrome::ChromeContentUtilityClient());
 100     content::SetUtilityClientForTesting(utility_content_client_.get());
 101 #endif
 102   }
 103
 104   virtual void OnTestEnd(const testing::TestInfo& test_info) OVERRIDE {
 105     // TODO(ios): Bring this back once ChromeContentBrowserClient is building.
 106 #if !defined(OS_IOS)
 107     browser_content_client_.reset();
 108     utility_content_client_.reset();
 109 #endif
 110     content_client_.reset();
 111     content::SetContentClient(NULL);
 112  }
 113
 114  private:
 115   // Client implementations for the content module.
 116   scoped_ptr<ChromeContentClient> content_client_;
 117   // TODO(ios): Bring this back once ChromeContentBrowserClient is building.
 118 #if !defined(OS_IOS)
 119   scoped_ptr<chrome::ChromeContentBrowserClient> browser_content_client_;
 120   scoped_ptr<chrome::ChromeContentUtilityClient> utility_content_client_;
 121 #endif
 122
 123   DISALLOW_COPY_AND_ASSIGN(ChromeTestSuiteInitializer);
 124 };
 125
 126 }  // namespace
 127
 128 ChromeTestSuite::ChromeTestSuite(int argc, char** argv)
 129     : content::ContentTestSuiteBase(argc, argv) {
 130 }
 131
 132 ChromeTestSuite::~ChromeTestSuite() {
 133 }
 134
 135 void ChromeTestSuite::Initialize() {
 136 #if defined(OS_MACOSX)
 137   base::mac::ScopedNSAutoreleasePool autorelease_pool;
 138 #if !defined(OS_IOS)
 139   chrome_browser_application_mac::RegisterBrowserCrApp();
 140 #endif  // !defined(OS_IOS)
 141 #endif
 142
 143 #if defined(OS_ANDROID)
 144   // Register JNI bindings for android.
 145   gfx::android::RegisterJni(base::android::AttachCurrentThread());
 146   net::android::RegisterJni(base::android::AttachCurrentThread());
 147   ui::android::RegisterJni(base::android::AttachCurrentThread());
 148   ui::gl::android::RegisterJni(base::android::AttachCurrentThread());
 149   chrome::android::RegisterJni(base::android::AttachCurrentThread());
 150 #endif
 151
 152   chrome::RegisterPathProvider();
 153 #if defined(OS_CHROMEOS)
 154   chromeos::RegisterPathProvider();
 155 #endif
 156   if (!browser_dir_.empty()) {
 157     PathService::Override(base::DIR_EXE, browser_dir_);
 158     PathService::Override(base::DIR_MODULE, browser_dir_);
 159   }
 160
 161 #if !defined(OS_IOS)
 162   extensions::RegisterPathProvider();
 163
 164   extensions::ExtensionsClient::Set(
 165       extensions::ChromeExtensionsClient::GetInstance());
 166
 167   // Only want to do this for unit tests.
 168   if (!content::GetCurrentTestLauncherDelegate()) {
 169     // For browser tests, this won't create the right object since
 170     // TestChromeWebUIControllerFactory is used. That's created and
 171     // registered in ChromeBrowserMainParts as in normal startup.
 172     content::WebUIControllerFactory::RegisterFactory(
 173         ChromeWebUIControllerFactory::GetInstance());
 174   }
 175 #endif
 176
 177   // Disable external libraries load if we are under python process in
 178   // ChromeOS.  That means we are autotest and, if ASAN is used,
 179   // external libraries load crashes.
 180   content::ContentTestSuiteBase::set_external_libraries_enabled(
 181       !IsCrosPythonProcess());
 182
 183   // Initialize after overriding paths as some content paths depend on correct
 184   // values for DIR_EXE and DIR_MODULE.
 185   content::ContentTestSuiteBase::Initialize();
 186
 187 #if defined(OS_MACOSX) && !defined(OS_IOS)
 188   // Look in the framework bundle for resources.
 189   base::FilePath path;
 190   PathService::Get(base::DIR_EXE, &path);
 191   path = path.Append(chrome::kFrameworkName);
 192   base::mac::SetOverrideFrameworkBundlePath(path);
 193 #endif
 194
 195   // Force unittests to run using en-US so if we test against string
 196   // output, it'll pass regardless of the system language.
 197   ResourceBundle::InitSharedInstanceWithLocale("en-US", NULL);
 198   base::FilePath resources_pack_path;
 199 #if defined(OS_MACOSX) && !defined(OS_IOS)
 200   PathService::Get(base::DIR_MODULE, &resources_pack_path);
 201   resources_pack_path =
 202       resources_pack_path.Append(FILE_PATH_LITERAL("resources.pak"));
 203 #else
 204   PathService::Get(chrome::FILE_RESOURCES_PACK, &resources_pack_path);
 205 #endif
 206   ResourceBundle::GetSharedInstance().AddDataPackFromPath(
 207       resources_pack_path, ui::SCALE_FACTOR_NONE);
 208
 209   stats_filename_ = base::StringPrintf("unit_tests-%d",
 210                                        base::GetCurrentProcId());
 211   RemoveSharedMemoryFile(stats_filename_);
 212   stats_table_.reset(new base::StatsTable(stats_filename_, 20, 200));
 213   base::StatsTable::set_current(stats_table_.get());
 214
 215   testing::TestEventListeners& listeners =
 216       testing::UnitTest::GetInstance()->listeners();
 217   listeners.Append(new ChromeTestSuiteInitializer);
 218 }
 219
 220 content::ContentClient* ChromeTestSuite::CreateClientForInitialization() {
 221   return new ChromeContentClient();
 222 }
 223
 224 void ChromeTestSuite::Shutdown() {
 225   ResourceBundle::CleanupSharedInstance();
 226
 227 #if defined(OS_MACOSX) && !defined(OS_IOS)
 228   base::mac::SetOverrideFrameworkBundle(NULL);
 229 #endif
 230
 231   base::StatsTable::set_current(NULL);
 232   stats_table_.reset();
 233   RemoveSharedMemoryFile(stats_filename_);
 234
 235   base::TestSuite::Shutdown();
 236 }