Index: base/metrics/statistics_recorder_unittest.cc |
diff --git a/base/metrics/statistics_recorder_unittest.cc b/base/metrics/statistics_recorder_unittest.cc |
index af5c1e7bbc316c2d8cb9b90cb8cac9805217ce07..880ab39c362222f65e670fd6a5d769fe1224dd48 100644 |
--- a/base/metrics/statistics_recorder_unittest.cc |
+++ b/base/metrics/statistics_recorder_unittest.cc |
@@ -10,6 +10,7 @@ |
#include "base/json/json_reader.h" |
#include "base/memory/scoped_ptr.h" |
#include "base/metrics/histogram_macros.h" |
+#include "base/metrics/histogram_persistence.h" |
#include "base/metrics/sparse_histogram.h" |
#include "base/metrics/statistics_recorder.h" |
#include "base/values.h" |
@@ -25,7 +26,10 @@ class StatisticsRecorderTest : public testing::Test { |
InitializeStatisticsRecorder(); |
} |
- void TearDown() override { UninitializeStatisticsRecorder(); } |
+ void TearDown() override { |
+ UninitializeStatisticsRecorder(); |
+ SetPersistentHistogramMemoryAllocator(nullptr); |
+ } |
void InitializeStatisticsRecorder() { |
statistics_recorder_ = new StatisticsRecorder(); |
@@ -315,6 +319,23 @@ TEST_F(StatisticsRecorderTest, ToJSON) { |
EXPECT_TRUE(json.empty()); |
} |
+TEST_F(StatisticsRecorderTest, IterationTest) { |
+ StatisticsRecorder::Histograms registered_histograms; |
+ LOCAL_HISTOGRAM_COUNTS("TestHistogram.IterationTest1", 30); |
+ SetPersistentHistogramMemoryAllocator( |
+ new LocalPersistentMemoryAllocator(64 << 10, 0, std::string())); |
+ LOCAL_HISTOGRAM_COUNTS("TestHistogram.IterationTest2", 30); |
+ |
+ StatisticsRecorder::HistogramIterator i1 = StatisticsRecorder::begin(true); |
+ EXPECT_NE(StatisticsRecorder::end(), i1); |
+ EXPECT_NE(StatisticsRecorder::end(), ++i1); |
+ EXPECT_EQ(StatisticsRecorder::end(), ++i1); |
+ |
+ StatisticsRecorder::HistogramIterator i2 = StatisticsRecorder::begin(false); |
+ EXPECT_NE(StatisticsRecorder::end(), i2); |
+ EXPECT_EQ(StatisticsRecorder::end(), ++i2); |
+} |
+ |
namespace { |
// CallbackCheckWrapper is simply a convenient way to check and store that |