#!/usr/bin/env python
# Copyright (c) 2012 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.

import os
import shutil
import unittest

from trend_graph import TrendGraph


class TestTrendGraph(unittest.TestCase):

  def testUpdate(self):
    test_graph_file_backup_path = os.path.join('test_data', 'graph.html.bak')
    test_graph_file_path = os.path.join('test_data', 'graph.html')
    shutil.copyfile(test_graph_file_backup_path, test_graph_file_path)
    trend_graph = TrendGraph(test_graph_file_path)
    data_map = {}
    data_map['whole'] = (['test1'], 'undefined', 'undefined')
    data_map['skip'] = (['test1', 'test2'], 'undefined', 'undefined')
    data_map['nonskip'] = (['test1', 'test2', 'test3'], 'undefined',
                           'undefined')
    data_map['passingrate'] = (str(4), 'undefined', 'undefined')

    trend_graph.Update('2008,1,1,13,45,00', data_map)
    # Assert the result graph from the file.
    f = open(test_graph_file_path)
    lines2 = f.readlines()
    f.close()
    line_count = 0
    for line in lines2:
      if '2008,0,1,13,45,00' in line:
        line_count += 1
    self.assertEqual(line_count, 2)


if __name__ == '__main__':
  unittest.main()