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