chrome/test/perf/perf_ui_test_suite.cc

   1 // Copyright (c) 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/perf/perf_ui_test_suite.h"
   6
   7 #include "base/file_util.h"
   8 #include "base/json/json_file_value_serializer.h"
   9 #include "base/lazy_instance.h"
  10 #include "base/logging.h"
  11 #include "base/message_loop/message_loop.h"
  12 #include "base/path_service.h"
  13 #include "base/strings/string_util.h"
  14 #include "base/strings/utf_string_conversions.h"
  15 #include "base/threading/platform_thread.h"
  16 #include "chrome/browser/themes/browser_theme_pack.h"
  17 #include "chrome/common/chrome_constants.h"
  18 #include "chrome/common/chrome_paths.h"
  19 #include "chrome/test/perf/generate_profile.h"
  20 #include "content/public/test/test_browser_thread.h"
  21 #include "extensions/common/extension.h"
  22
  23 #if defined(OS_WIN)
  24 #include <atlbase.h>
  25 #include "ui/base/resource/resource_bundle_win.h"
  26 #endif
  27
  28 using content::BrowserThread;
  29 using extensions::Extension;
  30
  31 namespace {
  32
  33 const int kNumURLs = 20000;
  34
  35 const char kThemeExtension[] = "mblmlcbknbnfebdfjnolmcapmdofhmme";
  36
  37 base::LazyInstance<base::FilePath> g_default_profile_dir =
  38     LAZY_INSTANCE_INITIALIZER;
  39 base::LazyInstance<base::FilePath> g_complex_profile_dir =
  40     LAZY_INSTANCE_INITIALIZER;
  41
  42 }  // namespace
  43
  44 PerfUITestSuite::PerfUITestSuite(int argc, char** argv)
  45     : UITestSuite(argc, argv) {
  46   base::PlatformThread::SetName("Tests_Main");
  47 }
  48
  49 PerfUITestSuite::~PerfUITestSuite() {
  50 }
  51
  52 base::FilePath PerfUITestSuite::GetPathForProfileType(
  53     ProfileType profile_type) {
  54   switch (profile_type) {
  55     case DEFAULT_THEME:
  56       return g_default_profile_dir.Get();
  57     case COMPLEX_THEME:
  58       return g_complex_profile_dir.Get();
  59     default:
  60       NOTREACHED();
  61       return base::FilePath();
  62   }
  63 }
  64
  65 void PerfUITestSuite::Initialize() {
  66 #if defined(OS_WIN)
  67   // On Windows, make sure we've loaded the main chrome dll for its resource
  68   // before we let our parent class initialize the shared resource bundle
  69   // infrastructure. We have to do this manually because the extension system
  70   // uses several resources which, on Windows, are only compiled into the
  71   // browser DLL, but the base chrome testing stuff is used outside of browser.
  72   //
  73   // TODO(darin): Kill this once http://crbug.com/52609 is fixed.
  74   base::FilePath dll;
  75   PathService::Get(base::DIR_MODULE, &dll);
  76   dll = dll.Append(chrome::kBrowserResourcesDll);
  77   HMODULE res_mod = ::LoadLibraryExW(dll.value().c_str(),
  78       NULL, LOAD_LIBRARY_AS_DATAFILE | LOAD_LIBRARY_AS_IMAGE_RESOURCE);
  79   DCHECK(res_mod);
  80   _AtlBaseModule.SetResourceInstance(res_mod);
  81
  82   ui::SetResourcesDataDLL(_AtlBaseModule.GetResourceInstance());
  83 #endif
  84
  85   UITestSuite::Initialize();
  86
  87   if (!default_profile_dir_.CreateUniqueTempDir()) {
  88     LOG(FATAL) << "Failed to create default profile directory...";
  89   }
  90
  91   // Build a profile in default profile dir.
  92   base::FilePath default_path =
  93       default_profile_dir_.path().AppendASCII("Default");
  94   if (!GenerateProfile(TOP_SITES, kNumURLs, default_path)) {
  95     LOG(FATAL) << "Failed to generate default profile for tests...";
  96   }
  97
  98   g_default_profile_dir.Get() = default_profile_dir_.path();
  99
 100   if (!complex_profile_dir_.CreateUniqueTempDir()) {
 101     LOG(FATAL) << "Failed to create complex profile directory...";
 102   }
 103
 104   if (!base::CopyDirectory(default_path,
 105                                 complex_profile_dir_.path(),
 106                                 true)) {
 107     LOG(FATAL) << "Failed to copy data to complex profile directory...";
 108   }
 109
 110   // Copy the Extensions directory from the template into the
 111   // complex_profile_dir dir.
 112   base::FilePath base_data_dir;
 113   if (!PathService::Get(chrome::DIR_TEST_DATA, &base_data_dir))
 114     LOG(FATAL) << "Failed to fetch test data dir";
 115
 116   base_data_dir = base_data_dir.AppendASCII("profiles");
 117   base_data_dir = base_data_dir.AppendASCII("profile_with_complex_theme");
 118   base_data_dir = base_data_dir.AppendASCII("Default");
 119
 120   if (!base::CopyDirectory(base_data_dir, complex_profile_dir_.path(), true))
 121     LOG(FATAL) << "Failed to copy default to complex profile";
 122
 123   // Parse the manifest and make a temporary extension object because the
 124   // theme system takes extensions as input.
 125   base::FilePath extension_base =
 126       complex_profile_dir_.path()
 127       .AppendASCII("Default")
 128       .AppendASCII("Extensions")
 129       .AppendASCII(kThemeExtension)
 130       .AppendASCII("1.1");
 131   BuildCachedThemePakIn(extension_base);
 132
 133   g_complex_profile_dir.Get() = complex_profile_dir_.path();
 134 }
 135
 136 void PerfUITestSuite::BuildCachedThemePakIn(
 137     const base::FilePath& extension_base) {
 138   int error_code = 0;
 139   std::string error;
 140   JSONFileValueSerializer serializer(
 141       extension_base.AppendASCII("manifest.json"));
 142   scoped_ptr<base::DictionaryValue> valid_value(
 143       static_cast<base::DictionaryValue*>(
 144           serializer.Deserialize(&error_code, &error)));
 145   if (error_code != 0 || !valid_value)
 146     LOG(FATAL) << "Error parsing theme manifest: " << error;
 147
 148   scoped_refptr<Extension> extension =
 149       Extension::Create(extension_base,
 150                         extensions::Manifest::INVALID_LOCATION,
 151                         *valid_value,
 152                         Extension::NO_FLAGS,
 153                         &error);
 154   if (!extension.get())
 155     LOG(FATAL) << "Error loading theme extension: " << error;
 156
 157   // Build the "Cached Theme.pak" file in the template. (It's not committed
 158   // both because committing large binary files is bad in a git world and
 159   // because people don't remember to update it anyway, meaning we usually
 160   // have the theme rebuild penalty in our data.)
 161   base::MessageLoopForUI message_loop_;
 162   content::TestBrowserThread ui_thread_(BrowserThread::UI, &message_loop_);
 163   content::TestBrowserThread file_thread_(BrowserThread::FILE,
 164                                           &message_loop_);
 165
 166   scoped_refptr<BrowserThemePack> theme(
 167       BrowserThemePack::BuildFromExtension(extension.get()));
 168   if (!theme.get())
 169     LOG(FATAL) << "Failed to load theme from extension";
 170
 171   theme->WriteToDisk(extension_base.AppendASCII("Cached Theme.pak"));
 172 }