summaryrefslogtreecommitdiffstats
path: root/chrome/test/data/webui/certificate_viewer_ui_test-inl.h
blob: 10b9417fa6165a546d60c2b2961d8cbf4f5cf3c0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
// 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.

#include "build/build_config.h"
#include "chrome/browser/certificate_viewer.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "chrome/browser/ui/webui/certificate_viewer_webui.h"
#include "chrome/browser/ui/webui/constrained_web_dialog_ui.h"
#include "chrome/common/url_constants.h"
#include "chrome/test/base/ui_test_utils.h"
#include "chrome/test/base/web_ui_browser_test.h"
#include "content/public/browser/render_view_host.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_ui.h"
#include "content/public/test/browser_test_utils.h"
#include "net/cert/x509_certificate.h"
#include "net/test/test_certificate_data.h"

// Test framework for chrome/test/data/webui/certificate_viewer_dialog_test.js.
class CertificateViewerUITest : public WebUIBrowserTest {
 public:
  CertificateViewerUITest();
  ~CertificateViewerUITest() override;

 protected:
  void CreateCertViewerDialog();
  void ShowCertificateViewer();
#if defined(OS_CHROMEOS)
  void ShowModalCertificateViewer();
#endif
};


void CertificateViewerUITest::CreateCertViewerDialog() {

}

void CertificateViewerUITest::ShowCertificateViewer() {
  scoped_refptr<net::X509Certificate> google_cert(
      net::X509Certificate::CreateFromBytes(
          reinterpret_cast<const char*>(google_der), sizeof(google_der)));

  ASSERT_TRUE(browser());
  ASSERT_TRUE(browser()->window());

  CertificateViewerDialog* dialog =
      new CertificateViewerDialog(google_cert.get());
  dialog->Show(browser()->tab_strip_model()->GetActiveWebContents(),
               browser()->window()->GetNativeWindow());
  content::WebContents* webui_webcontents =
      dialog->GetWebUI()->GetWebContents();
  content::WaitForLoadStop(webui_webcontents);
  content::WebUI* webui = webui_webcontents->GetWebUI();
  webui_webcontents->GetRenderViewHost()->SetWebUIProperty(
      "expectedUrl", chrome::kChromeUICertificateViewerURL);
  SetWebUIInstance(webui);
}

#if defined(OS_CHROMEOS)
void CertificateViewerUITest::ShowModalCertificateViewer() {
  scoped_refptr<net::X509Certificate> google_cert(
      net::X509Certificate::CreateFromBytes(
          reinterpret_cast<const char*>(google_der), sizeof(google_der)));

  ASSERT_TRUE(browser());
  ASSERT_TRUE(browser()->window());

  CertificateViewerModalDialog* dialog =
      new CertificateViewerModalDialog(google_cert.get());
  dialog->Show(browser()->tab_strip_model()->GetActiveWebContents(),
               browser()->window()->GetNativeWindow());
  content::WebContents* webui_webcontents =
      dialog->GetWebUI()->GetWebContents();
  content::WaitForLoadStop(webui_webcontents);
  content::WebUI* webui = webui_webcontents->GetWebUI();
  webui_webcontents->GetRenderViewHost()->SetWebUIProperty(
      "expectedUrl", chrome::kChromeUICertificateViewerDialogURL);
  SetWebUIInstance(webui);
}
#endif