media/tools/layout_tests/trend_graph_unittest.py

   1 #!/usr/bin/env python
   2 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 import os
   7 import shutil
   8 import unittest
   9
  10 from trend_graph import TrendGraph
  11
  12
  13 class TestTrendGraph(unittest.TestCase):
  14
  15   def testUpdate(self):
  16     test_graph_file_backup_path = os.path.join('test_data', 'graph.html.bak')
  17     test_graph_file_path = os.path.join('test_data', 'graph.html')
  18     shutil.copyfile(test_graph_file_backup_path, test_graph_file_path)
  19     trend_graph = TrendGraph(test_graph_file_path)
  20     data_map = {}
  21     data_map['whole'] = (['test1'], 'undefined', 'undefined')
  22     data_map['skip'] = (['test1', 'test2'], 'undefined', 'undefined')
  23     data_map['nonskip'] = (['test1', 'test2', 'test3'], 'undefined',
  24                            'undefined')
  25     data_map['passingrate'] = (str(4), 'undefined', 'undefined')
  26
  27     trend_graph.Update('2008,1,1,13,45,00', data_map)
  28     # Assert the result graph from the file.
  29     f = open(test_graph_file_path)
  30     lines2 = f.readlines()
  31     f.close()
  32     line_count = 0
  33     for line in lines2:
  34       if '2008,0,1,13,45,00' in line:
  35         line_count += 1
  36     self.assertEqual(line_count, 2)
  37
  38
  39 if __name__ == '__main__':
  40   unittest.main()