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
|
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "net/cert/cert_database.h"
#include "base/test/metrics/histogram_tester.h"
#include "base/test/task_environment.h"
#include "net/log/test_net_log.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace net {
namespace {
class Observer : public CertDatabase::Observer {
public:
void OnTrustStoreChanged() override { trust_store_change_count_++; }
void OnClientCertStoreChanged() override { client_cert_change_count_++; }
int trust_store_change_count_ = 0;
int client_cert_change_count_ = 0;
};
} // namespace
TEST(CertDatabaseTest, Notifications) {
base::test::SingleThreadTaskEnvironment task_environment;
CertDatabase* cert_database = CertDatabase::GetInstance();
ASSERT_TRUE(cert_database);
Observer observer_1;
Observer observer_2;
cert_database->AddObserver(&observer_1);
cert_database->AddObserver(&observer_2);
{
RecordingNetLogObserver net_log_observer;
base::HistogramTester histograms;
cert_database->NotifyObserversClientCertStoreChanged();
task_environment.RunUntilIdle();
EXPECT_EQ(observer_1.trust_store_change_count_, 0);
EXPECT_EQ(observer_1.client_cert_change_count_, 1);
EXPECT_EQ(observer_2.trust_store_change_count_,
observer_1.trust_store_change_count_);
EXPECT_EQ(observer_2.client_cert_change_count_,
observer_1.client_cert_change_count_);
EXPECT_EQ(net_log_observer.GetEntries().size(), 1u);
EXPECT_EQ(
net_log_observer
.GetEntriesWithType(
NetLogEventType::CERTIFICATE_DATABASE_CLIENT_CERT_STORE_CHANGED)
.size(),
1u);
histograms.ExpectUniqueSample(
"Net.Certificate.ChangeNotification",
CertDatabase::HistogramNotificationType::kClientCert, 1);
}
{
RecordingNetLogObserver net_log_observer;
base::HistogramTester histograms;
cert_database->NotifyObserversTrustStoreChanged();
task_environment.RunUntilIdle();
EXPECT_EQ(observer_1.trust_store_change_count_, 1);
EXPECT_EQ(observer_1.client_cert_change_count_, 1);
EXPECT_EQ(observer_2.trust_store_change_count_,
observer_1.trust_store_change_count_);
EXPECT_EQ(observer_2.client_cert_change_count_,
observer_1.client_cert_change_count_);
EXPECT_EQ(net_log_observer.GetEntries().size(), 1u);
EXPECT_EQ(net_log_observer
.GetEntriesWithType(
NetLogEventType::CERTIFICATE_DATABASE_TRUST_STORE_CHANGED)
.size(),
1u);
histograms.ExpectUniqueSample(
"Net.Certificate.ChangeNotification",
CertDatabase::HistogramNotificationType::kTrust, 1);
}
cert_database->RemoveObserver(&observer_1);
cert_database->RemoveObserver(&observer_2);
}
} // namespace net
|
