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 |