blob: 691a75fcb91ddbe612c27024cefccb040b892d84 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
// Copyright (c) 2006-2008 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 "chrome/test/ui/ui_test_suite.h"
#include "base/path_service.h"
#include "base/process_util.h"
#include "base/sys_info.h"
#include "chrome/common/env_vars.h"
// Force a test to use an already running browser instance. UI tests only.
const wchar_t UITestSuite::kUseExistingBrowser[] = L"use-existing-browser";
// Timeout for the test in milliseconds. UI tests only.
const wchar_t UITestSuite::kTestTimeout[] = L"test-timeout";
// Parameters to run test in parallel. UI tests only.
const wchar_t UITestSuite::kBatchCount[] = L"batch-count";
const wchar_t UITestSuite::kBatchIndex[] = L"batch-index";
const char UITestSuite::kGTestTotalShards[] = "GTEST_TOTAL_SHARDS=";
const char UITestSuite::kGTestShardIndex[] = "GTEST_SHARD_INDEX=";
UITestSuite::UITestSuite(int argc, char** argv)
: ChromeTestSuite(argc, argv) {
#if defined(OS_WIN)
crash_service_ = NULL;
#endif
}
void UITestSuite::Initialize() {
ChromeTestSuite::Initialize();
const CommandLine& parsed_command_line = *CommandLine::ForCurrentProcess();
UITest::set_in_process_renderer(
parsed_command_line.HasSwitch(switches::kSingleProcess));
UITest::set_no_sandbox(
parsed_command_line.HasSwitch(switches::kNoSandbox));
UITest::set_full_memory_dump(
parsed_command_line.HasSwitch(switches::kFullMemoryCrashReport));
UITest::set_safe_plugins(
parsed_command_line.HasSwitch(switches::kSafePlugins));
UITest::set_use_existing_browser(
parsed_command_line.HasSwitch(UITestSuite::kUseExistingBrowser));
UITest::set_dump_histograms_on_exit(
parsed_command_line.HasSwitch(switches::kDumpHistogramsOnExit));
UITest::set_enable_dcheck(
parsed_command_line.HasSwitch(switches::kEnableDCHECK));
UITest::set_silent_dump_on_dcheck(
parsed_command_line.HasSwitch(switches::kSilentDumpOnDCHECK));
UITest::set_disable_breakpad(
parsed_command_line.HasSwitch(switches::kDisableBreakpad));
std::wstring test_timeout =
parsed_command_line.GetSwitchValue(UITestSuite::kTestTimeout);
if (!test_timeout.empty()) {
UITest::set_test_timeout_ms(StringToInt(WideToUTF16Hack(test_timeout)));
}
#if defined(OS_WIN)
int batch_count = 0;
int batch_index = 0;
std::wstring batch_count_str =
parsed_command_line.GetSwitchValue(UITestSuite::kBatchCount);
if (!batch_count_str.empty()) {
batch_count = StringToInt(WideToUTF16Hack(batch_count_str));
}
std::wstring batch_index_str =
parsed_command_line.GetSwitchValue(UITestSuite::kBatchIndex);
if (!batch_index_str.empty()) {
batch_index = StringToInt(WideToUTF16Hack(batch_index_str));
}
if (batch_count > 0 && batch_index >= 0 && batch_index < batch_count) {
// Running UI test in parallel. Gtest supports running tests in shards,
// and every UI test instance is running with different user data dir.
// Thus all we need to do is to set up environment variables for gtest.
// See http://code.google.com/p/googletest/wiki/GoogleTestAdvancedGuide.
std::string batch_count_env(UITestSuite::kGTestTotalShards);
batch_count_env.append(WideToASCII(batch_count_str));
std::string batch_index_env(UITestSuite::kGTestShardIndex);
batch_index_env.append(WideToASCII(batch_index_str));
_putenv(batch_count_env.c_str());
_putenv(batch_index_env.c_str());
}
#endif
std::wstring js_flags =
parsed_command_line.GetSwitchValue(switches::kJavaScriptFlags);
if (!js_flags.empty()) {
UITest::set_js_flags(js_flags);
}
std::wstring log_level =
parsed_command_line.GetSwitchValue(switches::kLoggingLevel);
if (!log_level.empty()) {
UITest::set_log_level(log_level);
}
#if defined(OS_WIN)
LoadCrashService();
#endif
}
void UITestSuite::Shutdown() {
#if defined(OS_WIN)
if (crash_service_)
base::KillProcess(crash_service_, 0, false);
#endif
ChromeTestSuite::Shutdown();
}
void UITestSuite::SuppressErrorDialogs() {
#if defined(OS_WIN)
TestSuite::SuppressErrorDialogs();
#endif
UITest::set_show_error_dialogs(false);
}
#if defined(OS_WIN)
void UITestSuite::LoadCrashService() {
if (base::SysInfo::HasEnvVar(env_vars::kHeadless))
return;
if (base::GetProcessCount(L"crash_service.exe", NULL))
return;
FilePath exe_dir;
if (!PathService::Get(base::DIR_EXE, &exe_dir)) {
DCHECK(false);
return;
}
FilePath crash_service = exe_dir.Append(L"crash_service.exe");
if (!base::LaunchApp(crash_service.ToWStringHack(), false, false,
&crash_service_)) {
printf("Couldn't start crash_service.exe, so this ui_test run won't tell " \
"you if any test crashes!\n");
return;
}
printf("Started crash_service.exe so you know if a test crashes!\n");
}
#endif
|
