src/chrome/test/ui/ui_test_suite.cc

   1 // Copyright (c) 2011 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/test/ui/ui_test_suite.h"
   6
   7 #include <string>
   8
   9 #include "base/command_line.h"
  10 #include "base/environment.h"
  11 #include "base/logging.h"
  12 #include "base/memory/scoped_ptr.h"
  13 #include "base/path_service.h"
  14 #include "base/process/kill.h"
  15 #include "base/process/launch.h"
  16 #include "base/process/process_iterator.h"
  17 #include "chrome/common/chrome_switches.h"
  18 #include "chrome/common/env_vars.h"
  19
  20 UITestSuite::UITestSuite(int argc, char** argv) : ChromeTestSuite(argc, argv) {
  21 #if defined(OS_WIN)
  22   crash_service_ = NULL;
  23 #endif
  24 }
  25
  26 void UITestSuite::Initialize() {
  27   ChromeTestSuite::Initialize();
  28 #if defined(OS_WIN)
  29   LoadCrashService();
  30 #endif
  31 }
  32
  33 void UITestSuite::Shutdown() {
  34 #if defined(OS_WIN)
  35   if (crash_service_)
  36     base::KillProcess(crash_service_, 0, false);
  37   job_handle_.Close();
  38 #endif
  39   ChromeTestSuite::Shutdown();
  40 }
  41
  42 #if defined(OS_WIN)
  43 void UITestSuite::LoadCrashService() {
  44   scoped_ptr<base::Environment> env(base::Environment::Create());
  45   if (env->HasVar(env_vars::kHeadless))
  46     return;
  47
  48   if (base::GetProcessCount(L"crash_service.exe", NULL))
  49     return;
  50
  51   job_handle_.Set(CreateJobObject(NULL, NULL));
  52   if (!job_handle_.IsValid()) {
  53     LOG(ERROR) << "Could not create JobObject.";
  54     return;
  55   }
  56
  57   if (!base::SetJobObjectLimitFlags(job_handle_.Get(),
  58                                     JOB_OBJECT_LIMIT_KILL_ON_JOB_CLOSE)) {
  59     LOG(ERROR) << "Could not SetJobObjectLimitFlags.";
  60     return;
  61   }
  62
  63   base::FilePath exe_dir;
  64   if (!PathService::Get(base::DIR_EXE, &exe_dir)) {
  65     LOG(ERROR) << "Failed to get path to DIR_EXE, "
  66                << "not starting crash_service.exe!";
  67     return;
  68   }
  69
  70   base::LaunchOptions launch_options;
  71   launch_options.job_handle = job_handle_.Get();
  72   base::FilePath crash_service = exe_dir.Append(L"crash_service.exe");
  73   if (!base::LaunchProcess(crash_service.value(), base::LaunchOptions(),
  74                            &crash_service_)) {
  75     LOG(ERROR) << "Couldn't start crash_service.exe, so this ui_tests run "
  76                << "won't tell you if any test crashes!";
  77     return;
  78   }
  79 }
  80 #endif