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
|