Add WidgetTest::GetAllWidgets() to find dialogs created by TestBrowserDialog.

chrome/browser/ui/test/test_browser_dialog.h

 // 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.
 
 #ifndef CHROME_BROWSER_UI_TEST_TEST_BROWSER_DIALOG_H_
 #define CHROME_BROWSER_UI_TEST_TEST_BROWSER_DIALOG_H_
 
 #include <string>
 #include <vector>
 
 #include "base/macros.h"
-#include "chrome/browser/ui/browser_window.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/gfx/native_widget_types.h"
 
 // TestBrowserDialog provides a way to register an InProcessBrowserTest testing
 // harness with a framework that invokes Chrome browser dialogs in a consistent
 // way. It optionally provides a way to invoke dialogs "interactively". This
 // allows screenshots to be generated easily, with the same test data, to assist
 // with UI review. It also provides a registry of dialogs so they can be
 // systematically checked for subtle changes and regressions.
@@ skipping 35 matching lines @@
 class TestBrowserDialog {
  protected:
   TestBrowserDialog();
 
   // Runs the dialog whose name corresponds to the current test case.
   void RunDialog();
 
   // Show the dialog corresponding to |name| and leave it open.
   virtual void ShowDialog(const std::string& name) = 0;
 
-  // The window that owns the dialogs. Used to find where the dialog appears.
-  virtual gfx::NativeWindow DialogParent() = 0;
-
  private:
   DISALLOW_COPY_AND_ASSIGN(TestBrowserDialog);
 };
 
 // Helper to mix in a TestBrowserDialog to an existing test harness. |Base|
 // must be a descendant of InProcessBrowserTest.
 template <class Base>
 class SupportsTestDialog : public Base, public TestBrowserDialog {
  protected:
   SupportsTestDialog() {}
 
-  // TestBrowserDialog:
-  gfx::NativeWindow DialogParent() override {
-    return this->browser()->window()->GetNativeWindow();
-  }
-
  private:
   DISALLOW_COPY_AND_ASSIGN(SupportsTestDialog);
 };
 
 using DialogBrowserTest = SupportsTestDialog<InProcessBrowserTest>;
 
 namespace internal {
 
 // When present on the command line, runs the test in an interactive mode.
 constexpr const char kInteractiveSwitch[] = "interactive";
 
 }  // namespace internal
 
 #endif  // CHROME_BROWSER_UI_TEST_TEST_BROWSER_DIALOG_H_