1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
#!/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()
|
