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

ui/aura/test/aura_test_helper.h

 // Copyright (c) 2012 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 UI_AURA_TEST_AURA_TEST_HELPER_H_
 #define UI_AURA_TEST_AURA_TEST_HELPER_H_
 
 #include <memory>
 
 #include "base/macros.h"
@@ skipping 33 matching lines @@
 class TestWindowParentingClient;
 
 // A helper class owned by tests that does common initialization required for
 // Aura use. This class creates a root window with clients and other objects
 // that are necessary to run test on Aura.
 class AuraTestHelper {
  public:
   explicit AuraTestHelper(base::MessageLoopForUI* message_loop);
   ~AuraTestHelper();
 
+  // Returns the current AuraTestHelper, or nullptr if it's not alive.
+  static AuraTestHelper* GetInstance();
+
   // Makes aura target mus with a mock WindowTree (TestWindowTree). Must be
   // called before SetUp().
   void EnableMusWithTestWindowTree(
       WindowTreeClientDelegate* window_tree_delegate,
       WindowManagerDelegate* window_manager_delegate);
 
   // Makes aura target mus with the specified WindowTreeClient. Must be called
   // before SetUp().
   void EnableMusWithWindowTreeClient(WindowTreeClient* window_tree_client);
 
@@ skipping 60 matching lines @@
 
   WindowTreeClient* window_tree_client_ = nullptr;
 
   DISALLOW_COPY_AND_ASSIGN(AuraTestHelper);
 };
 
 }  // namespace test
 }  // namespace aura
 
 #endif  // UI_AURA_TEST_AURA_TEST_HELPER_H_