Adds testing infrastructure to web_ui_browsertest to support testing HtmlDialogs.

chrome/browser/ui/webui/web_ui_browsertest.cc

Index: chrome/browser/ui/webui/web_ui_browsertest.cc
diff --git a/chrome/browser/ui/webui/web_ui_browsertest.cc b/chrome/browser/ui/webui/web_ui_browsertest.cc
index 3130446c876a6f39270e088ac58650d5a2cddd17..2497b604d2b428d7c74d2a63fd46f928ec14e69c 100644
--- a/chrome/browser/ui/webui/web_ui_browsertest.cc
+++ b/chrome/browser/ui/webui/web_ui_browsertest.cc
@@ -207,7 +207,8 @@ const char WebUIBrowserTest::kDummyURL[] = "chrome://DummyURL";
 
 WebUIBrowserTest::WebUIBrowserTest()
     : test_handler_(new WebUITestHandler()),
-      libraries_preloaded_(false) {}
+      libraries_preloaded_(false),
+      web_ui_instance_(NULL) {}
 
 namespace {
 
@@ -254,6 +255,10 @@ void WebUIBrowserTest::TearDownInProcessBrowserTestFixture() {
   TestChromeWebUIFactory::RemoveFactoryOverride(GURL(kDummyURL).host());
 }
 
+void WebUIBrowserTest::SetWebUIInstance(WebUI* web_ui) {
+  web_ui_instance_ = web_ui;
+}
+
 WebUIMessageHandler* WebUIBrowserTest::GetMockMessageHandler() {
   return NULL;
 }
@@ -363,7 +368,7 @@ bool WebUIBrowserTest::RunJavascriptUsingHandler(
 }
 
 void WebUIBrowserTest::SetupHandlers() {
-  WebUI* web_ui_instance =
+  WebUI* web_ui_instance = web_ui_instance_ ? web_ui_instance_ :
       browser()->GetSelectedTabContents()->web_ui();
   ASSERT_TRUE(web_ui_instance != NULL);
   web_ui_instance->set_register_callback_overwrites(true);