#!/usr/bin/python
# Copyright (c) 2011 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'] = (str(1), 'undefined', 'undefined')
    data_map['skip'] = (str(2), 'undefined', 'undefined')
    data_map['nonskip'] = (str(3), '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()
    lineCount = 0
    for line in lines2:
      if '2008,1,1,13,45,00' in line:
        lineCount += 1
    self.assertEqual(lineCount, 2)