| 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 |