util_unittest.py revision 6e8cce623b6e4fe0c9e4af605d675dd9d0338c38 
1# Copyright 2012 The Chromium Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4import os
5import shutil
6import tempfile
7import unittest
8
9from telemetry.core import util
10
11
12class TestWait(unittest.TestCase):
13  @staticmethod
14  def testNonTimeout():
15    def test():
16      return True
17    util.WaitFor(test, 0.1)
18
19  def testTimeout(self):
20    def test():
21      return False
22    self.assertRaises(util.TimeoutException, lambda: util.WaitFor(test, 0.1))
23
24  def testCallable(self):
25    """Test methods and anonymous functions, functions are tested elsewhere."""
26    class Test(object):
27      def Method(self):
28        return 'test'
29    util.WaitFor(Test().Method, 0.1)
30
31    util.WaitFor(lambda: 1, 0.1)
32
33    # Test noncallable condition.
34    self.assertRaises(TypeError, lambda: util.WaitFor('test', 0.1))
35
36  def testReturn(self):
37    self.assertEquals('test', util.WaitFor(lambda: 'test', 0.1))
38
39class TestGetSequentialFileName(unittest.TestCase):
40  def __init__(self, *args, **kwargs):
41    super(TestGetSequentialFileName, self).__init__(*args, **kwargs)
42    self.test_directory = None
43
44  def setUp(self):
45    self.test_directory = tempfile.mkdtemp()
46
47  def testGetSequentialFileNameNoOtherSequentialFile(self):
48    next_json_test_file_path = util.GetSequentialFileName(
49        os.path.join(self.test_directory, 'test'))
50    self.assertEquals(os.path.join(self.test_directory, 'test_000'),
51                      next_json_test_file_path)
52
53  def testGetSequentialFileNameWithOtherSequentialFiles(self):
54    # Create test_000.json, test_001.json, test_002.json in test directory.
55    for i in xrange(3):
56      with open(
57          os.path.join(self.test_directory, 'test_%03d.json' % i), 'w') as _:
58        pass
59    next_json_test_file_path = util.GetSequentialFileName(
60        os.path.join(self.test_directory, 'test'))
61    self.assertEquals(os.path.join(self.test_directory, 'test_003'),
62                      next_json_test_file_path)
63
64  def tearDown(self):
65    shutil.rmtree(self.test_directory)
66