sdk/platform-tools/systrace/catapult/common/py_utils/py_utils/py_utils_unittest.py

   1 # Copyright 2015 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.
   4 import os
   5 import sys
   6 import unittest
   7
   8 import py_utils
   9
  10
  11 class PathTest(unittest.TestCase):
  12
  13   def testIsExecutable(self):
  14     self.assertFalse(py_utils.IsExecutable('nonexistent_file'))
  15     # We use actual files on disk instead of pyfakefs because the executable is
  16     # set different on win that posix platforms and pyfakefs doesn't support
  17     # win platform well.
  18     self.assertFalse(py_utils.IsExecutable(_GetFileInTestDir('foo.txt')))
  19     self.assertTrue(py_utils.IsExecutable(sys.executable))
  20
  21
  22 def _GetFileInTestDir(file_name):
  23   return os.path.join(os.path.dirname(__file__), 'test_data', file_name)
  24
  25
  26 class WaitForTest(unittest.TestCase):
  27
  28   def testWaitForTrue(self):
  29     def ReturnTrue():
  30       return True
  31     self.assertTrue(py_utils.WaitFor(ReturnTrue, .1))
  32
  33   def testWaitForFalse(self):
  34     def ReturnFalse():
  35       return False
  36
  37     with self.assertRaises(py_utils.TimeoutException):
  38       py_utils.WaitFor(ReturnFalse, .1)
  39
  40   def testWaitForEventuallyTrue(self):
  41     # Use list to pass to inner function in order to allow modifying the
  42     # variable from the outer scope.
  43     c = [0]
  44     def ReturnCounterBasedValue():
  45       c[0] += 1
  46       return c[0] > 2
  47
  48     self.assertTrue(py_utils.WaitFor(ReturnCounterBasedValue, .5))
  49
  50   def testWaitForTrueLambda(self):
  51     self.assertTrue(py_utils.WaitFor(lambda: True, .1))
  52
  53   def testWaitForFalseLambda(self):
  54     with self.assertRaises(py_utils.TimeoutException):
  55       py_utils.WaitFor(lambda: False, .1)