OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "components/suggestions/blacklist_store.h" | 5 #include "components/suggestions/blacklist_store.h" |
6 | 6 |
7 #include <set> | 7 #include <set> |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
11 #include "base/metrics/histogram_samples.h" | 11 |
12 #include "base/metrics/statistics_recorder.h" | 12 #include "base/test/histogram_tester.h" |
13 #include "base/test/statistics_delta_reader.h" | |
14 #include "components/pref_registry/testing_pref_service_syncable.h" | 13 #include "components/pref_registry/testing_pref_service_syncable.h" |
15 #include "components/suggestions/proto/suggestions.pb.h" | 14 #include "components/suggestions/proto/suggestions.pb.h" |
16 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
17 | 16 |
18 using user_prefs::TestingPrefServiceSyncable; | 17 using user_prefs::TestingPrefServiceSyncable; |
19 | 18 |
20 namespace suggestions { | 19 namespace suggestions { |
21 | 20 |
22 namespace { | 21 namespace { |
23 | 22 |
(...skipping 102 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
126 EXPECT_TRUE(blacklist_store.BlacklistUrl(GURL(kTestUrlB))); | 125 EXPECT_TRUE(blacklist_store.BlacklistUrl(GURL(kTestUrlB))); |
127 | 126 |
128 // Expect to retrieve A or B. | 127 // Expect to retrieve A or B. |
129 EXPECT_TRUE(blacklist_store.GetFirstUrlFromBlacklist(&retrieved)); | 128 EXPECT_TRUE(blacklist_store.GetFirstUrlFromBlacklist(&retrieved)); |
130 std::string retrieved_string = retrieved.spec(); | 129 std::string retrieved_string = retrieved.spec(); |
131 EXPECT_TRUE(retrieved_string == std::string(kTestUrlA) || | 130 EXPECT_TRUE(retrieved_string == std::string(kTestUrlA) || |
132 retrieved_string == std::string(kTestUrlB)); | 131 retrieved_string == std::string(kTestUrlB)); |
133 } | 132 } |
134 | 133 |
135 TEST_F(BlacklistStoreTest, LogsBlacklistSize) { | 134 TEST_F(BlacklistStoreTest, LogsBlacklistSize) { |
136 base::StatisticsDeltaReader statistics_delta_reader; | 135 base::HistogramTester histogram_tester; |
137 | 136 |
138 // Create a first store - blacklist is empty at this point. | 137 // Create a first store - blacklist is empty at this point. |
139 scoped_ptr<BlacklistStore> blacklist_store( | 138 scoped_ptr<BlacklistStore> blacklist_store( |
140 new BlacklistStore(pref_service())); | 139 new BlacklistStore(pref_service())); |
141 scoped_ptr<base::HistogramSamples> samples( | 140 |
142 statistics_delta_reader.GetHistogramSamplesSinceCreation( | 141 histogram_tester.ExpectTotalCount("Suggestions.LocalBlacklistSize", 1); |
143 "Suggestions.LocalBlacklistSize")); | 142 histogram_tester.ExpectUniqueSample("Suggestions.LocalBlacklistSize", 0, 1); |
144 EXPECT_EQ(1, samples->TotalCount()); | |
145 EXPECT_EQ(1, samples->GetCount(0)); | |
146 | 143 |
147 // Add some content to the blacklist. | 144 // Add some content to the blacklist. |
148 EXPECT_TRUE(blacklist_store->BlacklistUrl(GURL(kTestUrlA))); | 145 EXPECT_TRUE(blacklist_store->BlacklistUrl(GURL(kTestUrlA))); |
149 EXPECT_TRUE(blacklist_store->BlacklistUrl(GURL(kTestUrlB))); | 146 EXPECT_TRUE(blacklist_store->BlacklistUrl(GURL(kTestUrlB))); |
150 | 147 |
151 // Create a new BlacklistStore and verify the counts. | 148 // Create a new BlacklistStore and verify the counts. |
152 blacklist_store.reset(new BlacklistStore(pref_service())); | 149 blacklist_store.reset(new BlacklistStore(pref_service())); |
153 samples = statistics_delta_reader.GetHistogramSamplesSinceCreation( | 150 |
154 "Suggestions.LocalBlacklistSize"); | 151 histogram_tester.ExpectTotalCount("Suggestions.LocalBlacklistSize", 2); |
155 EXPECT_EQ(2, samples->TotalCount()); | 152 histogram_tester.ExpectBucketCount("Suggestions.LocalBlacklistSize", 0, 1); |
156 EXPECT_EQ(1, samples->GetCount(0)); | 153 histogram_tester.ExpectBucketCount("Suggestions.LocalBlacklistSize", 2, 1); |
157 EXPECT_EQ(1, samples->GetCount(2)); | |
158 } | 154 } |
159 | 155 |
160 } // namespace suggestions | 156 } // namespace suggestions |
OLD | NEW |