Extend Histogram class to support custom range definitions...

base/histogram_unittest.cc

 // Copyright (c) 2006-2008 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.
 
 // Test of Histogram class
 
 #include "base/histogram.h"
 #include "base/string_util.h"
 #include "base/time.h"
 #include "testing/gtest/include/gtest/gtest.h"
@@ skipping 11 matching lines @@
   scoped_refptr<Histogram> histogram = Histogram::FactoryGet(
       "TestHistogram", 1, 1000, 10, Histogram::kNoFlags);
   scoped_refptr<Histogram> histogram1 = Histogram::FactoryGet(
       "Test1Histogram", 1, 1000, 10, Histogram::kNoFlags);
 
   scoped_refptr<Histogram> linear_histogram = LinearHistogram::FactoryGet(
       "TestLinearHistogram", 1, 1000, 10, Histogram::kNoFlags);
   scoped_refptr<Histogram> linear_histogram1 = LinearHistogram::FactoryGet(
       "Test1LinearHistogram", 1, 1000, 10, Histogram::kNoFlags);
 
+  std::vector<int> custom_ranges;
+  custom_ranges.push_back(1);
+  custom_ranges.push_back(5);
+  custom_ranges.push_back(10);
+  custom_ranges.push_back(20);
+  custom_ranges.push_back(30);
+  scoped_refptr<Histogram> custom_histogram = CustomHistogram::FactoryGet(
+      "TestCustomHistogram", custom_ranges, Histogram::kNoFlags);
+  scoped_refptr<Histogram> custom_histogram1 = CustomHistogram::FactoryGet(
+      "Test1CustomHistogram", custom_ranges, Histogram::kNoFlags);
+
   // Use standard macros (but with fixed samples)
   HISTOGRAM_TIMES("Test2Histogram", TimeDelta::FromDays(1));
   HISTOGRAM_COUNTS("Test3Histogram", 30);
 
   DHISTOGRAM_TIMES("Test4Histogram", TimeDelta::FromDays(1));
   DHISTOGRAM_COUNTS("Test5Histogram", 30);
 
   HISTOGRAM_ENUMERATION("Test6Histogram", 129, 130);
 
   // Try to construct samples.
@@ skipping 25 matching lines @@
   StatisticsRecorder::GetHistograms(&histograms);  // Load up lists
   EXPECT_EQ(1U, histograms.size());
   scoped_refptr<Histogram> histogram1 = Histogram::FactoryGet(
       "Test1Histogram", 1, 1000, 10, Histogram::kNoFlags);
   histograms.clear();
   StatisticsRecorder::GetHistograms(&histograms);  // Load up lists
   EXPECT_EQ(2U, histograms.size());
 
   scoped_refptr<Histogram> linear_histogram = LinearHistogram::FactoryGet(
       "TestLinearHistogram", 1, 1000, 10, Histogram::kNoFlags);
+  histograms.clear();
+  StatisticsRecorder::GetHistograms(&histograms);  // Load up lists
+  EXPECT_EQ(3U, histograms.size());
+
   scoped_refptr<Histogram> linear_histogram1 = LinearHistogram::FactoryGet(
       "Test1LinearHistogram", 1, 1000, 10, Histogram::kNoFlags);
   histograms.clear();
   StatisticsRecorder::GetHistograms(&histograms);  // Load up lists
   EXPECT_EQ(4U, histograms.size());
 
+  std::vector<int> custom_ranges;
+  custom_ranges.push_back(1);
+  custom_ranges.push_back(5);
+  custom_ranges.push_back(10);
+  custom_ranges.push_back(20);
+  custom_ranges.push_back(30);
+  scoped_refptr<Histogram> custom_histogram = CustomHistogram::FactoryGet(
+      "TestCustomHistogram", custom_ranges, Histogram::kNoFlags);
+  scoped_refptr<Histogram> custom_histogram1 = CustomHistogram::FactoryGet(
+      "TestCustomHistogram", custom_ranges, Histogram::kNoFlags);
+
+  histograms.clear();
+  StatisticsRecorder::GetHistograms(&histograms);  // Load up lists
+  EXPECT_EQ(5U, histograms.size());
+
   // Use standard macros (but with fixed samples)
   HISTOGRAM_TIMES("Test2Histogram", TimeDelta::FromDays(1));
   HISTOGRAM_COUNTS("Test3Histogram", 30);
   histograms.clear();
   StatisticsRecorder::GetHistograms(&histograms);  // Load up lists
-  EXPECT_EQ(6U, histograms.size());
+  EXPECT_EQ(7U, histograms.size());
 
   HISTOGRAM_ENUMERATION("TestEnumerationHistogram", 20, 200);
   histograms.clear();
   StatisticsRecorder::GetHistograms(&histograms);  // Load up lists
-  EXPECT_EQ(7U, histograms.size());
+  EXPECT_EQ(8U, histograms.size());
 
   DHISTOGRAM_TIMES("Test4Histogram", TimeDelta::FromDays(1));
   DHISTOGRAM_COUNTS("Test5Histogram", 30);
   histograms.clear();
   StatisticsRecorder::GetHistograms(&histograms);  // Load up lists
 #ifndef NDEBUG
-  EXPECT_EQ(9U, histograms.size());
+  EXPECT_EQ(10U, histograms.size());
 #else
-  EXPECT_EQ(7U, histograms.size());
+  EXPECT_EQ(8U, histograms.size());
 #endif
 }
 
 TEST(HistogramTest, RangeTest) {
   StatisticsRecorder recorder;
   StatisticsRecorder::Histograms histograms;
 
   recorder.GetHistograms(&histograms);
   EXPECT_EQ(0U, histograms.size());
 
@@ skipping 44 matching lines @@
   EXPECT_EQ(7, transitioning_histogram->ranges(7));
   EXPECT_EQ(9, transitioning_histogram->ranges(8));
   EXPECT_EQ(11, transitioning_histogram->ranges(9));
   EXPECT_EQ(14, transitioning_histogram->ranges(10));
   EXPECT_EQ(17, transitioning_histogram->ranges(11));
   EXPECT_EQ(21, transitioning_histogram->ranges(12));
   EXPECT_EQ(26, transitioning_histogram->ranges(13));
   EXPECT_EQ(32, transitioning_histogram->ranges(14));
   EXPECT_EQ(INT_MAX, transitioning_histogram->ranges(15));
 
+  std::vector<int> custom_ranges;
+  custom_ranges.push_back(0);
+  custom_ranges.push_back(9);
+  custom_ranges.push_back(10);
+  custom_ranges.push_back(11);
+  custom_ranges.push_back(300);
+  scoped_refptr<Histogram> test_custom_histogram = CustomHistogram::FactoryGet(
+      "TestCustomRangeHistogram", custom_ranges, Histogram::kNoFlags);
+
+  EXPECT_EQ(custom_ranges[0], test_custom_histogram->ranges(0));
+  EXPECT_EQ(custom_ranges[1], test_custom_histogram->ranges(1));
+  EXPECT_EQ(custom_ranges[2], test_custom_histogram->ranges(2));
+  EXPECT_EQ(custom_ranges[3], test_custom_histogram->ranges(3));
+  EXPECT_EQ(custom_ranges[4], test_custom_histogram->ranges(4));
+
   recorder.GetHistograms(&histograms);
-  EXPECT_EQ(5U, histograms.size());
+  EXPECT_EQ(6U, histograms.size());
 }
 
+TEST(HistogramTest, CustomRangeTest) {
+  StatisticsRecorder recorder;
+  StatisticsRecorder::Histograms histograms;
+
+  // Check that missing leading zero is handled by an auto-insertion.
+  std::vector<int> custom_ranges;
+  // Don't include a zero.
+  custom_ranges.push_back(9);
+  custom_ranges.push_back(10);
+  custom_ranges.push_back(11);
+  scoped_refptr<Histogram> test_custom_histogram = CustomHistogram::FactoryGet(
+      "TestCustomRangeHistogram", custom_ranges, Histogram::kNoFlags);
+
+  EXPECT_EQ(0, test_custom_histogram->ranges(0));  // Auto added
+  EXPECT_EQ(custom_ranges[0], test_custom_histogram->ranges(1));
+  EXPECT_EQ(custom_ranges[1], test_custom_histogram->ranges(2));
+  EXPECT_EQ(custom_ranges[2], test_custom_histogram->ranges(3));
+
+  // Check that unsorted data with dups is handled gracefully.
+  const int kSmall = 7;
+  const int kMid = 8;
+  const int kBig = 9;
+  custom_ranges.clear();
+  custom_ranges.push_back(kBig);
+  custom_ranges.push_back(kMid);
+  custom_ranges.push_back(kSmall);
+  custom_ranges.push_back(kSmall);
+  custom_ranges.push_back(kMid);
+  custom_ranges.push_back(0);  // Push an explicit zero.
+  custom_ranges.push_back(kBig);
+
+  scoped_refptr<Histogram> unsorted_histogram = CustomHistogram::FactoryGet(
+      "TestCustomUnsortedDupedHistogram", custom_ranges, Histogram::kNoFlags);
+  EXPECT_EQ(0, unsorted_histogram->ranges(0));
+  EXPECT_EQ(kSmall, unsorted_histogram->ranges(1));
+  EXPECT_EQ(kMid, unsorted_histogram->ranges(2));
+  EXPECT_EQ(kBig, unsorted_histogram->ranges(3));
+}
+
+
 // Make sure histogram handles out-of-bounds data gracefully.
 TEST(HistogramTest, BoundsTest) {
   const size_t kBucketCount = 50;
   scoped_refptr<Histogram> histogram = Histogram::FactoryGet(
       "Bounded", 10, 100, kBucketCount, Histogram::kNoFlags);
 
   // Put two samples "out of bounds" above and below.
   histogram->Add(5);
   histogram->Add(-50);
 
@@ skipping 38 matching lines @@
   // Check to see that the bucket counts reflect our additions.
   Histogram::SampleSet sample;
   histogram->SnapshotSample(&sample);
   EXPECT_EQ(INT_MAX, histogram->ranges(8));
   for (int i = 0; i < 8; i++)
     EXPECT_EQ(i + 1, sample.counts(i));
 }
 
 
 }  // namespace