chrome_frame/test/reliability/run_all_unittests.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 "chrome_frame/test/reliability/reliability_test_suite.h"
   6
   7 #include "base/command_line.h"
   8 #include "base/process/kill.h"
   9 #include "chrome/common/chrome_paths.h"
  10 #include "chrome_frame/test/chrome_frame_test_utils.h"
  11 #include "chrome_frame/test_utils.h"
  12 #include "chrome_frame/utils.h"
  13
  14 static const char kRegisterDllFlag[] = "register";
  15
  16 int main(int argc, char **argv) {
  17   ScopedChromeFrameRegistrar::RegisterAndExitProcessIfDirected();
  18   // We create this slightly early as it is the one who instantiates THE
  19   // AtExitManager which some of the other stuff below relies on.
  20   ReliabilityTestSuite test_suite(argc, argv);
  21
  22   SetConfigBool(kChromeFrameHeadlessMode, true);
  23   base::ProcessHandle crash_service = chrome_frame_test::StartCrashService();
  24
  25   int result = -1;
  26   // If --register is passed, then we need to ensure that Chrome Frame is
  27   // registered before starting up the reliability tests.
  28   CommandLine* cmd_line = CommandLine::ForCurrentProcess();
  29   DCHECK(cmd_line);
  30   if (!cmd_line) {
  31     NOTREACHED() << "CommandLine object not initialized";
  32     return result;
  33   }
  34   if (cmd_line->HasSwitch(kRegisterDllFlag)) {
  35     std::wstring dll_path = cmd_line->GetSwitchValueNative(kRegisterDllFlag);
  36
  37     // Run() must be called within the scope of the ScopedChromeFrameRegistrar
  38     // to ensure that the correct DLL remains registered during the tests.
  39     ScopedChromeFrameRegistrar scoped_chrome_frame_registrar(
  40         dll_path, ScopedChromeFrameRegistrar::SYSTEM_LEVEL);
  41     result = test_suite.Run();
  42   } else {
  43     result = test_suite.Run();
  44   }
  45
  46   DeleteConfigValue(kChromeFrameHeadlessMode);
  47   if (crash_service)
  48     base::KillProcess(crash_service, 0, false);
  49
  50   return result;
  51 }
  52