extensions/test/extensions_unittests_main.cc

   1 // Copyright 2014 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 "base/base_paths.h"
   6 #include "base/bind.h"
   7 #include "base/macros.h"
   8 #include "base/path_service.h"
   9 #include "base/test/launcher/unit_test_launcher.h"
  10 #include "content/public/common/content_client.h"
  11 #include "content/public/test/content_test_suite_base.h"
  12 #include "content/public/test/unittest_test_suite.h"
  13 #include "extensions/common/constants.h"
  14 #include "extensions/common/extension_paths.h"
  15 #include "extensions/test/test_extensions_client.h"
  16 #include "third_party/mojo/src/mojo/edk/embedder/test_embedder.h"
  17 #include "ui/base/resource/resource_bundle.h"
  18 #include "ui/gl/gl_surface.h"
  19
  20 namespace {
  21
  22 // Content client that exists only to register chrome-extension:// scheme with
  23 // the url module.
  24 // TODO(jamescook): Should this be merged with ShellContentClient? Should this
  25 // be a persistent object available to tests?
  26 class ExtensionsContentClient : public content::ContentClient {
  27  public:
  28   ExtensionsContentClient() {}
  29   ~ExtensionsContentClient() override {}
  30
  31   // content::ContentClient overrides:
  32   void AddAdditionalSchemes(
  33       std::vector<std::string>* standard_schemes,
  34       std::vector<std::string>* savable_schemes) override {
  35     standard_schemes->push_back(extensions::kExtensionScheme);
  36     savable_schemes->push_back(extensions::kExtensionScheme);
  37     standard_schemes->push_back(extensions::kExtensionResourceScheme);
  38     savable_schemes->push_back(extensions::kExtensionResourceScheme);
  39   }
  40
  41  private:
  42   DISALLOW_COPY_AND_ASSIGN(ExtensionsContentClient);
  43 };
  44
  45 // The test suite for extensions_unittests.
  46 class ExtensionsTestSuite : public content::ContentTestSuiteBase {
  47  public:
  48   ExtensionsTestSuite(int argc, char** argv);
  49   ~ExtensionsTestSuite() override;
  50
  51  private:
  52   // base::TestSuite:
  53   void Initialize() override;
  54   void Shutdown() override;
  55
  56   scoped_ptr<extensions::TestExtensionsClient> client_;
  57
  58   DISALLOW_COPY_AND_ASSIGN(ExtensionsTestSuite);
  59 };
  60
  61 ExtensionsTestSuite::ExtensionsTestSuite(int argc, char** argv)
  62     : content::ContentTestSuiteBase(argc, argv) {}
  63
  64 ExtensionsTestSuite::~ExtensionsTestSuite() {}
  65
  66 void ExtensionsTestSuite::Initialize() {
  67   content::ContentTestSuiteBase::Initialize();
  68   gfx::GLSurface::InitializeOneOffForTests();
  69
  70   // Register the chrome-extension:// scheme via this circuitous path. Note
  71   // that this does not persistently set up a ContentClient; individual tests
  72   // must use content::SetContentClient().
  73   {
  74     ExtensionsContentClient content_client;
  75     RegisterContentSchemes(&content_client);
  76   }
  77   RegisterInProcessThreads();
  78
  79   extensions::RegisterPathProvider();
  80
  81   base::FilePath extensions_shell_and_test_pak_path;
  82   PathService::Get(base::DIR_MODULE, &extensions_shell_and_test_pak_path);
  83   ui::ResourceBundle::InitSharedInstanceWithPakPath(
  84       extensions_shell_and_test_pak_path.AppendASCII(
  85           "extensions_shell_and_test.pak"));
  86
  87   client_.reset(new extensions::TestExtensionsClient());
  88   extensions::ExtensionsClient::Set(client_.get());
  89 }
  90
  91 void ExtensionsTestSuite::Shutdown() {
  92   extensions::ExtensionsClient::Set(NULL);
  93   client_.reset();
  94
  95   ui::ResourceBundle::CleanupSharedInstance();
  96   content::ContentTestSuiteBase::Shutdown();
  97 }
  98
  99 }  // namespace
 100
 101 int main(int argc, char** argv) {
 102   content::UnitTestTestSuite test_suite(new ExtensionsTestSuite(argc, argv));
 103
 104   mojo::embedder::test::InitWithSimplePlatformSupport();
 105   return base::LaunchUnitTests(argc,
 106                                argv,
 107                                base::Bind(&content::UnitTestTestSuite::Run,
 108                                           base::Unretained(&test_suite)));
 109 }