blob: f866aa6cf02ce1abb89f5c28c24dd1b8a6b4cb5b [file] [log] [blame]
#!/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',
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()
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__':