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
3// found in the LICENSE file.
4
5#include <string>
6
7#include "base/memory/scoped_ptr.h"
8#include "base/metrics/histogram_base.h"
9#include "base/metrics/histogram_samples.h"
10#include "base/metrics/sample_map.h"
11#include "base/metrics/sparse_histogram.h"
12#include "base/metrics/statistics_recorder.h"
13#include "base/pickle.h"
14#include "base/strings/stringprintf.h"
15#include "testing/gtest/include/gtest/gtest.h"
16
17namespace base {
18
19class SparseHistogramTest : public testing::Test {
20 protected:
21  virtual void SetUp() {
22    // Each test will have a clean state (no Histogram / BucketRanges
23    // registered).
24    InitializeStatisticsRecorder();
25  }
26
27  virtual void TearDown() {
28    UninitializeStatisticsRecorder();
29  }
30
31  void InitializeStatisticsRecorder() {
32    statistics_recorder_ = new StatisticsRecorder();
33  }
34
35  void UninitializeStatisticsRecorder() {
36    delete statistics_recorder_;
37    statistics_recorder_ = NULL;
38  }
39
40  scoped_ptr<SparseHistogram> NewSparseHistogram(const std::string& name) {
41    return scoped_ptr<SparseHistogram>(new SparseHistogram(name));
42  }
43
44  StatisticsRecorder* statistics_recorder_;
45};
46
47TEST_F(SparseHistogramTest, BasicTest) {
48  scoped_ptr<SparseHistogram> histogram(NewSparseHistogram("Sparse"));
49  scoped_ptr<HistogramSamples> snapshot(histogram->SnapshotSamples());
50  EXPECT_EQ(0, snapshot->TotalCount());
51  EXPECT_EQ(0, snapshot->sum());
52
53  histogram->Add(100);
54  scoped_ptr<HistogramSamples> snapshot1(histogram->SnapshotSamples());
55  EXPECT_EQ(1, snapshot1->TotalCount());
56  EXPECT_EQ(1, snapshot1->GetCount(100));
57
58  histogram->Add(100);
59  histogram->Add(101);
60  scoped_ptr<HistogramSamples> snapshot2(histogram->SnapshotSamples());
61  EXPECT_EQ(3, snapshot2->TotalCount());
62  EXPECT_EQ(2, snapshot2->GetCount(100));
63  EXPECT_EQ(1, snapshot2->GetCount(101));
64}
65
66TEST_F(SparseHistogramTest, MacroBasicTest) {
67  UMA_HISTOGRAM_SPARSE_SLOWLY("Sparse", 100);
68  UMA_HISTOGRAM_SPARSE_SLOWLY("Sparse", 200);
69  UMA_HISTOGRAM_SPARSE_SLOWLY("Sparse", 100);
70
71  StatisticsRecorder::Histograms histograms;
72  StatisticsRecorder::GetHistograms(&histograms);
73
74  ASSERT_EQ(1U, histograms.size());
75  HistogramBase* sparse_histogram = histograms[0];
76
77  EXPECT_EQ(SPARSE_HISTOGRAM, sparse_histogram->GetHistogramType());
78  EXPECT_EQ("Sparse", sparse_histogram->histogram_name());
79  EXPECT_EQ(HistogramBase::kUmaTargetedHistogramFlag,
80            sparse_histogram->flags());
81
82  scoped_ptr<HistogramSamples> samples = sparse_histogram->SnapshotSamples();
83  EXPECT_EQ(3, samples->TotalCount());
84  EXPECT_EQ(2, samples->GetCount(100));
85  EXPECT_EQ(1, samples->GetCount(200));
86}
87
88TEST_F(SparseHistogramTest, MacroInLoopTest) {
89  // Unlike the macros in histogram.h, SparseHistogram macros can have a
90  // variable as histogram name.
91  for (int i = 0; i < 2; i++) {
92    std::string name = StringPrintf("Sparse%d", i + 1);
93    UMA_HISTOGRAM_SPARSE_SLOWLY(name, 100);
94  }
95
96  StatisticsRecorder::Histograms histograms;
97  StatisticsRecorder::GetHistograms(&histograms);
98  ASSERT_EQ(2U, histograms.size());
99
100  std::string name1 = histograms[0]->histogram_name();
101  std::string name2 = histograms[1]->histogram_name();
102  EXPECT_TRUE(("Sparse1" == name1 && "Sparse2" == name2) ||
103              ("Sparse2" == name1 && "Sparse1" == name2));
104}
105
106TEST_F(SparseHistogramTest, Serialize) {
107  scoped_ptr<SparseHistogram> histogram(NewSparseHistogram("Sparse"));
108  histogram->SetFlags(HistogramBase::kIPCSerializationSourceFlag);
109
110  Pickle pickle;
111  histogram->SerializeInfo(&pickle);
112
113  PickleIterator iter(pickle);
114
115  int type;
116  EXPECT_TRUE(iter.ReadInt(&type));
117  EXPECT_EQ(SPARSE_HISTOGRAM, type);
118
119  std::string name;
120  EXPECT_TRUE(iter.ReadString(&name));
121  EXPECT_EQ("Sparse", name);
122
123  int flag;
124  EXPECT_TRUE(iter.ReadInt(&flag));
125  EXPECT_EQ(HistogramBase::kIPCSerializationSourceFlag, flag);
126
127  // No more data in the pickle.
128  EXPECT_FALSE(iter.SkipBytes(1));
129}
130
131}  // namespace base
132