OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "chrome/test/base/uma_histogram_helper.h" | 5 #include "chrome/test/base/uma_histogram_helper.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/metrics/statistics_recorder.h" | 8 #include "base/metrics/statistics_recorder.h" |
9 #include "base/test/test_timeouts.h" | 9 #include "base/test/test_timeouts.h" |
10 #include "chrome/test/base/ui_test_utils.h" | 10 #include "chrome/test/base/ui_test_utils.h" |
11 #include "content/public/browser/histogram_fetcher.h" | 11 #include "content/public/browser/histogram_fetcher.h" |
12 | 12 |
13 UMAHistogramHelper::UMAHistogramHelper() { | 13 UMAHistogramHelper::UMAHistogramHelper() { |
14 base::StatisticsRecorder::Initialize(); | |
15 } | |
16 | |
17 UMAHistogramHelper::~UMAHistogramHelper() { | |
18 } | |
19 | |
20 void UMAHistogramHelper::PrepareSnapshot(const char* const histogram_names[], | |
21 size_t num_histograms) { | |
22 for (size_t i = 0; i < num_histograms; ++i) { | |
23 std::string histogram_name = histogram_names[i]; | |
24 | |
25 base::HistogramBase* histogram = | |
26 base::StatisticsRecorder::FindHistogram(histogram_name); | |
27 // If there is no histogram present, then don't record a snapshot. The logic | |
28 // in the Expect* methods will act to treat no histogram equivalent to | |
29 // samples with zeros. | |
30 if (histogram) { | |
31 histogram_snapshots[histogram_name] = | |
32 make_linked_ptr(histogram->SnapshotSamples().release()); | |
33 } | |
34 } | |
35 } | 14 } |
36 | 15 |
37 void UMAHistogramHelper::Fetch() { | 16 void UMAHistogramHelper::Fetch() { |
38 base::Closure callback = base::Bind(&UMAHistogramHelper::FetchCallback, | 17 base::Closure callback = base::Bind(&UMAHistogramHelper::FetchCallback, |
39 base::Unretained(this)); | 18 base::Unretained(this)); |
40 | 19 |
41 content::FetchHistogramsAsynchronously( | 20 content::FetchHistogramsAsynchronously( |
42 base::MessageLoop::current(), | 21 base::MessageLoop::current(), |
43 callback, | 22 callback, |
44 // If this call times out, it means that a child process is not | 23 // If this call times out, it means that a child process is not |
(...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
96 | 75 |
97 void UMAHistogramHelper::FetchCallback() { | 76 void UMAHistogramHelper::FetchCallback() { |
98 base::MessageLoopForUI::current()->Quit(); | 77 base::MessageLoopForUI::current()->Quit(); |
99 } | 78 } |
100 | 79 |
101 void UMAHistogramHelper::CheckBucketCount( | 80 void UMAHistogramHelper::CheckBucketCount( |
102 const std::string& name, | 81 const std::string& name, |
103 base::HistogramBase::Sample sample, | 82 base::HistogramBase::Sample sample, |
104 base::HistogramBase::Count expected_count, | 83 base::HistogramBase::Count expected_count, |
105 base::HistogramSamples& samples) { | 84 base::HistogramSamples& samples) { |
106 int actual_count = samples.GetCount(sample); | 85 EXPECT_EQ(expected_count, samples.GetCount(sample)) |
107 if (histogram_snapshots.count(name)) | |
108 actual_count -= histogram_snapshots[name]->GetCount(sample); | |
109 EXPECT_EQ(expected_count, actual_count) | |
110 << "Histogram \"" << name | 86 << "Histogram \"" << name |
111 << "\" does not have the right number of samples (" << expected_count | 87 << "\" does not have the right number of samples (" << expected_count |
112 << ") in the expected bucket (" << sample << "). It has (" << actual_count | 88 << ") in the expected bucket (" << sample << ")."; |
113 << ")."; | |
114 } | 89 } |
115 | 90 |
116 void UMAHistogramHelper::CheckTotalCount( | 91 void UMAHistogramHelper::CheckTotalCount( |
117 const std::string& name, | 92 const std::string& name, |
118 base::HistogramBase::Count expected_count, | 93 base::HistogramBase::Count expected_count, |
119 base::HistogramSamples& samples) { | 94 base::HistogramSamples& samples) { |
120 int actual_count = samples.TotalCount(); | 95 EXPECT_EQ(expected_count, samples.TotalCount()) |
121 if (histogram_snapshots.count(name)) | |
122 actual_count -= histogram_snapshots[name]->TotalCount(); | |
123 EXPECT_EQ(expected_count, actual_count) | |
124 << "Histogram \"" << name | 96 << "Histogram \"" << name |
125 << "\" does not have the right total number of samples (" | 97 << "\" does not have the right total number of samples (" |
126 << expected_count << "). It has (" << actual_count << ")."; | 98 << expected_count << ")."; |
127 } | 99 } |
OLD | NEW |