Adding command line switch to run UI test in parallel.

chrome/test/ui/ui_test_suite.cc

 // 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();
@@ skipping 15 matching lines @@
       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);
   }
@@ skipping 37 matching lines @@
   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