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
|
// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/rappor/rappor_service.h"
#include "base/base64.h"
#include "base/prefs/testing_pref_service.h"
#include "components/metrics/metrics_hashes.h"
#include "components/rappor/byte_vector_utils.h"
#include "components/rappor/proto/rappor_metric.pb.h"
#include "components/rappor/rappor_parameters.h"
#include "components/rappor/rappor_pref_names.h"
#include "components/rappor/test_log_uploader.h"
#include "components/rappor/test_rappor_service.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace rappor {
TEST(RapporServiceTest, Update) {
TestRapporService rappor_service;
EXPECT_LT(base::TimeDelta(), rappor_service.next_rotation());
EXPECT_TRUE(rappor_service.test_uploader()->is_running());
rappor_service.Update(RECORDING_DISABLED, false);
EXPECT_EQ(base::TimeDelta(), rappor_service.next_rotation());
EXPECT_FALSE(rappor_service.test_uploader()->is_running());
rappor_service.Update(FINE_LEVEL, false);
EXPECT_LT(base::TimeDelta(), rappor_service.next_rotation());
EXPECT_FALSE(rappor_service.test_uploader()->is_running());
rappor_service.Update(COARSE_LEVEL, true);
EXPECT_LT(base::TimeDelta(), rappor_service.next_rotation());
EXPECT_TRUE(rappor_service.test_uploader()->is_running());
}
// Check that samples can be recorded and exported.
TEST(RapporServiceTest, RecordAndExportMetrics) {
TestRapporService rappor_service;
// Multiple samples for the same metric should only generate one report.
rappor_service.RecordSample("MyMetric", ETLD_PLUS_ONE_RAPPOR_TYPE, "foo");
rappor_service.RecordSample("MyMetric", ETLD_PLUS_ONE_RAPPOR_TYPE, "bar");
RapporReports reports;
rappor_service.GetReports(&reports);
EXPECT_EQ(1, reports.report_size());
const RapporReports::Report& report = reports.report(0);
EXPECT_TRUE(report.name_hash());
// ETLD_PLUS_ONE_RAPPOR_TYPE has 128 bits
EXPECT_EQ(16u, report.bits().size());
}
// Check that the reporting level is respected.
TEST(RapporServiceTest, RecordingLevel) {
TestRapporService rappor_service;
rappor_service.Update(COARSE_LEVEL, false);
// ETLD_PLUS_ONE_RAPPOR_TYPE is a FINE_LEVEL metric
rappor_service.RecordSample("FineMetric", ETLD_PLUS_ONE_RAPPOR_TYPE, "foo");
RapporReports reports;
rappor_service.GetReports(&reports);
EXPECT_EQ(0, reports.report_size());
}
// Check that GetRecordedSampleForMetric works as expected.
TEST(RapporServiceTest, GetRecordedSampleForMetric) {
TestRapporService rappor_service;
// Multiple samples for the same metric; only the latest is remembered.
rappor_service.RecordSample("MyMetric", ETLD_PLUS_ONE_RAPPOR_TYPE, "foo");
rappor_service.RecordSample("MyMetric", ETLD_PLUS_ONE_RAPPOR_TYPE, "bar");
std::string sample;
RapporType type;
EXPECT_FALSE(
rappor_service.GetRecordedSampleForMetric("WrongMetric", &sample, &type));
EXPECT_TRUE(
rappor_service.GetRecordedSampleForMetric("MyMetric", &sample, &type));
EXPECT_EQ("bar", sample);
EXPECT_EQ(ETLD_PLUS_ONE_RAPPOR_TYPE, type);
}
// Check that the incognito is respected.
TEST(RapporServiceTest, Incognito) {
TestRapporService rappor_service;
rappor_service.set_is_incognito(true);
rappor_service.RecordSample("MyMetric", COARSE_RAPPOR_TYPE, "foo");
RapporReports reports;
rappor_service.GetReports(&reports);
EXPECT_EQ(0, reports.report_size());
}
// Check that Sample objects record correctly.
TEST(RapporServiceTest, RecordSample) {
TestRapporService rappor_service;
scoped_ptr<Sample> sample = rappor_service.CreateSample(COARSE_RAPPOR_TYPE);
sample->SetStringField("Url", "example.com");
sample->SetFlagsField("Flags1", 0xbcd, 12);
rappor_service.RecordSampleObj("ObjMetric", sample.Pass());
uint64_t url_hash = metrics::HashMetricName("ObjMetric.Url");
uint64_t flags_hash = metrics::HashMetricName("ObjMetric.Flags1");
RapporReports reports;
rappor_service.GetReports(&reports);
EXPECT_EQ(2, reports.report_size());
size_t url_index = reports.report(0).name_hash() == url_hash ? 0 : 1;
size_t flags_index = url_index == 0 ? 1 : 0;
EXPECT_EQ(url_hash, reports.report(url_index).name_hash());
EXPECT_EQ(1u, reports.report(url_index).bits().size());
EXPECT_EQ(flags_hash, reports.report(flags_index).name_hash());
EXPECT_EQ(2u, reports.report(flags_index).bits().size());
}
} // namespace rappor
|
