1 files changed, 163 insertions, 0 deletions
diff --git a/chrome/test/unit/chrome_test_suite.cc b/chrome/test/unit/chrome_test_suite.cc
new file mode 100644
index 0000000..d9d08cf
--- /dev/null
+++ b/chrome/test/unit/chrome_test_suite.cc
@@ -0,0 +1,163 @@
+// 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/test/unit/chrome_test_suite.h"
+
+#include "app/app_paths.h"
+#include "app/resource_bundle.h"
+#include "base/command_line.h"
+#include "base/process_util.h"
+#include "base/scoped_nsautorelease_pool.h"
+#include "base/stats_table.h"
+#include "base/utf_string_conversions.h"
+#include "chrome/browser/browser_process.h"
+#include "chrome/common/chrome_constants.h"
+#include "chrome/common/chrome_paths.h"
+#include "chrome/common/chrome_switches.h"
+#include "chrome/common/url_constants.h"
+#include "chrome/test/testing_browser_process.h"
+
+#if defined(OS_MACOSX)
+#include "base/mac_util.h"
+#endif
+
+#if defined(OS_POSIX)
+#include "base/shared_memory.h"
+#endif
+
+static void RemoveSharedMemoryFile(const std::string& filename) {
+  // Stats uses SharedMemory under the hood. On posix, this results in a file
+  // on disk.
+#if defined(OS_POSIX)
+  base::SharedMemory memory;
+  memory.Delete(UTF8ToWide(filename));
+#endif
+}
+
+WarningHostResolverProc::WarningHostResolverProc()
+    : HostResolverProc(NULL) {
+}
+
+WarningHostResolverProc::~WarningHostResolverProc() {
+}
+
+int WarningHostResolverProc::Resolve(const std::string& host,
+                                     net::AddressFamily address_family,
+                                     net::HostResolverFlags host_resolver_flags,
+                                     net::AddressList* addrlist,
+                                     int* os_error) {
+  const char* kLocalHostNames[] = {"localhost", "127.0.0.1"};
+  bool local = false;
+
+  if (host == net::GetHostName()) {
+    local = true;
+  } else {
+    for (size_t i = 0; i < arraysize(kLocalHostNames); i++)
+      if (host == kLocalHostNames[i]) {
+        local = true;
+        break;
+      }
+  }
+
+  // Make the test fail so it's harder to ignore.
+  // If you really need to make real DNS query, use
+  // net::RuleBasedHostResolverProc and its AllowDirectLookup method.
+  EXPECT_TRUE(local) << "Making external DNS lookup of " << host;
+
+  return ResolveUsingPrevious(host, address_family, host_resolver_flags,
+                              addrlist, os_error);
+}
+
+ChromeTestSuite::ChromeTestSuite(int argc, char** argv)
+    : base::TestSuite(argc, argv),
+      stats_table_(NULL),
+      created_user_data_dir_(false) {
+}
+
+ChromeTestSuite::~ChromeTestSuite() {
+}
+
+void ChromeTestSuite::Initialize() {
+  base::ScopedNSAutoreleasePool autorelease_pool;
+
+  base::TestSuite::Initialize();
+
+  chrome::RegisterChromeSchemes();
+  host_resolver_proc_ = new WarningHostResolverProc();
+  scoped_host_resolver_proc_.Init(host_resolver_proc_.get());
+
+  chrome::RegisterPathProvider();
+  app::RegisterPathProvider();
+  g_browser_process = new TestingBrowserProcess;
+
+  // Notice a user data override, and otherwise default to using a custom
+  // user data directory that lives alongside the current app.
+  // NOTE: The user data directory will be erased before each UI test that
+  //       uses it, in order to ensure consistency.
+  FilePath user_data_dir =
+      CommandLine::ForCurrentProcess()->GetSwitchValuePath(
+          switches::kUserDataDir);
+  if (user_data_dir.empty() &&
+      file_util::CreateNewTempDirectory(FILE_PATH_LITERAL("chrome_test_"),
+                                        &user_data_dir)) {
+    user_data_dir = user_data_dir.AppendASCII("test_user_data");
+    created_user_data_dir_ = true;
+  }
+  if (!user_data_dir.empty())
+    PathService::Override(chrome::DIR_USER_DATA, user_data_dir);
+
+  if (!browser_dir_.empty()) {
+    PathService::Override(base::DIR_EXE, browser_dir_);
+    PathService::Override(base::DIR_MODULE, browser_dir_);
+  }
+
+#if defined(OS_MACOSX)
+  // Look in the framework bundle for resources.
+  FilePath path;
+  PathService::Get(base::DIR_EXE, &path);
+  path = path.Append(chrome::kFrameworkName);
+  mac_util::SetOverrideAppBundlePath(path);
+#endif
+
+  // Force unittests to run using en-US so if we test against string
+  // output, it'll pass regardless of the system language.
+  ResourceBundle::InitSharedInstance("en-US");
+
+  // initialize the global StatsTable for unit_tests (make sure the file
+  // doesn't exist before opening it so the test gets a clean slate)
+  stats_filename_ = "unit_tests";
+  std::string pid_string = StringPrintf("-%d", base::GetCurrentProcId());
+  stats_filename_ += pid_string;
+  RemoveSharedMemoryFile(stats_filename_);
+  stats_table_ = new StatsTable(stats_filename_, 20, 200);
+  StatsTable::set_current(stats_table_);
+}
+
+void ChromeTestSuite::Shutdown() {
+  ResourceBundle::CleanupSharedInstance();
+
+#if defined(OS_MACOSX)
+  mac_util::SetOverrideAppBundle(NULL);
+#endif
+
+  delete g_browser_process;
+  g_browser_process = NULL;
+
+  // Tear down shared StatsTable; prevents unit_tests from leaking it.
+  StatsTable::set_current(NULL);
+  delete stats_table_;
+  RemoveSharedMemoryFile(stats_filename_);
+
+  // Delete the test_user_data dir recursively
+  // NOTE: user_data_dir will be deleted only if it was automatically
+  // created.
+  FilePath user_data_dir;
+  if (created_user_data_dir_ &&
+      PathService::Get(chrome::DIR_USER_DATA, &user_data_dir) &&
+      !user_data_dir.empty()) {
+    file_util::Delete(user_data_dir, true);
+    file_util::Delete(user_data_dir.DirName(), false);
+  }
+  base::TestSuite::Shutdown();
+}