# 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 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('benchmark')
    result = recorder.get_chart_data()
    self.assertEquals({
        'format_version': '1.0',
        'benchmark_name': 'benchmark',
        'charts': {}}, result)

  def test_one_chart(self):
    """Tests chart data with two samples in one chart."""
    recorder = ChartDataRecorder('benchmark')
    recorder.record_scalar('chart', 'val1', 'ms', 1)
    recorder.record_scalar('chart', 'val2', 'ms', 2)

    result = recorder.get_chart_data()
    self.assertEquals('1.0', result['format_version'])
    self.assertEquals('benchmark', result['benchmark_name'])

    charts = result['charts']
    self.assertEquals(1, len(charts))
    self.assertEquals(2, len(charts['chart']))
    self.assertEquals({
        'type': 'scalar',
        'units': 'ms',
        'value': 1}, charts['chart']['val1'])
    self.assertEquals({
        'type': 'scalar',
        'units': 'ms',
        'value': 2}, charts['chart']['val2'])

  def test_two_charts(self):
    """Tests chart data with two samples over two charts."""
    recorder = ChartDataRecorder('benchmark')
    recorder.record_scalar('chart1', 'val1', 'ms', 1)
    recorder.record_scalar('chart2', 'val2', 'ms', 2)

    result = recorder.get_chart_data()
    self.assertEquals('1.0', result['format_version'])
    self.assertEquals('benchmark', result['benchmark_name'])

    charts = result['charts']
    self.assertEquals(2, len(charts))
    self.assertEquals(1, len(charts['chart1']))
    self.assertEquals({
        'type': 'scalar',
        'units': 'ms',
        'value': 1}, charts['chart1']['val1'])
    self.assertEquals(1, len(charts['chart2']))
    self.assertEquals({
        'type': 'scalar',
        'units': 'ms',
        'value': 2}, charts['chart2']['val2'])

  def test_vectors(self):
    """Test recording a list of scalar values."""
    recorder = ChartDataRecorder('benchmark')
    recorder.record_vector('chart1', 'val1', 'ms', [1, 2])
    recorder.record_vector('chart2', 'val2', 'ms', [])

    result = recorder.get_chart_data()
    self.assertEquals('1.0', result['format_version'])
    self.assertEquals('benchmark', result['benchmark_name'])

    charts = result['charts']
    self.assertEquals(2, len(charts))
    self.assertEquals(1, len(charts['chart1']))
    self.assertEquals({
        'type': 'list_of_scalar_values',
        'units': 'ms',
        'values': [1, 2]}, charts['chart1']['val1'])
    self.assertEquals(1, len(charts['chart2']))
    self.assertEquals({
        'type': 'list_of_scalar_values',
        'units': 'ms',
        'values': []}, charts['chart2']['val2'])