Index: gm/rebaseline_server/base_unittest.py |
diff --git a/gm/rebaseline_server/base_unittest.py b/gm/rebaseline_server/base_unittest.py |
new file mode 100755 |
index 0000000000000000000000000000000000000000..b8a653866b1479c7da0d25f3553f2c667b067dee |
--- /dev/null |
+++ b/gm/rebaseline_server/base_unittest.py |
@@ -0,0 +1,41 @@ |
+#!/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. |
+""" |
+ |
+# System-level imports. |
+import os |
+import sys |
+ |
+PARENT_DIR = os.path.abspath(os.path.dirname(__file__)) |
+TRUNK_DIR = os.path.abspath(os.path.join(PARENT_DIR, os.pardir, os.pardir)) |
+ |
+# Import the superclass base_unittest module from the tools dir. |
+# |
+# TODO(epoger): If I don't put this at the beginning of sys.path, the import of |
+# tests.base_unittest fails. That's bad. I need to come up with a cleaner way |
+# of doing this... I think this will involve changing how we import the "boto" |
+# library in gs_utils.py, within the common repo. |
+TOOLS_DIR = os.path.join(TRUNK_DIR, 'tools') |
+if TOOLS_DIR != sys.path[0]: |
+ sys.path.insert(0, TOOLS_DIR) |
+import tests.base_unittest as superclass_module |
+ |
+ |
+class TestCase(superclass_module.TestCase): |
+ |
+ def __init__(self, *args, **kwargs): |
+ super(TestCase, self).__init__(*args, **kwargs) |
+ # Some of the tests within this package want their output validated, |
+ # so we declare where the expected and actual output will be. |
+ self._testdata_dir = os.path.join(PARENT_DIR, 'testdata') |
+ |
+def main(*args, **kwargs): |
+ superclass_module.main(*args, **kwargs) |