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
|
// Copyright 2024 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/metrics/structured/chrome_event_storage.h"
#include <cstdint>
#include <vector>
#include "base/files/file_path.h"
#include "base/files/scoped_temp_dir.h"
#include "base/test/metrics/histogram_tester.h"
#include "base/test/task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/metrics_proto/chrome_user_metrics_extension.pb.h"
#include "third_party/metrics_proto/structured_data.pb.h"
namespace metrics::structured {
namespace {
StructuredEventProto BuildTestEvent(
uint64_t id = 0,
const std::vector<int64_t>& metrics = std::vector<int64_t>()) {
StructuredEventProto event;
event.set_device_project_id(id);
int metric_id = 0;
for (int64_t metric : metrics) {
auto* m = event.add_metrics();
m->set_name_hash(metric_id++);
m->set_value_int64(metric);
}
return event;
}
} // namespace
class ChromeEventStorageTest : public testing::Test {
public:
void Wait() { task_environment_.RunUntilIdle(); }
StructuredDataProto GetReport(ChromeEventStorage* storage) {
StructuredDataProto structured_data;
*structured_data.mutable_events() = storage->TakeEvents();
return structured_data;
}
void ExpectNoErrors() {
histogram_tester_.ExpectTotalCount("UMA.StructuredMetrics.InternalError",
0);
}
private:
base::test::TaskEnvironment task_environment_{
base::test::TaskEnvironment::MainThreadType::UI,
base::test::TaskEnvironment::ThreadPoolExecutionMode::QUEUED,
base::test::TaskEnvironment::TimeSource::MOCK_TIME};
base::HistogramTester histogram_tester_;
base::ScopedTempDir temp_dir_;
};
TEST_F(ChromeEventStorageTest, StoreAndProvideEvents) {
ChromeEventStorage storage;
Wait();
ASSERT_TRUE(storage.IsReady());
storage.AddEvent(BuildTestEvent());
EXPECT_EQ(storage.RecordedEventsCount(), 1);
EventsProto events;
storage.CopyEvents(&events);
EXPECT_EQ(events.events_size(), 1);
StructuredDataProto proto = GetReport(&storage);
EXPECT_EQ(proto.events_size(), 1);
// Storage should have no events after a successful dump.
events.Clear();
storage.CopyEvents(&events);
EXPECT_EQ(events.events_size(), 0);
ExpectNoErrors();
}
TEST_F(ChromeEventStorageTest, EventsClearedAfterReport) {
ChromeEventStorage storage;
Wait();
storage.AddEvent(BuildTestEvent());
storage.AddEvent(BuildTestEvent());
// Should provide both the previous events.
EXPECT_EQ(GetReport(&storage).events_size(), 2);
// But the previous events shouldn't appear in the second report.
EXPECT_EQ(GetReport(&storage).events_size(), 0);
storage.AddEvent(BuildTestEvent());
// The third request should only contain the third event.
EXPECT_EQ(GetReport(&storage).events_size(), 1);
ExpectNoErrors();
}
TEST_F(ChromeEventStorageTest, Purge) {
ChromeEventStorage storage;
storage.AddEvent(BuildTestEvent());
storage.AddEvent(BuildTestEvent());
EXPECT_EQ(storage.RecordedEventsCount(), 2);
storage.Purge();
EXPECT_EQ(storage.RecordedEventsCount(), 0);
EXPECT_EQ(GetReport(&storage).events_size(), 0);
}
} // namespace metrics::structured
|
