| Index: tools/tests/base_unittest.py
|
| diff --git a/tools/tests/base_unittest.py b/tools/tests/base_unittest.py
|
| new file mode 100755
|
| index 0000000000000000000000000000000000000000..096039915b12cc03c0737034cc4601a05d6db054
|
| --- /dev/null
|
| +++ b/tools/tests/base_unittest.py
|
| @@ -0,0 +1,85 @@
|
| +#!/usr/bin/python
|
| +
|
| +"""
|
| +Copyright 2014 Google Inc.
|
| +
|
| +Use of this source code is governed by a BSD-style license that can be
|
| +found in the LICENSE file.
|
| +
|
| +A wrapper around the standard Python unittest library, adding features we need
|
| +for various unittests within this directory.
|
| +"""
|
| +
|
| +import os
|
| +import subprocess
|
| +import unittest
|
| +
|
| +
|
| +class TestCase(unittest.TestCase):
|
| +
|
| + def shortDescription(self):
|
| + """Tell unittest framework to not print docstrings for test cases."""
|
| + return None
|
| +
|
| + def run_command(self, args):
|
| + """Runs a program from the command line and returns stdout.
|
| +
|
| + Args:
|
| + args: Command line to run, as a list of string parameters. args[0] is the
|
| + binary to run.
|
| +
|
| + Returns:
|
| + stdout from the program, as a single string.
|
| +
|
| + Raises:
|
| + Exception: the program exited with a nonzero return code.
|
| + """
|
| + proc = subprocess.Popen(args,
|
| + stdout=subprocess.PIPE,
|
| + stderr=subprocess.PIPE)
|
| + (stdout, stderr) = proc.communicate()
|
| + if proc.returncode is not 0:
|
| + raise Exception('command "%s" failed: %s' % (args, stderr))
|
| + return stdout
|
| +
|
| + def find_path_to_program(self, program):
|
| + """Returns path to an existing program binary.
|
| +
|
| + Args:
|
| + program: Basename of the program to find (e.g., 'render_pictures').
|
| +
|
| + Returns:
|
| + Absolute path to the program binary, as a string.
|
| +
|
| + Raises:
|
| + Exception: unable to find the program binary.
|
| + """
|
| + trunk_path = os.path.abspath(os.path.join(os.path.dirname(__file__),
|
| + os.pardir, os.pardir))
|
| + possible_paths = [os.path.join(trunk_path, 'out', 'Release', program),
|
| + os.path.join(trunk_path, 'out', 'Debug', program),
|
| + os.path.join(trunk_path, 'out', 'Release',
|
| + program + '.exe'),
|
| + os.path.join(trunk_path, 'out', 'Debug',
|
| + program + '.exe')]
|
| + for try_path in possible_paths:
|
| + if os.path.isfile(try_path):
|
| + return try_path
|
| + raise Exception('cannot find %s in paths %s; maybe you need to '
|
| + 'build %s?' % (program, possible_paths, program))
|
| +
|
| +
|
| +def main(test_case_class):
|
| + """Run the unit tests within the given class.
|
| +
|
| + Raises an Exception if any of those tests fail (in case we are running in the
|
| + context of run_all.py, which depends on that Exception to signal failures).
|
| +
|
| + TODO(epoger): Make all of our unit tests use the Python unittest framework,
|
| + so we can leverage its ability to run *all* the tests and report failures at
|
| + the end.
|
| + """
|
| + suite = unittest.TestLoader().loadTestsFromTestCase(test_case_class)
|
| + results = unittest.TextTestRunner(verbosity=2).run(suite)
|
| + if not results.wasSuccessful():
|
| + raise Exception('failed unittest %s' % test_case_class)
|
|
|