Teach mojo_benchmark to produce chart_data for the perf dashboard.

mojo/devtools/common/devtoolslib/perf_dashboard_unittest.py

+# Copyright 2015 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.
+
+"""Tests for the Chromium Performance Dashboard data format implementation."""
+
+import imp
+import json
+import os.path
+import sys
+import unittest
+
+try:
+  imp.find_module("devtoolslib")
+except ImportError:
+  sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from devtoolslib.perf_dashboard import ChartDataRecorder
+
+class ChartDataRecorderTest(unittest.TestCase):
+  """Tests the chart data recorder."""
+
+  def test_empty(self):
+    """Tests chart data with no charts."""
+    recorder = ChartDataRecorder()
+    result = json.loads(recorder.get_json())
+    self.assertEquals(0, len(result))
+
+  def test_one_chart(self):
+    """Tests chart data with two samples in one chart."""
+    recorder = ChartDataRecorder()
+    recorder.record_scalar('chart', 'val1', 'ms', 1)
+    recorder.record_scalar('chart', 'val2', 'ms', 2)
+
+    result = json.loads(recorder.get_json())
+    self.assertEquals(1, len(result))
+    self.assertEquals(2, len(result['chart']))
+    self.assertEquals({
+        'type': 'scalar',
+        'name': 'val1',
+        'units': 'ms',
+        'value': 1}, result['chart'][0])
+    self.assertEquals({
+        'type': 'scalar',
+        'name': 'val2',
+        'units': 'ms',
+        'value': 2}, result['chart'][1])
+
+  def test_two_charts(self):
+    """Tests chart data with two samples over two charts."""
+    recorder = ChartDataRecorder()
+    recorder.record_scalar('chart1', 'val1', 'ms', 1)
+    recorder.record_scalar('chart2', 'val2', 'ms', 2)
+
+    result = json.loads(recorder.get_json())
+    self.assertEquals(2, len(result))
+    self.assertEquals(1, len(result['chart1']))
+    self.assertEquals({
+        'type': 'scalar',
+        'name': 'val1',
+        'units': 'ms',
+        'value': 1}, result['chart1'][0])
+    self.assertEquals(1, len(result['chart2']))
+    self.assertEquals({
+        'type': 'scalar',
+        'name': 'val2',
+        'units': 'ms',
+        'value': 2}, result['chart2'][0])