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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
|
// Copyright 2020 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/ash/cert_provisioning/cert_provisioning_test_helpers.h"
#include <optional>
#include "base/test/gmock_callback_support.h"
#include "base/time/time.h"
#include "chrome/test/base/testing_browser_process.h"
#include "google_apis/gaia/gaia_id.h"
#include "net/test/cert_builder.h"
#include "testing/gtest/include/gtest/gtest.h"
using base::test::RunOnceCallback;
using testing::_;
using testing::Invoke;
namespace ash::cert_provisioning {
namespace {
std::vector<uint8_t> StrToBytes(const std::string& val) {
return std::vector<uint8_t>(val.begin(), val.end());
}
} // namespace
//================ CertificateHelperForTesting =================================
// Generated by chrome/test/data/policy/test_certs/create_test_certs.sh
const char kFakeCertificate[] = R"(-----BEGIN CERTIFICATE-----
MIIDJzCCAg+gAwIBAgIBATANBgkqhkiG9w0BAQsFADAXMRUwEwYDVQQDDAxyb290
X2NhX2NlcnQwHhcNMjAwMjI1MTUyNTU2WhcNMzAwMjIyMTUyNTU2WjAUMRIwEAYD
VQQDDAkxMjcuMC4wLjEwggEiMA0GCSqGSIb3DQEBAQUAA4IBDwAwggEKAoIBAQDW
druvpaJovmyWzIcjtsSk/lp319+zNPSYGLzJzTeEmnFoDf/b89ft6xR1NIahmvVd
UHGOMlzgDKnNkqWw+pgpn6U8dk+leWnwlUefzDz7OY8qXfX29Vh0m/kATQc64lnp
rX19fEi2DOgH6heCQDSaHI/KAnAXccwl8kdGuTEnvdzbdHqQq8pPGpEqzC/NOjk7
kDNkUt0J74ZVMm4+jhVOgZ35mFLtC+xjfycBgbnt8yfPOzmOMwXTjYDPNaIy32AZ
t66oIToteoW5Ilg+j5Mto3unBDHrw8rml3+W/nwHuOPEIgBqLQFfWtXpuX8CbcS6
SFNK4hxCJOvlzUbgTpsrAgMBAAGjgYAwfjAMBgNVHRMBAf8EAjAAMB0GA1UdDgQW
BBRDEl1/2pL5LtKnpIly+XCj3N6MwDAfBgNVHSMEGDAWgBQrwVEnUQZlX850A2N+
URfS8BxoyzAdBgNVHSUEFjAUBggrBgEFBQcDAQYIKwYBBQUHAwIwDwYDVR0RBAgw
BocEfwAAATANBgkqhkiG9w0BAQsFAAOCAQEAXZd+Ul7GUFZPLSiTZ618hUI2UdO0
7rtPwBw3TephWuyEeHht+WhzA3sRL3nprEiJqIg5w/Tlfz4dsObpSU3vKmDhLzAx
HJrN5vKdbEj9wyuhYSRJwvJka1ZOgPzhQcDQOp1SqonNxLx/sSMDR2UIDMBGzrkQ
sDkn58N5eWm+hZADOAKROHR47j85VcsmYGK7z2x479YzsyWyOm0dbACXv7/HvFkz
56KvgxRaPZQzQUg5yuXa21IjQz07wyWSYnHpm2duAbYFl6CTR9Rlj5vpRkKsQP1W
mMhGDBfgEskdbM+0agsZrJupoQMBUbD5gflcJlW3kwlboi3dTtiGixfYWw==
-----END CERTIFICATE-----)";
CertificateHelperForTesting::CertificateHelperForTesting(
platform_keys::MockPlatformKeysService* platform_keys_service)
: platform_keys_service_(platform_keys_service) {
template_cert_ = CreateSingleCertificateFromBytes(kFakeCertificate,
sizeof(kFakeCertificate));
DCHECK(template_cert_);
EXPECT_CALL(*platform_keys_service_, GetCertificates)
.WillRepeatedly(
Invoke(this, &CertificateHelperForTesting::GetCertificates));
}
CertificateHelperForTesting::~CertificateHelperForTesting() = default;
void CertificateHelperForTesting::GetCertificates(
chromeos::platform_keys::TokenId token_id,
platform_keys::GetCertificatesCallback callback) {
auto result = std::make_unique<net::CertificateList>();
*result = cert_list_;
std::move(callback).Run(std::move(result),
chromeos::platform_keys::Status::kSuccess);
}
scoped_refptr<net::X509Certificate> CertificateHelperForTesting::AddCert(
CertScope cert_scope,
const std::optional<CertProfileId>& cert_profile_id,
chromeos::platform_keys::Status status,
base::Time not_valid_before,
base::Time not_valid_after) {
net::CertBuilder cert_builder(template_cert_->cert_buffer(),
/*issuer=*/nullptr);
cert_builder.SetValidity(not_valid_before, not_valid_after);
auto cert = cert_builder.GetX509Certificate();
std::optional<std::vector<uint8_t>> attribute;
if (cert_profile_id.has_value()) {
attribute = StrToBytes(cert_profile_id.value());
}
EXPECT_CALL(
*platform_keys_service_,
GetAttributeForKey(
GetPlatformKeysTokenId(cert_scope),
chromeos::platform_keys::GetSubjectPublicKeyInfoBlob(cert),
chromeos::platform_keys::KeyAttributeType::kCertificateProvisioningId,
_))
.WillRepeatedly(
base::test::RunOnceCallbackRepeatedly<3>(attribute, status));
cert_list_.push_back(cert);
return cert;
}
scoped_refptr<net::X509Certificate> CertificateHelperForTesting::AddCert(
CertScope cert_scope,
const std::optional<CertProfileId>& cert_profile_id) {
base::Time not_valid_before = base::Time::Now() - base::Days(1);
base::Time not_valid_after = base::Time::Now() + base::Days(365);
return AddCert(cert_scope, cert_profile_id,
chromeos::platform_keys::Status::kSuccess, not_valid_before,
not_valid_after);
}
scoped_refptr<net::X509Certificate> CertificateHelperForTesting::AddCert(
CertScope cert_scope,
const std::optional<CertProfileId>& cert_profile_id,
chromeos::platform_keys::Status status) {
base::Time not_valid_before = base::Time::Now() - base::Days(1);
base::Time not_valid_after = base::Time::Now() + base::Days(365);
return AddCert(cert_scope, cert_profile_id, status, not_valid_before,
not_valid_after);
}
void CertificateHelperForTesting::ClearCerts() {
cert_list_.clear();
}
const net::CertificateList& CertificateHelperForTesting::GetCerts() const {
return cert_list_;
}
//================ ProfileHelperForTesting =====================================
namespace {
const char kTestUserEmail[] = "user@gmail.com";
const GaiaId::Literal kTestUserGaiaId("test_gaia_id");
} // namespace
ProfileHelperForTesting::ProfileHelperForTesting()
: ProfileHelperForTesting(/*user_is_affiilated=*/false) {}
ProfileHelperForTesting::ProfileHelperForTesting(bool user_is_affiliated)
: testing_profile_manager_(TestingBrowserProcess::GetGlobal()) {
Init(user_is_affiliated);
}
ProfileHelperForTesting::~ProfileHelperForTesting() = default;
void ProfileHelperForTesting::Init(bool user_is_affiliated) {
ASSERT_TRUE(testing_profile_manager_.SetUp());
testing_profile_ =
testing_profile_manager_.CreateTestingProfile(kTestUserEmail);
ASSERT_TRUE(testing_profile_);
auto test_account =
AccountId::FromUserEmailGaiaId(kTestUserEmail, kTestUserGaiaId);
user_ = fake_user_manager_->AddUserWithAffiliation(test_account,
user_is_affiliated);
}
Profile* ProfileHelperForTesting::GetProfile() const {
return testing_profile_;
}
user_manager::User* ProfileHelperForTesting::GetUser() const {
return user_;
}
} // namespace ash::cert_provisioning
|
