OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/python |
| 2 |
| 3 """ |
| 4 Copyright 2014 Google Inc. |
| 5 |
| 6 Use of this source code is governed by a BSD-style license that can be |
| 7 found in the LICENSE file. |
| 8 |
| 9 A wrapper around the standard Python unittest library, adding features we need |
| 10 for various unittests within this directory. |
| 11 """ |
| 12 |
| 13 # System-level imports. |
| 14 import os |
| 15 import sys |
| 16 |
| 17 PARENT_DIR = os.path.abspath(os.path.dirname(__file__)) |
| 18 TRUNK_DIR = os.path.abspath(os.path.join(PARENT_DIR, os.pardir, os.pardir)) |
| 19 |
| 20 # Import the superclass base_unittest module from the tools dir. |
| 21 # |
| 22 # TODO(epoger): If I don't put this at the beginning of sys.path, the import of |
| 23 # tests.base_unittest fails. That's bad. I need to come up with a cleaner way |
| 24 # of doing this... I think this will involve changing how we import the "boto" |
| 25 # library in gs_utils.py, within the common repo. |
| 26 TOOLS_DIR = os.path.join(TRUNK_DIR, 'tools') |
| 27 if TOOLS_DIR != sys.path[0]: |
| 28 sys.path.insert(0, TOOLS_DIR) |
| 29 import tests.base_unittest as superclass_module |
| 30 |
| 31 |
| 32 class TestCase(superclass_module.TestCase): |
| 33 |
| 34 def __init__(self, *args, **kwargs): |
| 35 super(TestCase, self).__init__(*args, **kwargs) |
| 36 # Some of the tests within this package want their output validated, |
| 37 # so we declare where the expected and actual output will be. |
| 38 self._testdata_dir = os.path.join(PARENT_DIR, 'testdata') |
| 39 |
| 40 def main(*args, **kwargs): |
| 41 superclass_module.main(*args, **kwargs) |
OLD | NEW |