content/test/content_test_launcher.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 "content/public/test/test_launcher.h"
   6
   7 #include "base/base_paths.h"
   8 #include "base/command_line.h"
   9 #include "base/debug/stack_trace.h"
  10 #include "base/i18n/icu_util.h"
  11 #include "base/logging.h"
  12 #include "base/process/memory.h"
  13 #include "base/sys_info.h"
  14 #include "base/test/test_suite.h"
  15 #include "base/test/test_timeouts.h"
  16 #include "content/public/common/content_switches.h"
  17 #include "content/public/test/content_test_suite_base.h"
  18 #include "content/shell/app/shell_main_delegate.h"
  19 #include "content/shell/common/shell_switches.h"
  20 #include "media/base/media_switches.h"
  21 #include "testing/gtest/include/gtest/gtest.h"
  22
  23 #ifdef V8_USE_EXTERNAL_STARTUP_DATA
  24 #include "gin/v8_initializer.h"
  25 #endif
  26
  27 #if defined(OS_ANDROID)
  28 #include "base/message_loop/message_loop.h"
  29 #include "content/app/mojo/mojo_init.h"
  30 #include "content/common/url_schemes.h"
  31 #include "content/public/common/content_paths.h"
  32 #include "content/public/test/nested_message_pump_android.h"
  33 #include "content/shell/browser/shell_content_browser_client.h"
  34 #include "content/shell/common/shell_content_client.h"
  35 #include "ui/base/ui_base_paths.h"
  36 #endif
  37
  38 namespace content {
  39
  40 #if defined(OS_ANDROID)
  41 scoped_ptr<base::MessagePump> CreateMessagePumpForUI() {
  42   return scoped_ptr<base::MessagePump>(new NestedMessagePumpAndroid());
  43 };
  44 #endif
  45
  46 class ContentBrowserTestSuite : public ContentTestSuiteBase {
  47  public:
  48   ContentBrowserTestSuite(int argc, char** argv)
  49       : ContentTestSuiteBase(argc, argv) {
  50   }
  51   ~ContentBrowserTestSuite() override {}
  52
  53  protected:
  54   void Initialize() override {
  55 #if defined(OS_ANDROID)
  56     base::i18n::AllowMultipleInitializeCallsForTesting();
  57     base::i18n::InitializeICU();
  58
  59 #ifdef V8_USE_EXTERNAL_STARTUP_DATA
  60     gin::V8Initializer::LoadV8Snapshot();
  61     gin::V8Initializer::LoadV8Natives();
  62 #endif
  63
  64     // This needs to be done before base::TestSuite::Initialize() is called,
  65     // as it also tries to set MessagePumpForUIFactory.
  66     if (!base::MessageLoop::InitMessagePumpForUIFactory(
  67             &CreateMessagePumpForUI))
  68       VLOG(0) << "MessagePumpForUIFactory already set, unable to override.";
  69
  70     // For all other platforms, we call ContentMain for browser tests which goes
  71     // through the normal browser initialization paths. For Android, we must set
  72     // things up manually.
  73     content_client_.reset(new ShellContentClient);
  74     browser_content_client_.reset(new ShellContentBrowserClient());
  75     SetContentClient(content_client_.get());
  76     SetBrowserClientForTesting(browser_content_client_.get());
  77
  78     content::RegisterContentSchemes(false);
  79     RegisterPathProvider();
  80     ui::RegisterPathProvider();
  81     RegisterInProcessThreads();
  82
  83     InitializeMojo();
  84 #endif
  85
  86     ContentTestSuiteBase::Initialize();
  87   }
  88
  89 #if defined(OS_ANDROID)
  90   scoped_ptr<ShellContentClient> content_client_;
  91   scoped_ptr<ShellContentBrowserClient> browser_content_client_;
  92 #endif
  93
  94   DISALLOW_COPY_AND_ASSIGN(ContentBrowserTestSuite);
  95 };
  96
  97 class ContentTestLauncherDelegate : public TestLauncherDelegate {
  98  public:
  99   ContentTestLauncherDelegate() {}
 100   ~ContentTestLauncherDelegate() override {}
 101
 102   int RunTestSuite(int argc, char** argv) override {
 103     return ContentBrowserTestSuite(argc, argv).Run();
 104   }
 105
 106   bool AdjustChildProcessCommandLine(
 107       base::CommandLine* command_line,
 108       const base::FilePath& temp_data_dir) override {
 109     command_line->AppendSwitchPath(switches::kContentShellDataPath,
 110                                    temp_data_dir);
 111     command_line->AppendSwitch(switches::kUseFakeDeviceForMediaStream);
 112     command_line->AppendSwitch(switches::kUseFakeUIForMediaStream);
 113     return true;
 114   }
 115
 116  protected:
 117   ContentMainDelegate* CreateContentMainDelegate() override {
 118     return new ShellMainDelegate();
 119   }
 120
 121  private:
 122   DISALLOW_COPY_AND_ASSIGN(ContentTestLauncherDelegate);
 123 };
 124
 125 }  // namespace content
 126
 127 int main(int argc, char** argv) {
 128   int default_jobs = std::max(1, base::SysInfo::NumberOfProcessors() / 2);
 129   content::ContentTestLauncherDelegate launcher_delegate;
 130   return LaunchTests(&launcher_delegate, default_jobs, argc, argv);
 131 }