Adds testing infrastructure to web_ui_browsertest to support testing HtmlDialogs.

chrome/browser/ui/webui/web_ui_browsertest.h

Index: chrome/browser/ui/webui/web_ui_browsertest.h
diff --git a/chrome/browser/ui/webui/web_ui_browsertest.h b/chrome/browser/ui/webui/web_ui_browsertest.h
index 50876783ec58f35f14291ee6a767b5da6e292ee6..267ecfeb1972ac65b5a3504504042b062bd3a2a9 100644
--- a/chrome/browser/ui/webui/web_ui_browsertest.h
+++ b/chrome/browser/ui/webui/web_ui_browsertest.h
@@ -76,7 +76,7 @@ class WebUIBrowserTest
                          const ConstValueVector& test_arguments);
 
   // Runs a test that may include calls to functions in test_api.js, and waits
-  // for call to asyncTestDone().  Takes ownership of Value arguments.
+  // for call to testDone().  Takes ownership of Value arguments.
   bool RunJavascriptAsyncTest(const std::string& test_name);
   bool RunJavascriptAsyncTest(const std::string& test_name,
                               base::Value* arg);
@@ -125,6 +125,9 @@ class WebUIBrowserTest
   // Tear down override for |kDummyURL|.
   virtual void TearDownInProcessBrowserTestFixture() OVERRIDE;
 
+  // Set a WebUI instance to run tests on.
+  virtual void SetWebUIInstance(WebUI* web_ui);

[Sheridan Rawlins, 2011/09/15 19:11:41]

Do we need virtual here?

[flackr, 2011/09/15 21:02:37]

Unlikely. Done.

+
   // Returns a mock WebUI object under test (if any).
   virtual WebUIMessageHandler* GetMockMessageHandler();
 
@@ -178,6 +181,9 @@ class WebUIBrowserTest
   // PreloadJavascriptLibraries().
   std::string preload_test_fixture_;
   std::string preload_test_name_;
+
+  // The WebUI instance used for testing.
+  WebUI* web_ui_instance_;
 };
 
 #endif  // CHROME_BROWSER_UI_WEBUI_WEB_UI_BROWSERTEST_H_