Chromium Code Reviews| Index: media/tools/layout_tests/trend_graph_unittest.py |
| diff --git a/media/tools/layout_tests/trend_graph_unittest.py b/media/tools/layout_tests/trend_graph_unittest.py |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..5fcde5a79ca35d9d3fd470bc05d66ced8cce78fd |
| --- /dev/null |
| +++ b/media/tools/layout_tests/trend_graph_unittest.py |
| @@ -0,0 +1,37 @@ |
| +#!/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 unittest |
| +import shutil |
| +import os |
|
dennis_jeffrey
2011/08/25 00:36:46
Reverse the order of the above 3 lines to put them
imasaki1
2011/08/25 23:57:03
Done.
|
| + |
| +from trend_graph import TrendGraph |
| + |
| + |
| +class TestTrendGraph(unittest.TestCase): |
| + |
| + def testUpdate(self): |
| + test_graph_file_path = os.path.join('test_data', 'graph.html') |
| + test_graph_file_backup_path = os.path.join('test_data', 'graph.html.bk') |
| + shutil.copyfile(test_graph_file_path, test_graph_file_backup_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) |
| + # Recover the original file. |
| + shutil.copyfile(test_graph_file_backup_path, test_graph_file_path) |
|
dennis_jeffrey
2011/08/25 00:36:46
It's hacky to save the original graph file and the
imasaki1
2011/08/25 23:57:03
Created back up file and changed the code to copy
dennis_jeffrey
2011/08/26 19:01:26
This is better, but the approach of having a backu
|