Add a TestBrowserDialog helper class for testing browser dialogs in a consistent way.

chrome/browser/ui/test/browser_dialog_browsertest.cc

+// Copyright 2016 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 "base/command_line.h"
+#include "base/process/launch.h"
+#include "base/test/launcher/test_launcher.h"
+#include "base/test/test_switches.h"
+#include "base/test/test_timeouts.h"
+#include "chrome/browser/ui/test/test_browser_dialog.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "ui/compositor/compositor_switches.h"
+
+namespace {
+
+// Switch for BrowserDialogTest.Invoke to spawn a subprocess testing the
+// provided argument under a consistent setup.
+constexpr const char kDialogSwitch[] = "dialog";
+
+// Pattern to search in test names that indicate support for dialog testing.
+constexpr const char kDialogPattern[] = "InvokeDialog_";
+
+}  // namespace
+
+// Adds a browser_test entry point into the dialog testing framework. Without a
+// --dialog specified, just lists the available dialogs and exits.
+TEST(BrowserDialogTest, Invoke) {
+  const base::CommandLine& invoker = *base::CommandLine::ForCurrentProcess();
+  const std::string dialog_name = invoker.GetSwitchValueASCII(kDialogSwitch);
+
+  std::set<std::string> dialog_cases;
+  const testing::UnitTest* unit_test = testing::UnitTest::GetInstance();
+  for (int i = 0; i < unit_test->total_test_case_count(); ++i) {
+    const testing::TestCase* test_case = unit_test->GetTestCase(i);
+    for (int j = 0; j < test_case->total_test_count(); ++j) {
+      const char* name = test_case->GetTestInfo(j)->name();
+      if (strstr(name, kDialogPattern))
+        dialog_cases.insert(test_case->name() + std::string(".") + name);
+    }
+  }
+
+  if (dialog_name.empty()) {
+    std::string case_list;
+    for (const std::string& name : dialog_cases)
+      case_list += "\t" + name + "\n";
+    VLOG(0) << "\nPass one of the following after --" << kDialogSwitch << "=\n"
+            << case_list;
+    return;
+  }
+
+  auto it = dialog_cases.find(dialog_name);
+  ASSERT_NE(it, dialog_cases.end()) << "Dialog '" << dialog_name
+                                    << "' not found.";
+
+  base::CommandLine command(invoker);
+
+  // Replace TestBrowserDialog.Invoke with |dialog_name|.
+  command.AppendSwitchASCII(base::kGTestFilterFlag, dialog_name);
+
+  base::LaunchOptions options;
+
+  // Disable timeouts and generate screen output if --interactive was specified.
+  if (command.HasSwitch(internal::kInteractiveSwitch)) {
+    command.AppendSwitchASCII(switches::kUiTestActionMaxTimeout,
+                              TestTimeouts::kNoTimeoutSwitchValue);
+    command.AppendSwitchASCII(switches::kTestLauncherTimeout,
+                              TestTimeouts::kNoTimeoutSwitchValue);
+    command.AppendSwitch(switches::kEnablePixelOutputInTests);
+  } else {
+    options.wait = true;
+  }
+
+  base::LaunchProcess(command, options);
+}