ios/chrome/test/ios_chrome_unit_test_suite.cc

// Copyright 2015 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 "ios/chrome/test/ios_chrome_unit_test_suite.h"

#include "ios/chrome/browser/browser_state/browser_state_keyed_service_factories.h"
#include "ios/chrome/browser/chrome_paths.h"
#include "ios/chrome/test/testing_application_context.h"
#include "ios/public/test/test_chrome_browser_provider.h"
#include "ios/public/test/test_chrome_provider_initializer.h"
#include "ios/web/public/web_client.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/base/resource/resource_bundle.h"
#include "ui/base/ui_base_paths.h"
#include "url/url_util.h"

namespace {

class IOSChromeUnitTestSuiteInitializer
    : public testing::EmptyTestEventListener {
 public:
  IOSChromeUnitTestSuiteInitializer() {}
  ~IOSChromeUnitTestSuiteInitializer() override {}

  void OnTestStart(const testing::TestInfo& test_info) override {
    web_client_.reset(new web::WebClient());
    web::SetWebClient(web_client_.get());
    test_ios_chrome_provider_initializer_.reset(
        new ios::TestChromeProviderInitializer());
    application_context_.reset(new TestingApplicationContext);
  }

  void OnTestEnd(const testing::TestInfo& test_info) override {
    web_client_.reset();
    web::SetWebClient(nullptr);
    test_ios_chrome_provider_initializer_.reset();
    application_context_.reset();
  }

 private:
  scoped_ptr<web::WebClient> web_client_;
  scoped_ptr<ios::TestChromeProviderInitializer>
      test_ios_chrome_provider_initializer_;
  scoped_ptr<ApplicationContext> application_context_;
  DISALLOW_COPY_AND_ASSIGN(IOSChromeUnitTestSuiteInitializer);
};

}  // namespace

IOSChromeUnitTestSuite::IOSChromeUnitTestSuite(int argc, char** argv)
    : base::TestSuite(argc, argv) {
}

IOSChromeUnitTestSuite::~IOSChromeUnitTestSuite() {
}

void IOSChromeUnitTestSuite::Initialize() {
  // Add an additional listener to do the extra initialization for unit tests.
  // It will be started before the base class listeners and ended after the
  // base class listeners.
  testing::TestEventListeners& listeners =
      testing::UnitTest::GetInstance()->listeners();
  listeners.Append(new IOSChromeUnitTestSuiteInitializer);

  ui::ResourceBundle::InitSharedInstanceWithLocale(
      "en-US", nullptr, ui::ResourceBundle::LOAD_COMMON_RESOURCES);

  // Ensure that all BrowserStateKeyedServiceFactories are built before any
  // test is run so that the dependencies are correctly resolved.
  EnsureBrowserStateKeyedServiceFactoriesBuilt();

  ios::RegisterPathProvider();
  ui::RegisterPathProvider();

  {
    ios::TestChromeBrowserProvider provider;
    url::AddStandardScheme(provider.GetChromeUIScheme(),
                           url::SCHEME_WITHOUT_PORT);
  }

  base::TestSuite::Initialize();
}