chrome/browser/ui/views/certificate_selector_browsertest.cc

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/bind.h"
   6 #include "base/files/file_path.h"
   7 #include "base/strings/utf_string_conversions.h"
   8 #include "chrome/browser/ui/browser.h"
   9 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  10 #include "chrome/browser/ui/views/certificate_selector.h"
  11 #include "chrome/test/base/in_process_browser_test.h"
  12 #include "chrome/test/base/interactive_test_utils.h"
  13 #include "content/public/test/browser_test_utils.h"
  14 #include "net/base/test_data_directory.h"
  15 #include "net/cert/x509_certificate.h"
  16 #include "net/test/cert_test_util.h"
  17 #include "testing/gtest/include/gtest/gtest.h"
  18
  19 namespace {
  20
  21 class TestCertificateSelector : public chrome::CertificateSelector {
  22  public:
  23   TestCertificateSelector(const net::CertificateList& certificates,
  24                           content::WebContents* web_contents)
  25       : CertificateSelector(certificates, web_contents) {}
  26
  27   void Init() { InitWithText(base::ASCIIToUTF16("some arbitrary text")); }
  28
  29  private:
  30   DISALLOW_COPY_AND_ASSIGN(TestCertificateSelector);
  31 };
  32
  33 class CertificateSelectorTest : public InProcessBrowserTest {
  34  public:
  35   void SetUpInProcessBrowserTestFixture() override {
  36     client_1_ =
  37         net::ImportCertFromFile(net::GetTestCertsDirectory(), "client_1.pem");
  38     ASSERT_NE(nullptr, client_1_);
  39
  40     client_2_ =
  41         net::ImportCertFromFile(net::GetTestCertsDirectory(), "client_2.pem");
  42     ASSERT_NE(nullptr, client_2_);
  43   }
  44
  45   void SetUpOnMainThread() override {
  46     ASSERT_TRUE(content::WaitForLoadStop(
  47         browser()->tab_strip_model()->GetActiveWebContents()));
  48
  49     net::CertificateList certificates;
  50     certificates.push_back(client_1_);
  51     certificates.push_back(client_2_);
  52
  53     selector_ = new TestCertificateSelector(
  54         certificates, browser()->tab_strip_model()->GetActiveWebContents());
  55     selector_->Init();
  56     selector_->Show();
  57   }
  58
  59   void TearDownOnMainThread() override {
  60     selector_->Close();
  61     selector_ = nullptr;
  62   }
  63
  64  protected:
  65   scoped_refptr<net::X509Certificate> client_1_;
  66   scoped_refptr<net::X509Certificate> client_2_;
  67
  68   // The selector will be deleted when the browser is shutdown.
  69   TestCertificateSelector* selector_ = nullptr;
  70 };
  71
  72 }  // namespace
  73
  74 IN_PROC_BROWSER_TEST_F(CertificateSelectorTest, GetSelectedCert) {
  75   EXPECT_EQ(client_1_.get(), selector_->GetSelectedCert());
  76   EXPECT_TRUE(ui_test_utils::SendKeyPressSync(browser(), ui::VKEY_DOWN, false,
  77                                               false, false, false));
  78   EXPECT_EQ(client_2_.get(), selector_->GetSelectedCert());
  79   EXPECT_TRUE(ui_test_utils::SendKeyPressSync(browser(), ui::VKEY_UP, false,
  80                                               false, false, false));
  81   EXPECT_EQ(client_1_.get(), selector_->GetSelectedCert());
  82 }