| // Copyright (c) 2012 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. |
| |
| #ifndef CHROME_TEST_BASE_UMA_HISTOGRAM_HELPER_H_ |
| #define CHROME_TEST_BASE_UMA_HISTOGRAM_HELPER_H_ |
| |
| #include "base/memory/linked_ptr.h" |
| #include "base/metrics/histogram.h" |
| #include "base/metrics/histogram_base.h" |
| #include "base/metrics/histogram_samples.h" |
| |
| // UMAHistogramHelper provides a simple interface for examining UMA histograms. |
| // Tests can use this interface to verify that UMA data is getting logged as |
| // intended. |
| class UMAHistogramHelper { |
| public: |
| // UMAHistogramHelper should be created before the execution of the test case. |
| UMAHistogramHelper(); |
| |
| ~UMAHistogramHelper(); |
| |
| // Parameters should be string literals of all histograms to snapshot. |
| // Call this before executing the test code. This method can be called |
| // multiple times. The existing snapshots are preserved, except when one of |
| // the |histogram_names| was previously passed as a parameter, then a new |
| // snapshot will replace the existing one. |
| void PrepareSnapshot(const char* const histogram_names[], |
| size_t num_histograms); |
| |
| // Each child process may have its own histogram data, make sure this data |
| // gets accumulated into the browser process before we examine the histograms. |
| void Fetch(); |
| |
| // We know the exact number of samples in a bucket, and that no other bucket |
| // should have samples. If |PrepareSnapshot| was called for the histogram |
| // named |name| then the |expected_count| is the diff from the snapshot. |
| void ExpectUniqueSample(const std::string& name, |
| base::HistogramBase::Sample sample, |
| base::HistogramBase::Count expected_count); |
| |
| // We know the exact number of samples in a bucket, but other buckets may |
| // have samples as well. If |PrepareSnapshot| was called for histogram named |
| // |name| then the |expected_count| is the diff from the snapshot. |
| void ExpectBucketCount(const std::string& name, |
| base::HistogramBase::Sample sample, |
| base::HistogramBase::Count expected_count); |
| |
| // We don't know the values of the samples, but we know how many there are. If |
| // |PrepareSnapshot| was called for |name| histogram, then the |
| // |count| is the diff from the snapshot. |
| void ExpectTotalCount(const std::string& name, |
| base::HistogramBase::Count count); |
| |
| private: |
| void FetchCallback(); |
| |
| void CheckBucketCount(const std::string& name, |
| base::HistogramBase::Sample sample, |
| base::Histogram::Count expected_count, |
| base::HistogramSamples& samples); |
| |
| void CheckTotalCount(const std::string& name, |
| base::Histogram::Count expected_count, |
| base::HistogramSamples& samples); |
| |
| DISALLOW_COPY_AND_ASSIGN(UMAHistogramHelper); |
| |
| // The map from histogram names to their snapshots |
| std::map<std::string, linked_ptr<base::HistogramSamples> > |
| histogram_snapshots; |
| }; |
| |
| #endif // CHROME_TEST_BASE_UMA_HISTOGRAM_HELPER_H_ |