From 3e6096fb5bcb9c9e8a50faa76461da96d145ca8f Mon Sep 17 00:00:00 2001 From: "W. Trevor King" Date: Tue, 8 Dec 2009 20:02:34 -0500 Subject: [PATCH] Reworked test.py to handle deeper directory structure --- libbe/bug.py | 2 +- libbe/bugdir.py | 2 +- libbe/comment.py | 2 +- libbe/{ => util}/plugin.py | 2 +- test.py | 105 +++++++++++++++++++++++-------------- 5 files changed, 69 insertions(+), 44 deletions(-) rename libbe/{ => util}/plugin.py (98%) diff --git a/libbe/bug.py b/libbe/bug.py index 84eb1ed..5755b63 100644 --- a/libbe/bug.py +++ b/libbe/bug.py @@ -34,7 +34,7 @@ import xml.sax.saxutils import libbe import libbe.util.id -from libbe.storage.util.properties import Property, doc_property, +from libbe.storage.util.properties import Property, doc_property, \ local_property, defaulting_property, checked_property, cached_property, \ primed_property, change_hook_property, settings_property import libbe.storage.settings_object as settings_object diff --git a/libbe/bugdir.py b/libbe/bugdir.py index 4435624..02ff96f 100644 --- a/libbe/bugdir.py +++ b/libbe/bugdir.py @@ -32,7 +32,7 @@ import time import libbe import libbe.util.encoding as encoding import libbe.storage as storage -from libbe.storage.util.properties import Property, doc_property, +from libbe.storage.util.properties import Property, doc_property, \ local_property, defaulting_property, checked_property, \ fn_checked_property, cached_property, primed_property, \ change_hook_property, settings_property diff --git a/libbe/comment.py b/libbe/comment.py index 987e39c..f5a6309 100644 --- a/libbe/comment.py +++ b/libbe/comment.py @@ -34,7 +34,7 @@ import xml.sax.saxutils import libbe import libbe.util.id -from libbe.storage.util.properties import Property, doc_property, +from libbe.storage.util.properties import Property, doc_property, \ local_property, defaulting_property, checked_property, cached_property, \ primed_property, change_hook_property, settings_property import libbe.storage.settings_object as settings_object diff --git a/libbe/plugin.py b/libbe/util/plugin.py similarity index 98% rename from libbe/plugin.py rename to libbe/util/plugin.py index 03f68fc..edb4922 100644 --- a/libbe/plugin.py +++ b/libbe/util/plugin.py @@ -30,7 +30,7 @@ import libbe if libbe.TESTING == True: import doctest -def my_import(mod_name): +def import_by_name(mod_name): module = __import__(mod_name) components = mod_name.split('.') for comp in components[1:]: 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) -- 2.26.2