ssl_client_certificate_selector_test.cc revision 2a99a7e74a7f215066514fe81d2bfa6639d9eddd 
1// Copyright (c) 2012 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 "chrome/browser/ssl/ssl_client_certificate_selector_test.h"
6
7#include "base/bind.h"
8#include "base/files/file_path.h"
9#include "chrome/browser/profiles/profile.h"
10#include "chrome/browser/ui/browser.h"
11#include "chrome/browser/ui/tabs/tab_strip_model.h"
12#include "chrome/test/base/ui_test_utils.h"
13#include "content/public/browser/web_contents.h"
14#include "content/public/test/browser_test_utils.h"
15#include "net/base/cert_test_util.h"
16#include "net/base/test_data_directory.h"
17#include "net/base/x509_certificate.h"
18#include "net/http/http_transaction_factory.h"
19#include "net/ssl/ssl_cert_request_info.h"
20#include "net/url_request/url_request.h"
21#include "net/url_request/url_request_context.h"
22#include "net/url_request/url_request_context_getter.h"
23
24using ::testing::Mock;
25using ::testing::StrictMock;
26using content::BrowserThread;
27
28SSLClientCertificateSelectorTestBase::SSLClientCertificateSelectorTestBase()
29    : io_loop_finished_event_(false, false) {
30}
31
32SSLClientCertificateSelectorTestBase::~SSLClientCertificateSelectorTestBase() {
33}
34
35void SSLClientCertificateSelectorTestBase::SetUpInProcessBrowserTestFixture() {
36  base::FilePath certs_dir = net::GetTestCertsDirectory();
37
38  mit_davidben_cert_ = net::ImportCertFromFile(certs_dir, "mit.davidben.der");
39  ASSERT_TRUE(mit_davidben_cert_);
40
41  foaf_me_chromium_test_cert_ = net::ImportCertFromFile(
42      certs_dir, "foaf.me.chromium-test-cert.der");
43  ASSERT_TRUE(foaf_me_chromium_test_cert_);
44
45  cert_request_info_ = new net::SSLCertRequestInfo;
46  cert_request_info_->host_and_port = "foo:123";
47  cert_request_info_->client_certs.push_back(mit_davidben_cert_);
48  cert_request_info_->client_certs.push_back(foaf_me_chromium_test_cert_);
49}
50
51void SSLClientCertificateSelectorTestBase::SetUpOnMainThread() {
52  url_request_context_getter_ = browser()->profile()->GetRequestContext();
53
54  BrowserThread::PostTask(
55      BrowserThread::IO,
56      FROM_HERE,
57      base::Bind(&SSLClientCertificateSelectorTestBase::SetUpOnIOThread,
58                 this));
59
60  io_loop_finished_event_.Wait();
61
62  content::WaitForLoadStop(
63      browser()->tab_strip_model()->GetActiveWebContents());
64}
65
66// Have to release our reference to the auth handler during the test to allow
67// it to be destroyed while the Browser and its IO thread still exist.
68void SSLClientCertificateSelectorTestBase::CleanUpOnMainThread() {
69  BrowserThread::PostTask(
70      BrowserThread::IO,
71      FROM_HERE,
72      base::Bind(&SSLClientCertificateSelectorTestBase::CleanUpOnIOThread,
73                 this));
74
75  io_loop_finished_event_.Wait();
76
77  auth_requestor_ = NULL;
78}
79
80void SSLClientCertificateSelectorTestBase::SetUpOnIOThread() {
81  url_request_ = MakeURLRequest(url_request_context_getter_);
82
83  auth_requestor_ = new StrictMock<SSLClientAuthRequestorMock>(
84      url_request_,
85      cert_request_info_);
86
87  io_loop_finished_event_.Signal();
88}
89
90void SSLClientCertificateSelectorTestBase::CleanUpOnIOThread() {
91  delete url_request_;
92
93  io_loop_finished_event_.Signal();
94}
95
96net::URLRequest* SSLClientCertificateSelectorTestBase::MakeURLRequest(
97    net::URLRequestContextGetter* context_getter) {
98  net::URLRequest* request =
99      context_getter->GetURLRequestContext()->CreateRequest(
100          GURL("https://example"), NULL);
101  return request;
102}
103