ash/test/test_suite.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 "ash/test/test_suite.h"
   6
   7 #include "ash/ash_switches.h"
   8 #include "base/command_line.h"
   9 #include "base/files/file_path.h"
  10 #include "build/build_config.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12 #include "ui/base/resource/resource_bundle.h"
  13 #include "ui/base/ui_base_paths.h"
  14 #include "ui/gfx/gfx_paths.h"
  15 #include "ui/gl/gl_surface.h"
  16
  17 #if defined(OS_WIN)
  18 #include "base/win/windows_version.h"
  19 #include "ui/base/win/atl_module.h"
  20 #include "win8/test/metro_registration_helper.h"
  21 #include "win8/test/test_registrar_constants.h"
  22 #endif
  23
  24 namespace ash {
  25 namespace test {
  26
  27 AuraShellTestSuite::AuraShellTestSuite(int argc, char** argv)
  28     : TestSuite(argc, argv) {}
  29
  30 void AuraShellTestSuite::Initialize() {
  31   base::TestSuite::Initialize();
  32   gfx::GLSurface::InitializeOneOffForTests();
  33
  34 #if defined(OS_WIN)
  35   base::win::Version version = base::win::GetVersion();
  36   // Although Ash officially is only supported for users on Win7+, we still run
  37   // ash_unittests on Vista builders, so we still need to initialize COM.
  38   if (version >= base::win::VERSION_VISTA &&
  39       !base::CommandLine::ForCurrentProcess()->HasSwitch(
  40           ash::switches::kForceAshToDesktop)) {
  41     com_initializer_.reset(new base::win::ScopedCOMInitializer());
  42     ui::win::CreateATLModuleIfNeeded();
  43     if (version >= base::win::VERSION_WIN8)
  44       ASSERT_TRUE(win8::MakeTestDefaultBrowserSynchronously());
  45   }
  46 #endif
  47
  48   gfx::RegisterPathProvider();
  49   ui::RegisterPathProvider();
  50
  51   // Force unittests to run using en-US so if we test against string
  52   // output, it'll pass regardless of the system language.
  53   ui::ResourceBundle::InitSharedInstanceWithLocale(
  54       "en-US", NULL, ui::ResourceBundle::LOAD_COMMON_RESOURCES);
  55
  56   base::DiscardableMemoryAllocator::SetInstance(&discardable_memory_allocator_);
  57 }
  58
  59 void AuraShellTestSuite::Shutdown() {
  60   ui::ResourceBundle::CleanupSharedInstance();
  61 #if defined(OS_WIN)
  62   com_initializer_.reset();
  63 #endif
  64   base::TestSuite::Shutdown();
  65 }
  66
  67 }  // namespace test
  68 }  // namespace ash