Update of /cvsroot/docstring/dps/test
In directory usw-pr-cvs1:/tmp/cvs-serv14792/dps/test
Added Files:
UnitTestFolder.py
Log Message:
Copy of restructuredtext's test/UnitTestFolder.py
--- NEW FILE: UnitTestFolder.py ---
#! /usr/bin/env python
"""
:Author: Garth Kidd
:Contact: ga...@de...
:Revision: $Revision: 1.1 $
:Date: $Date: 2001/09/17 04:12:13 $
:Copyright: This module has been placed in the public domain.
"""
import sys, os, getopt, types, unittest, re
__all__ = ( 'debug', 'verbosity', 'main' )
# So that individual test modules can share a bit of state,
# `UnitTestPackage` acts as an intermediary for the following
# variables:
debug = 0
verbosity = 1
USAGE = """\
Usage: test_whatever [options]
Options:
-h, --help Show this message
-v, --verbose Verbose output
-q, --quiet Minimal output
-d, --debug Debug mode
"""
def usageExit(msg=None):
"""Print usage and exit."""
if msg:
print msg
print USAGE
sys.exit(2)
def parseArgs(argv=sys.argv):
"""Parse command line arguments and set TestFramework state.
State is to be acquired by test_* modules by a grotty hack:
``from TestFramework import *``. For this stylistic
transgression, I expect to be first up against the wall
when the revolution comes. --Garth"""
global verbosity, debug
try:
options, args = getopt.getopt(argv[1:], 'hHvqd',
['help', 'verbose', 'quiet', 'debug'])
for opt, value in options:
if opt in ('-h', '-H', '--help'):
usageExit()
if opt in ('-q', '--quiet'):
verbosity = 0
if opt in ('-v', '--verbose'):
verbosity = 2
if opt in ('-d', '--debug'):
debug =1
if len(args) != 0:
usageExit("No command-line arguments supported yet.")
except getopt.error, msg:
self.usageExit(msg)
def loadModulesFromFolder(path, name='', subfolders=None):
"""
Return a test suite composed of all the tests from modules in a folder.
Search for modules in directory `path`, beginning with `name`. If
`subfolders` is true, search subdirectories (also beginning with `name`)
recursively.
"""
testLoader = unittest.defaultTestLoader
testSuite = unittest.TestSuite()
testModules = []
paths = [path]
while paths:
p = paths.pop(0)
if not p:
p = os.curdir
files = os.listdir(p)
for filename in files:
if filename.startswith(name):
fullpath = os.path.join(p, filename)
if filename.endswith('.py'):
testModules.append(fullpath)
elif subfolders and os.path.isdir(fullpath):
paths.append(fullpath)
sys.path.insert(0, '')
# Import modules and add their tests to the suite.
for modpath in testModules:
if debug:
print >>sys.stderr, "importing %s" % modpath
sys.path[0], filename = os.path.split(modpath)
modname = filename[:-3] # strip off the '.py'
module = __import__(modname)
# if there's a suite defined, incorporate its contents
try:
suite = getattr(module, 'suite')
if type(suite) == types.FunctionType:
testSuite.addTest(suite())
elif type(suite) == types.InstanceType \
and isinstance(suite, unittest.TestSuite):
testSuite.addTest(suite)
else:
raise AssertionError, "don't understand suite"
except AttributeError:
# Look for individual tests
moduleTests = testLoader.loadTestsFromModule(module)
# unittest.TestSuite.addTests() doesn't work as advertised,
# as it can't load tests from another TestSuite, so we have
# to cheat:
testSuite.addTest(moduleTests)
return testSuite
def main(suite=None):
"""
Shared `main` for any individual test_* file.
suite -- TestSuite to run. If not specified, look for any globally defined
tests and run them.
"""
parseArgs()
if suite is None:
# Load any globally defined tests.
suite = unittest.defaultTestLoader.loadTestsFromModule(__import__('__main__'))
if debug:
print >>sys.stderr, "Debug: Suite=%s" % suite
testRunner = unittest.TextTestRunner(verbosity=verbosity)
# run suites (if we were called from test_all) or suite...
if type(suite) == type([]):
for s in suite:
testRunner.run(s)
else:
testRunner.run(suite)
|
