From 12d74be7d7f62400663911d94783320bdfba86d4 Mon Sep 17 00:00:00 2001
From: Armin Ronacher <armin.ronacher@active-4.com>
Date: Fri, 26 Aug 2011 11:24:10 +0100
Subject: [PATCH] Added better test runner

---
 run-tests.py | 57 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 run-tests.py

diff --git a/run-tests.py b/run-tests.py
new file mode 100644
index 00000000..96db7f9a
--- /dev/null
+++ b/run-tests.py
@@ -0,0 +1,57 @@
+import sys
+import unittest
+from unittest.loader import TestLoader
+from flask.testsuite import suite
+
+common_prefix = suite.__module__ + '.'
+
+
+def find_all_tests():
+    suites = [suite()]
+    while suites:
+        s = suites.pop()
+        try:
+            suites.extend(s)
+        except TypeError:
+            yield s
+
+
+def find_all_tests_with_name():
+    for testcase in find_all_tests():
+        yield testcase, '%s.%s.%s' % (
+            testcase.__class__.__module__,
+            testcase.__class__.__name__,
+            testcase._testMethodName
+        )
+
+
+class BetterLoader(TestLoader):
+
+    def loadTestsFromName(self, name, module=None):
+        if name == 'suite':
+            return suite()
+        for testcase, testname in find_all_tests_with_name():
+            if testname == name:
+                return testcase
+            if testname.startswith(common_prefix):
+                if testname[len(common_prefix):] == name:
+                    return testcase
+
+        all_results = []
+        for testcase, testname in find_all_tests_with_name():
+            if testname.endswith('.' + name):
+                all_results.append((testcase, testname))
+
+        if len(all_results) == 1:
+            return all_results[0][0]
+        elif not len(all_results):
+            error = 'could not find testcase "%s"' % name
+        else:
+            error = 'Too many matches: for "%s"\n%s' % \
+                (name, '\n'.join('  - ' + n for c, n in all_results))
+
+        print >> sys.stderr, 'Error: %s' % error
+        sys.exit(1)
+
+
+unittest.main(testLoader=BetterLoader(), defaultTest='suite')