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
|
// Copyright 2017 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/reporting/reporting_garbage_collector.h"
#include <string>
#include "base/test/simple_test_tick_clock.h"
#include "base/time/time.h"
#include "base/timer/mock_timer.h"
#include "net/base/isolation_info.h"
#include "net/base/network_anonymization_key.h"
#include "net/reporting/reporting_cache.h"
#include "net/reporting/reporting_policy.h"
#include "net/reporting/reporting_report.h"
#include "net/reporting/reporting_test_util.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/abseil-cpp/absl/types/optional.h"
namespace net {
namespace {
class ReportingGarbageCollectorTest : public ReportingTestBase {
protected:
size_t report_count() {
std::vector<const ReportingReport*> reports;
cache()->GetReports(&reports);
return reports.size();
}
const absl::optional<base::UnguessableToken> kReportingSource_ =
base::UnguessableToken::Create();
const NetworkAnonymizationKey kNik_;
const IsolationInfo kIsolationInfo_;
const GURL kUrl_ = GURL("https://origin/path");
const std::string kUserAgent_ = "Mozilla/1.0";
const std::string kGroup_ = "group";
const std::string kType_ = "default";
};
// Make sure the garbage collector is actually present in the context.
TEST_F(ReportingGarbageCollectorTest, Created) {
EXPECT_NE(nullptr, garbage_collector());
}
// Make sure that the garbage collection timer is started and stopped correctly.
TEST_F(ReportingGarbageCollectorTest, Timer) {
EXPECT_FALSE(garbage_collection_timer()->IsRunning());
cache()->AddReport(absl::nullopt, kNik_, kUrl_, kUserAgent_, kGroup_, kType_,
base::Value::Dict(), 0, tick_clock()->NowTicks(), 0);
EXPECT_TRUE(garbage_collection_timer()->IsRunning());
garbage_collection_timer()->Fire();
EXPECT_FALSE(garbage_collection_timer()->IsRunning());
}
TEST_F(ReportingGarbageCollectorTest, Report) {
cache()->AddReport(absl::nullopt, kNik_, kUrl_, kUserAgent_, kGroup_, kType_,
base::Value::Dict(), 0, tick_clock()->NowTicks(), 0);
garbage_collection_timer()->Fire();
EXPECT_EQ(1u, report_count());
}
TEST_F(ReportingGarbageCollectorTest, ExpiredReport) {
cache()->AddReport(absl::nullopt, kNik_, kUrl_, kUserAgent_, kGroup_, kType_,
base::Value::Dict(), 0, tick_clock()->NowTicks(), 0);
tick_clock()->Advance(2 * policy().max_report_age);
garbage_collection_timer()->Fire();
EXPECT_EQ(0u, report_count());
}
TEST_F(ReportingGarbageCollectorTest, FailedReport) {
cache()->AddReport(absl::nullopt, kNik_, kUrl_, kUserAgent_, kGroup_, kType_,
base::Value::Dict(), 0, tick_clock()->NowTicks(), 0);
std::vector<const ReportingReport*> reports;
cache()->GetReports(&reports);
for (int i = 0; i < policy().max_report_attempts; i++) {
cache()->IncrementReportsAttempts(reports);
}
garbage_collection_timer()->Fire();
EXPECT_EQ(0u, report_count());
}
TEST_F(ReportingGarbageCollectorTest, ExpiredSource) {
ReportingEndpointGroupKey group_key(kNik_, kReportingSource_,
url::Origin::Create(kUrl_), kGroup_);
cache()->SetV1EndpointForTesting(group_key, *kReportingSource_,
kIsolationInfo_, kUrl_);
// Mark the source as expired. The source should be removed as soon as
// garbage collection runs, as there are no queued reports for it.
cache()->SetExpiredSource(*kReportingSource_);
// Before garbage collection, the endpoint should still exist.
EXPECT_EQ(1u, cache()->GetReportingSourceCountForTesting());
EXPECT_TRUE(cache()->GetV1EndpointForTesting(*kReportingSource_, kGroup_));
// Fire garbage collection. The endpoint configuration should be removed.
garbage_collection_timer()->Fire();
EXPECT_EQ(0u, cache()->GetReportingSourceCountForTesting());
EXPECT_FALSE(cache()->GetV1EndpointForTesting(*kReportingSource_, kGroup_));
}
TEST_F(ReportingGarbageCollectorTest, ExpiredSourceWithPendingReports) {
ReportingEndpointGroupKey group_key(kNik_, kReportingSource_,
url::Origin::Create(kUrl_), kGroup_);
cache()->SetV1EndpointForTesting(group_key, *kReportingSource_,
kIsolationInfo_, kUrl_);
cache()->AddReport(kReportingSource_, kNik_, kUrl_, kUserAgent_, kGroup_,
kType_, base::Value::Dict(), 0, tick_clock()->NowTicks(),
0);
// Mark the source as expired. The source data should be removed as soon as
// all reports are delivered.
cache()->SetExpiredSource(*kReportingSource_);
// Even though expired, GC should not delete the source as there is still a
// queued report.
garbage_collection_timer()->Fire();
EXPECT_EQ(1u, report_count());
EXPECT_EQ(1u, cache()->GetReportingSourceCountForTesting());
EXPECT_TRUE(cache()->GetV1EndpointForTesting(*kReportingSource_, kGroup_));
// Deliver report.
std::vector<const ReportingReport*> reports;
cache()->GetReports(&reports);
cache()->RemoveReports(reports);
EXPECT_EQ(0u, report_count());
// Fire garbage collection again. The endpoint configuration should be
// removed.
garbage_collection_timer()->Fire();
EXPECT_EQ(0u, cache()->GetReportingSourceCountForTesting());
EXPECT_FALSE(cache()->GetV1EndpointForTesting(*kReportingSource_, kGroup_));
}
} // namespace
} // namespace net
|
