base/test/histogram_tester.h: Add a way to query a whole family of related

base/test/histogram_tester.cc

 // 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 "base/test/histogram_tester.h"
 
+#include <algorithm>
+#include <vector>
+
 #include "base/metrics/histogram.h"
 #include "base/metrics/histogram_samples.h"
 #include "base/metrics/statistics_recorder.h"
 #include "base/stl_util.h"
+#include "base/strings/string_util.h"
+#include "base/strings/stringprintf.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace base {
 
 HistogramTester::HistogramTester() {
   StatisticsRecorder::Initialize();  // Safe to call multiple times.
 
   // Record any histogram data that exists when the object is created so it can
   // be subtracted later.
   StatisticsRecorder::Histograms histograms;
@@ skipping 45 matching lines @@
       base::StatisticsRecorder::FindHistogram(name);
   if (histogram) {
     scoped_ptr<base::HistogramSamples> samples(histogram->SnapshotSamples());
     CheckTotalCount(name, count, *samples);
   } else {
     // No histogram means there were zero samples.
     EXPECT_EQ(count, 0) << "Histogram \"" << name << "\" does not exist.";
   }
 }
 
+std::string HistogramTester::GetTotalCountsForQuery(
+    const std::string& query) const {
+  EXPECT_TRUE(query.find('.') != std::string::npos)
+      << "|query| ought to contain at least one period, to avoid matching too"
+      << " many histograms.";

[Ilya Sherman, 2015/06/23 00:57:40]

nit: Perhaps output the query?

+
+  std::vector<std::string> result;
+  StatisticsRecorder::Histograms histograms;
+  StatisticsRecorder::GetSnapshot(query, &histograms);
+  for (base::HistogramBase* histogram : histograms) {
+    scoped_ptr<HistogramSamples> samples = histogram->SnapshotSamples();
+    auto old_histogram = histograms_snapshot_.find(histogram->histogram_name());
+    if (old_histogram != histograms_snapshot_.end())
+      samples->Subtract(*old_histogram->second);
+
+    if (histogram->GetHistogramType() == LINEAR_HISTOGRAM ||
+        histogram->GetHistogramType() == BOOLEAN_HISTOGRAM) {
+      for (scoped_ptr<SampleCountIterator> bucket_it = samples->Iterator();
+           !bucket_it->Done(); bucket_it->Next()) {
+        HistogramBase::Sample min, max;
+        HistogramBase::Count count;
+        bucket_it->Get(&min, &max, &count);
+        if (count != 0) {
+          result.push_back(base::StringPrintf(
+              "%s[%d]=%d", histogram->histogram_name().c_str(), min, count));
+        }
+      }
+    } else {
+      HistogramBase::Count count = samples->TotalCount();
+      if (count != 0) {
+        result.push_back(base::StringPrintf(
+            "%s=%d", histogram->histogram_name().c_str(), count));
+      }
+    }
+  }
+  std::sort(result.begin(), result.end());
+  return JoinString(result, "\n");
+}
+
 scoped_ptr<HistogramSamples> HistogramTester::GetHistogramSamplesSinceCreation(
     const std::string& histogram_name) {
   HistogramBase* histogram = StatisticsRecorder::FindHistogram(histogram_name);
   if (!histogram)
     return scoped_ptr<HistogramSamples>();
   scoped_ptr<HistogramSamples> named_samples(histogram->SnapshotSamples());
   HistogramSamples* named_original_samples =
       histograms_snapshot_[histogram_name];
   if (named_original_samples)
     named_samples->Subtract(*named_original_samples);
@@ skipping 28 matching lines @@
   if (histogram_data != histograms_snapshot_.end())
     actual_count -= histogram_data->second->TotalCount();
 
   EXPECT_EQ(expected_count, actual_count)
       << "Histogram \"" << name
       << "\" does not have the right total number of samples ("
       << expected_count << "). It has (" << actual_count << ").";
 }
 
 }  // namespace base