Adds testing infrastructure to web_ui_browsertest to support testing HtmlDialogs.

chrome/browser/ui/webui/certificate_viewer_ui_browsertest.cc

Index: chrome/browser/ui/webui/certificate_viewer_ui_browsertest.cc
diff --git a/chrome/browser/ui/webui/certificate_viewer_ui_browsertest.cc b/chrome/browser/ui/webui/certificate_viewer_ui_browsertest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..b7554298820a6d8400af596bbb131df739093605
--- /dev/null
+++ b/chrome/browser/ui/webui/certificate_viewer_ui_browsertest.cc
@@ -0,0 +1,43 @@
+// Copyright (c) 2011 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 "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/browser_window.h"
+// #include "chrome/browser/ui/webui/chrome_web_ui.h"
+#include "chrome/browser/ui/webui/web_ui_browsertest.h"
+#include "chrome/browser/certificate_viewer.h"
+#include "chrome/common/url_constants.h"
+#include "chrome/test/base/ui_test_utils.h"
+#include "chrome/test/test_html_dialog_observer.h"
+#include "content/browser/renderer_host/render_view_host.h"
+#include "content/browser/tab_contents/tab_contents.h"
+#include "content/browser/webui/web_ui.h"
+#include "net/base/test_certificate_data.h"
+#include "net/base/x509_certificate.h"
+
+// Test framework for chrome/test/data/webui/certificate_viewer_dialog_test.js.
+class CertificateViewerUITest : public WebUIBrowserTest {
+ public:
+  virtual ~CertificateViewerUITest() {}
+
+  virtual void SetUpOnMainThread() {
+    // Need to enable more WebUI when runtime flag is in use.
+    // ChromeWebUI::SetMoreWebUI(true);
+
+    scoped_refptr<net::X509Certificate> google_cert(
+        net::X509Certificate::CreateFromBytes(
+            reinterpret_cast<const char*>(google_der), sizeof(google_der)));
+
+    DCHECK(browser());

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

If these are tests, they should be ASSERT_TRUE; not DCHECK, so the test runner
doesn't fail.

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

Done.

+    DCHECK(browser()->window());
+
+    TestHtmlDialogObserver dialog_observer;
+    ShowCertificateViewer(browser()->window()->GetNativeHandle(), google_cert);
+    WebUI* webui = dialog_observer.GetWebUI();
+    webui->tab_contents()->render_view_host()->SetWebUIProperty(
+        "expectedUrl", chrome::kChromeUICertificateViewerURL);
+    SetWebUIInstance(webui);
+    WebUIBrowserTest::SetUpOnMainThread();
+  }
+};