Create a histogram serialization mechanism in components/metrics

base/metrics/chromeos_metrics_unittest.cc

+// Copyright (c) 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.
+
+// Test of crash class
+
+#include "base/file_util.h"
+#include "base/logging.h"
+#include "base/metrics/chromeos_metrics.h"
+#include "base/metrics/crash_sample.h"
+#include "base/metrics/histogram_sample.h"
+#include "base/metrics/linearhistogram_sample.h"
+#include "base/metrics/metric_sample.h"
+#include "base/metrics/sparsehistogram_sample.h"
+#include "base/metrics/useraction_sample.h"
+#include "base/strings/stringprintf.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace base {
+namespace {
+
+void AreEqHistogram(HistogramSample* a, HistogramSample* b) {
+  ASSERT_EQ(a->name(), b->name());
+  ASSERT_EQ(a->sample(), b->sample());
+  ASSERT_EQ(a->min(), b->min());
+  ASSERT_EQ(a->max(), b->max());
+  ASSERT_EQ(a->nbucket(), b->nbucket());
+}
+
+void AreEqCrash(CrashSample* a, CrashSample* b) {
+  ASSERT_EQ(a->name(), b->name());
+}
+
+void AreEqLinear(LinearHistogramSample* a, LinearHistogramSample* b) {
+  ASSERT_EQ(a->name(), b->name());
+  ASSERT_EQ(a->sample(), b->sample());
+  ASSERT_EQ(a->max(), b->max());
+}
+
+void AreEqSparse(SparseHistogramSample* a, SparseHistogramSample* b) {
+  ASSERT_EQ(a->name(), b->name());
+  ASSERT_EQ(a->sample(), b->sample());
+}
+
+void AreEqAction(UserActionSample* a, UserActionSample* b) {
+  ASSERT_EQ(a->name(), b->name());
+}
+
+void AreEq(MetricSample* a, MetricSample* b) {
+  ASSERT_TRUE(a);
+  ASSERT_TRUE(b);
+  MetricSample::SampleType type = a->type();
+  ASSERT_EQ(a->type(), b->type());
+  if (type == MetricSample::CRASH) {
+    AreEqCrash(reinterpret_cast<CrashSample*>(a),
+               reinterpret_cast<CrashSample*>(b));
+  } else if (type == MetricSample::HISTOGRAM) {
+    AreEqHistogram(reinterpret_cast<HistogramSample*>(a),
+                   reinterpret_cast<HistogramSample*>(b));
+  } else if (type == MetricSample::SPARSE_HISTOGRAM) {
+    AreEqSparse(reinterpret_cast<SparseHistogramSample*>(a),
+                reinterpret_cast<SparseHistogramSample*>(b));
+  } else if (type == MetricSample::LINEAR_HISTOGRAM) {
+    AreEqLinear(reinterpret_cast<LinearHistogramSample*>(a),
+                reinterpret_cast<LinearHistogramSample*>(b));
+  } else if (type == MetricSample::USER_ACTION) {
+    AreEqAction(reinterpret_cast<UserActionSample*>(a),
+                reinterpret_cast<UserActionSample*>(b));
+  } else {
+    DLOG(ERROR) << "could not recognize type";
+    FAIL();
+  }
+}
+
+class ChromeOSMetricTest : public testing::Test {
+ protected:
+  static const int buff_size = 1024;
+  char* buffer;
+  std::string filename;
+  FilePath filepath;
+  ChromeOSMetricTest() {
+    filename = "/tmp/chromeossampletest";
+    filepath = FilePath(filename);
+  }
+  virtual void SetUp() {
+    DeleteFile(filepath, false);
+    buffer = new char[buff_size];
+  }
+
+  void TestSerialization(MetricSample* sample) {
+    string serialized(sample->toString());
+    ASSERT_EQ('\0', serialized[serialized.length() - 1]);
+    AreEq(sample, ChromeOSMetrics::ReadSample(serialized));
+  }
+};
+
+TEST_F(ChromeOSMetricTest, CrashSerializeTest) {
+  CrashSample crash("test");
+  TestSerialization(&crash);
+}
+
+TEST_F(ChromeOSMetricTest, HistogramSerializeTest) {
+  HistogramSample hist("myhist", 13, 1, 100, 10);
+  TestSerialization(&hist);
+}
+
+TEST_F(ChromeOSMetricTest, LinearSerializeTest) {
+  LinearHistogramSample lhist("linearhist", 12, 30);
+  TestSerialization(&lhist);
+}
+
+TEST_F(ChromeOSMetricTest, SparseSerializeTest) {
+  SparseHistogramSample shist("mysparse", 30);
+  TestSerialization(&shist);
+}
+
+TEST_F(ChromeOSMetricTest, UserActionSerializeTest) {
+  TestSerialization(new UserActionSample("myaction"));
+}
+
+TEST_F(ChromeOSMetricTest, IllegalNameAreFilteredTest) {
+  SparseHistogramSample sample1("no space", 10);
+  LinearHistogramSample sample2(StringPrintf("here%cbhe", '\0'), 1, 3);
+  EXPECT_FALSE(ChromeOSMetrics::WriteMetricToFile(sample1, filename));
+  EXPECT_FALSE(ChromeOSMetrics::WriteMetricToFile(sample2, filename));
+  int64 size = GetFileSize(filepath, &size);
+
+  EXPECT_EQ(0, size);
+}
+
+TEST_F(ChromeOSMetricTest, BadInputIsCaughtTest) {
+  string input(StringPrintf("sparsehistogram%cname foo%c", '\0', '\0'));
+  EXPECT_EQ(NULL, SparseHistogramSample::ReadSparseHistogram(input));
+}
+
+TEST_F(ChromeOSMetricTest, MessageSeparatedByZero) {
+  CrashSample crash("crash");
+
+  ChromeOSMetrics::WriteMetricToFile(crash, filename);
+  int64 size;
+  GetFileSize(filepath, &size);
+  ASSERT_EQ(size, 16);
+}
+
+TEST_F(ChromeOSMetricTest, WriteReadTest) {
+  HistogramSample hist("myhist", 1, 2, 3, 4);
+  CrashSample crash("crash");
+  LinearHistogramSample lhist("linear", 1, 10);
+  SparseHistogramSample shist("mysparse", 30);
+  UserActionSample action("myaction");
+
+  ChromeOSMetrics::WriteMetricToFile(hist, filename);
+  ChromeOSMetrics::WriteMetricToFile(crash, filename);
+  ChromeOSMetrics::WriteMetricToFile(lhist, filename);
+  ChromeOSMetrics::WriteMetricToFile(shist, filename);
+  ChromeOSMetrics::WriteMetricToFile(action, filename);
+  ScopedVector<MetricSample> vect;
+  ChromeOSMetrics::ReadAndTruncateMetricsFromFile(filename, &vect);
+  ASSERT_EQ(vect.size(), size_t(5));
+  AreEq(&hist, vect[0]);
+  AreEq(&crash, vect[1]);
+  AreEq(&lhist, vect[2]);
+  AreEq(&shist, vect[3]);
+  AreEq(&action, vect[4]);
+
+  int64 size;
+  GetFileSize(filepath, &size);
+  ASSERT_EQ(0, size);
+}
+}  // namespace
+}  // namespace base