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
|
// 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 "chrome/browser/metrics/extensions_metrics_provider.h"
#include <string>
#include "base/memory/scoped_ptr.h"
#include "base/prefs/testing_pref_service.h"
#include "components/metrics/client_info.h"
#include "components/metrics/metrics_service.h"
#include "components/metrics/metrics_state_manager.h"
#include "components/metrics/proto/system_profile.pb.h"
#include "extensions/common/extension.h"
#include "extensions/common/extension_builder.h"
#include "extensions/common/extension_set.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
bool IsMetricsReportingEnabled() {
return true;
}
void StoreNoClientInfoBackup(const metrics::ClientInfo& /* client_info */) {
}
scoped_ptr<metrics::ClientInfo> ReturnNoBackup() {
return scoped_ptr<metrics::ClientInfo>();
}
class TestExtensionsMetricsProvider : public ExtensionsMetricsProvider {
public:
explicit TestExtensionsMetricsProvider(
metrics::MetricsStateManager* metrics_state_manager)
: ExtensionsMetricsProvider(metrics_state_manager) {}
// Makes the protected HashExtension method available to testing code.
using ExtensionsMetricsProvider::HashExtension;
protected:
// Override the GetInstalledExtensions method to return a set of extensions
// for tests.
scoped_ptr<extensions::ExtensionSet> GetInstalledExtensions(
Profile* profile) override {
scoped_ptr<extensions::ExtensionSet> extensions(
new extensions::ExtensionSet());
scoped_refptr<const extensions::Extension> extension;
extension = extensions::ExtensionBuilder()
.SetManifest(extensions::DictionaryBuilder()
.Set("name", "Test extension")
.Set("version", "1.0.0")
.Set("manifest_version", 2))
.SetID("ahfgeienlihckogmohjhadlkjgocpleb")
.Build();
extensions->Insert(extension);
extension = extensions::ExtensionBuilder()
.SetManifest(extensions::DictionaryBuilder()
.Set("name", "Test extension 2")
.Set("version", "1.0.0")
.Set("manifest_version", 2))
.SetID("pknkgggnfecklokoggaggchhaebkajji")
.Build();
extensions->Insert(extension);
extension = extensions::ExtensionBuilder()
.SetManifest(extensions::DictionaryBuilder()
.Set("name", "Colliding Extension")
.Set("version", "1.0.0")
.Set("manifest_version", 2))
.SetID("mdhofdjgenpkhlmddfaegdjddcecipmo")
.Build();
extensions->Insert(extension);
return extensions.Pass();
}
// Override GetClientID() to return a specific value on which test
// expectations are based.
uint64 GetClientID() override { return 0x3f1bfee9; }
};
} // namespace
// Checks that the hash function used to hide precise extension IDs produces
// the expected values.
TEST(ExtensionsMetricsProvider, HashExtension) {
EXPECT_EQ(978,
TestExtensionsMetricsProvider::HashExtension(
"ahfgeienlihckogmohjhadlkjgocpleb", 0));
EXPECT_EQ(10,
TestExtensionsMetricsProvider::HashExtension(
"ahfgeienlihckogmohjhadlkjgocpleb", 3817));
EXPECT_EQ(1007,
TestExtensionsMetricsProvider::HashExtension(
"pknkgggnfecklokoggaggchhaebkajji", 3817));
EXPECT_EQ(10,
TestExtensionsMetricsProvider::HashExtension(
"mdhofdjgenpkhlmddfaegdjddcecipmo", 3817));
}
// Checks that the fake set of extensions provided by
// TestExtensionsMetricsProvider is encoded properly.
TEST(ExtensionsMetricsProvider, SystemProtoEncoding) {
metrics::SystemProfileProto system_profile;
TestingPrefServiceSimple local_state;
metrics::MetricsService::RegisterPrefs(local_state.registry());
scoped_ptr<metrics::MetricsStateManager> metrics_state_manager(
metrics::MetricsStateManager::Create(
&local_state,
base::Bind(&IsMetricsReportingEnabled),
base::Bind(&StoreNoClientInfoBackup),
base::Bind(&ReturnNoBackup)));
TestExtensionsMetricsProvider extension_metrics(metrics_state_manager.get());
extension_metrics.ProvideSystemProfileMetrics(&system_profile);
ASSERT_EQ(2, system_profile.occupied_extension_bucket_size());
EXPECT_EQ(10, system_profile.occupied_extension_bucket(0));
EXPECT_EQ(1007, system_profile.occupied_extension_bucket(1));
}
|
