Imported Upstream version 1.0.1

system/run.py

@@ -0,0 +1,131 @@
+#!/usr/local/bin/python
+
+import glob
+import importlib
+import os
+import inspect
+import fnmatch
+import sys
+import traceback
+import random
+
+from lib import BaseTest
+from s3_lib import S3Test
+from swift_lib import SwiftTest
+from api_lib import APITest
+
+try:
+    from termcolor import colored
+except ImportError:
+    def colored(s, **kwargs):
+        return s
+
+
+def run(include_long_tests=False, capture_results=False, tests=None, filters=None):
+    """
+    Run system test.
+    """
+    if not tests:
+        tests = glob.glob("t*_*")
+    fails = []
+    numTests = numFailed = numSkipped = 0
+    lastBase = None
+
+    for test in tests:
+
+        testModule = importlib.import_module(test)
+
+        for name in dir(testModule):
+            o = getattr(testModule, name)
+
+            if not (inspect.isclass(o) and issubclass(o, BaseTest) and o is not BaseTest and
+                    o is not SwiftTest and o is not S3Test and o is not APITest):
+                continue
+
+            newBase = o.__bases__[0]
+            if lastBase is not None and lastBase is not newBase:
+                lastBase.shutdown_class()
+
+            lastBase = newBase
+
+            if filters:
+                matches = False
+
+                for filt in filters:
+                    if fnmatch.fnmatch(o.__name__, filt):
+                        matches = True
+                        break
+
+                if not matches:
+                    continue
+
+            t = o()
+            if t.longTest and not include_long_tests or not t.fixture_available():
+                numSkipped += 1
+                continue
+
+            numTests += 1
+
+            sys.stdout.write("%s:%s... " % (test, o.__name__))
+
+            try:
+                t.captureResults = capture_results
+                t.test()
+            except BaseException:
+                numFailed += 1
+                typ, val, tb = sys.exc_info()
+                fails.append((test, t, typ, val, tb, testModule))
+                sys.stdout.write(colored("FAIL\n", color="red"))
+            else:
+                sys.stdout.write(colored("OK\n", color="green"))
+
+            t.shutdown()
+
+    if lastBase is not None:
+        lastBase.shutdown_class()
+
+    print "TESTS: %d SUCCESS: %d FAIL: %d SKIP: %d" % (numTests, numTests - numFailed, numFailed, numSkipped)
+
+    if len(fails) > 0:
+        print "\nFAILURES (%d):" % (len(fails), )
+
+        for (test, t, typ, val, tb, testModule) in fails:
+            print "%s:%s %s" % (test, t.__class__.__name__, testModule.__doc__.strip() + ": " + t.__doc__.strip())
+            #print "ERROR: %s" % (val, )
+            traceback.print_exception(typ, val, tb)
+            print "=" * 60
+
+        sys.exit(1)
+
+if __name__ == "__main__":
+    if 'APTLY_VERSION' not in os.environ:
+        try:
+            os.environ['APTLY_VERSION'] = os.popen("make version").read().strip()
+        except BaseException, e:
+            print "Failed to capture current version: ", e
+
+    os.chdir(os.path.realpath(os.path.dirname(sys.argv[0])))
+    random.seed()
+    include_long_tests = False
+    capture_results = False
+    tests = None
+    args = sys.argv[1:]
+
+    while len(args) > 0 and args[0].startswith("--"):
+        if args[0] == "--long":
+            include_long_tests = True
+        elif args[0] == "--capture":
+            capture_results = True
+
+        args = args[1:]
+
+    tests = []
+    filters = []
+
+    for arg in args:
+        if arg.startswith('t'):
+            tests.append(arg)
+        else:
+            filters.append(arg)
+
+    run(include_long_tests, capture_results, tests, filters)