| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/ui/browser.h" | 5 #include "chrome/browser/ui/browser.h" |
| 6 #include "chrome/browser/ui/browser_window.h" | 6 #include "chrome/browser/ui/browser_window.h" |
| 7 #include "chrome/browser/ui/webui/chrome_web_ui.h" | 7 #include "chrome/browser/ui/webui/chrome_web_ui.h" |
| 8 #include "chrome/browser/ui/webui/web_ui_browsertest.h" | 8 #include "chrome/browser/ui/webui/web_ui_browsertest.h" |
| 9 #include "chrome/browser/certificate_viewer.h" | 9 #include "chrome/browser/certificate_viewer.h" |
| 10 #include "chrome/common/url_constants.h" | 10 #include "chrome/common/url_constants.h" |
| 11 #include "chrome/test/base/ui_test_utils.h" | 11 #include "chrome/test/base/ui_test_utils.h" |
| 12 #include "chrome/test/base/test_html_dialog_observer.h" | 12 #include "chrome/test/base/test_html_dialog_observer.h" |
| 13 #include "content/browser/renderer_host/render_view_host.h" | 13 #include "content/browser/renderer_host/render_view_host.h" |
| 14 #include "content/browser/tab_contents/tab_contents.h" | 14 #include "content/browser/tab_contents/tab_contents.h" |
| 15 #include "content/browser/webui/web_ui.h" | 15 #include "content/browser/webui/web_ui.h" |
| 16 #include "net/base/test_certificate_data.h" | 16 #include "net/base/test_certificate_data.h" |
| 17 #include "net/base/x509_certificate.h" | 17 #include "net/base/x509_certificate.h" |
| 18 | 18 |
| 19 // Test framework for chrome/test/data/webui/certificate_viewer_dialog_test.js. | 19 // Test framework for chrome/test/data/webui/certificate_viewer_dialog_test.js. |
| 20 class CertificateViewerUITest : public WebUIBrowserTest { | 20 class CertificateViewerUITest : public WebUIBrowserTest { |
| 21 public: | 21 public: |
| 22 CertificateViewerUITest(); | 22 CertificateViewerUITest(); |
| 23 virtual ~CertificateViewerUITest(); | 23 virtual ~CertificateViewerUITest(); |
| 24 | 24 |
| 25 private: | 25 protected: |
| 26 virtual void SetUpOnMainThread() OVERRIDE; | 26 virtual void ShowCertificateViewer(); |
| 27 }; | 27 }; |
| 28 | 28 |
| 29 void CertificateViewerUITest::SetUpOnMainThread() { | 29 void CertificateViewerUITest::ShowCertificateViewer() { |
| 30 // Enable more WebUI to use WebUI certificate viewer. | 30 // Enable more WebUI to use WebUI certificate viewer. |
| 31 ChromeWebUI::OverrideMoreWebUI(true); | 31 ChromeWebUI::OverrideMoreWebUI(true); |
| 32 | 32 |
| 33 scoped_refptr<net::X509Certificate> google_cert( | 33 scoped_refptr<net::X509Certificate> google_cert( |
| 34 net::X509Certificate::CreateFromBytes( | 34 net::X509Certificate::CreateFromBytes( |
| 35 reinterpret_cast<const char*>(google_der), sizeof(google_der))); | 35 reinterpret_cast<const char*>(google_der), sizeof(google_der))); |
| 36 | 36 |
| 37 ASSERT_TRUE(browser()); | 37 ASSERT_TRUE(browser()); |
| 38 ASSERT_TRUE(browser()->window()); | 38 ASSERT_TRUE(browser()->window()); |
| 39 | 39 |
| 40 TestHtmlDialogObserver dialog_observer; | 40 TestHtmlDialogObserver dialog_observer(this); |
| 41 ShowCertificateViewer(browser()->window()->GetNativeHandle(), google_cert); | 41 ::ShowCertificateViewer(browser()->window()->GetNativeHandle(), google_cert); |
| 42 WebUI* webui = dialog_observer.GetWebUI(); | 42 WebUI* webui = dialog_observer.GetWebUI(); |
| 43 webui->tab_contents()->render_view_host()->SetWebUIProperty( | 43 webui->tab_contents()->render_view_host()->SetWebUIProperty( |
| 44 "expectedUrl", chrome::kChromeUICertificateViewerURL); | 44 "expectedUrl", chrome::kChromeUICertificateViewerURL); |
| 45 SetWebUIInstance(webui); | 45 SetWebUIInstance(webui); |
| 46 WebUIBrowserTest::SetUpOnMainThread(); | 46 WebUIBrowserTest::SetUpOnMainThread(); |
| 47 } | 47 } |
| OLD | NEW |