Improve UMA tests; improve no-op reconcilor; new reconcilor test.

chrome/test/base/uma_histogram_helper.cc

 // 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.
 
 #include "chrome/test/base/uma_histogram_helper.h"
 
 #include "base/bind.h"
 #include "base/metrics/statistics_recorder.h"
 #include "base/test/test_timeouts.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/histogram_fetcher.h"
 
 UMAHistogramHelper::UMAHistogramHelper() {
+  base::StatisticsRecorder::Initialize();
+}
+
+void UMAHistogramHelper::PrepareSnapshot(const char** histogram_names,
+                                         int num_histograms) {
+  for (int i = 0; i < num_histograms; i++) {
+    std::string histogram_name = histogram_names[i];
+
+    base::HistogramBase* histogram =
+        base::StatisticsRecorder::FindHistogram(histogram_name);
+    // If there is no histogram present, then don't record a snapshot. The logic
+    // in the Expect* methods will act to treat no histogram equivalent to
+    // samples with zeros.
+    if (histogram)
+      histogram_snapshots[histogram_name] = histogram->SnapshotSamples();
+  }
 }
 
 void UMAHistogramHelper::Fetch() {
   base::Closure callback = base::Bind(&UMAHistogramHelper::FetchCallback,
                                       base::Unretained(this));
 
   content::FetchHistogramsAsynchronously(
       base::MessageLoop::current(),
       callback,
       // If this call times out, it means that a child process is not
@@ skipping 51 matching lines @@
 
 void UMAHistogramHelper::FetchCallback() {
   base::MessageLoopForUI::current()->Quit();
 }
 
 void UMAHistogramHelper::CheckBucketCount(
     const std::string& name,
     base::HistogramBase::Sample sample,
     base::HistogramBase::Count expected_count,
     base::HistogramSamples& samples) {
-  EXPECT_EQ(expected_count, samples.GetCount(sample))
+  int actual_count;
+  if (histogram_snapshots.count(name)) {
+    actual_count = samples.GetCount(sample) -
+                   histogram_snapshots[name]->GetCount(sample);
+  } else {
+    actual_count = samples.GetCount(sample);
+  }
+  EXPECT_EQ(expected_count, actual_count)
       << "Histogram \"" << name
       << "\" does not have the right number of samples (" << expected_count
-      << ") in the expected bucket (" << sample << ").";
+      << ") in the expected bucket (" << sample << "). It has ("
+      << actual_count << ").";
 }
 
 void UMAHistogramHelper::CheckTotalCount(
     const std::string& name,
     base::HistogramBase::Count expected_count,
     base::HistogramSamples& samples) {
-  EXPECT_EQ(expected_count, samples.TotalCount())
+  int actual_count;
+  if (histogram_snapshots.count(name)) {

[Alexei Svitkine (slow), 2014/06/12 17:09:11]

Nit: How about:

int actual_count = samples.TotalCount();
if (histogram_snapshots.count(name))
  actual_count -= histogram_snapshots[name]->TotalCount();

Same above.

[Mike Lerman, 2014/06/12 18:12:15]

Slick. Done.

+    actual_count = samples.TotalCount() -
+                   histogram_snapshots[name]->TotalCount();
+  } else {
+    actual_count = samples.TotalCount();
+  }
+  EXPECT_EQ(expected_count, actual_count)
       << "Histogram \"" << name
       << "\" does not have the right total number of samples ("
-      << expected_count << ").";
+      << expected_count << "). It has (" << actual_count << ").";
 }