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
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/certificate_provider/test_certificate_provider_extension_mixin.h"
#include <memory>
#include "chrome/browser/certificate_provider/test_certificate_provider_extension.h"
#include "chrome/browser/policy/extension_force_install_mixin.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/test/base/mixin_based_in_process_browser_test.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace ash {
TestCertificateProviderExtensionMixin::TestCertificateProviderExtensionMixin(
InProcessBrowserTestMixinHost* host,
ExtensionForceInstallMixin* extension_force_install_mixin)
: InProcessBrowserTestMixin(host),
extension_force_install_mixin_(extension_force_install_mixin) {}
TestCertificateProviderExtensionMixin::
~TestCertificateProviderExtensionMixin() = default;
void TestCertificateProviderExtensionMixin::TearDownOnMainThread() {
certificate_provider_extension_.reset();
}
void TestCertificateProviderExtensionMixin::ForceInstall(
Profile* profile,
bool wait_on_extension_loaded,
bool immediately_provide_certificates) {
DCHECK(wait_on_extension_loaded || !immediately_provide_certificates)
<< "When wait_on_extension_loaded is unset, "
"immediately_provide_certificates must also be unset!";
DCHECK(extension_force_install_mixin_->initialized())
<< "ExtensionForceInstallMixin must be initialized before calling "
"ForceInstall!";
DCHECK(!certificate_provider_extension_) << "ForceInstall already called!";
certificate_provider_extension_ =
std::make_unique<TestCertificateProviderExtension>(profile);
ASSERT_TRUE(extension_force_install_mixin_->ForceInstallFromSourceDir(
TestCertificateProviderExtension::GetExtensionSourcePath(),
TestCertificateProviderExtension::GetExtensionPemPath(),
wait_on_extension_loaded
? ExtensionForceInstallMixin::WaitMode::kBackgroundPageFirstLoad
: ExtensionForceInstallMixin::WaitMode::kPrefSet));
if (wait_on_extension_loaded && immediately_provide_certificates)
certificate_provider_extension_->TriggerSetCertificates();
}
} // namespace ash
|
