Reworked test.py to handle deeper directory structure

1 files changed, 65 insertions, 40 deletions

diff --git a/test.py b/test.py
index 4f20808..2b93bcf 100644
--- a/test.py
+++ b/test.py
@@ -1,54 +1,79 @@
 """Usage: python test.py [module(s) ...]
 
-When called without optional module names, run the doctests from *all*
-modules.  This may raise lots of errors if you haven't installed one
-of the versioning control systems.
+When called without optional module names, run the test suites for
+*all* modules.  This may raise lots of errors if you haven't installed
+one of the versioning control systems.
 
-When called with module name arguments, only run the doctests from
-those modules.
+When called with module name arguments, only run the test suites from
+those modules and their submodules.  For example:
+  python test.py libbe.bugdir libbe.storage
 """
 
-import libbe
-libbe.TESTING = True
-from libbe import plugin, vcs
-import unittest
 import doctest
+import os
+import os.path
 import sys
+import unittest
 
-suite = unittest.TestSuite()
+import libbe
+libbe.TESTING = True
+from libbe.util.tree import Tree
+from libbe.util.plugin import import_by_name
 
-if len(sys.argv) > 1:
-    for submodname in sys.argv[1:]:
-        match = False
-        mod = plugin.get_plugin("libbe", submodname)
-        if mod is not None:
-            if hasattr(mod, "suite"):
-                suite.addTest(mod.suite)
-                match = True
-            else:
-                print "Module \"%s\" has no test suite" % submodname
-        mod = plugin.get_plugin("becommands", submodname)
-        if mod is not None:
-            if hasattr(mod, "suite"):
-                suite.addTest(mod.suite)
-            else:
-                suite.addTest(doctest.DocTestSuite(mod))
-            match = True
-        if not match:
-            print "No modules match \"%s\"" % submodname
-            sys.exit(1)
-else:
-    failed = False
-    for modname,module in plugin.iter_plugins("libbe"):
-        if not hasattr(module, "suite"):
+def python_tree(root_path='libbe', root_modname='libbe'):
+    tree = Tree()
+    tree.path = root_path
+    tree.parent = None
+    stack = [tree]
+    while len(stack) > 0:
+        f = stack.pop(0)
+        if f.path.endswith('.py'):
+            f.name = os.path.basename(f.path)[:-len('.py')]
+        elif os.path.isdir(f.path) \
+                and os.path.exists(os.path.join(f.path, '__init__.py')):
+            f.name = os.path.basename(f.path)
+            f.is_module = True
+            for child in os.listdir(f.path):
+                if child == '__init__.py':
+                    continue
+                c = Tree()
+                c.path = os.path.join(f.path, child)
+                c.parent = f
+                stack.append(c)
+        else:
             continue
-        suite.addTest(module.suite)
-    for modname,module in plugin.iter_plugins("becommands"):
-        suite.addTest(doctest.DocTestSuite(module))
+        if f.parent == None:
+            f.modname = root_modname
+        else:
+            f.modname = f.parent.modname + '.' + f.name
+            f.parent.append(f)
+    return tree
+
+def add_module_tests(suite, module_name):
+    mod = import_by_name(module_name)
+    if mod == None:
+        raise KeyError, 'module "%s" not found' % module_name
+    if hasattr(mod, 'suite'):
+        s = mod.suite
+    else:
+        s = unittest.TestLoader().loadTestsFromModule(mod)
+    suite.addTest(s)
 
-_vcs = vcs.installed_vcs()
-vcs.set_preferred_vcs(_vcs.name)
-print 'Using %s as the testing VCS' % _vcs.name
+suite = unittest.TestSuite()
+tree = python_tree()
+if len(sys.argv) <= 1:
+    for node in tree.traverse():
+        add_module_tests(suite, node.modname)
+else:
+    added = []
+    for modname in sys.argv[1:]:
+        for node in tree.traverse():
+            if node.modname == modname:
+                for n in node.traverse():
+                    if n.modname not in added:
+                        add_module_tests(suite, n.modname)
+                        added.append(n.modname)
+                break
 
 result = unittest.TextTestRunner(verbosity=2).run(suite)