#!python
# Copyright (c) 2002 Trent Mick
#
# Permission is hereby granted, free of charge, to any person obtaining a
# copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be included
# in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
# IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
# CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
# TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
"""
px and p4lib.py Regression Test Suite Harness
Usage:
python test.py [<options>...] [<tests>...]
Options:
-x <testname>, --exclude=<testname>
Exclude the named test from the set of tests to be
run. This can be used multiple times to specify
multiple exclusions.
-v, --verbose run tests in verbose mode with output to stdout
-q, --quiet don't print anything except if a test fails
-h, --help print this text and exit
This will find all modules whose name is "test_*" in the test
directory, and run them. Various command line options provide
additional facilities.
If non-option arguments are present, they are names for tests to run.
If no test names are given, all tests are run.
Test Setup Options:
--p4d=<path-to-p4d> Server executable to use for test server.
-c, --clean Don't setup, just clean up the test workspace.
-n, --no-clean Don't clean up after setting up and running the
test suite.
Because this test suite intends to test scripts interfacing with
Perforce, a test p4d server is setup (with associated test users and
clients), before the tests are run, and shutdown and cleaned up afterwards.
"""
import os
import sys
import getopt
import glob
import time
import types
import tempfile
import unittest
import testsupport
#---- exceptions
class TestError(Exception):
pass
#---- globals
gVerbosity = 2
#---- utility routines
def _rmtreeOnError(rmFunction, filePath, excInfo):
if excInfo[0] == OSError:
# presuming because file is read-only
os.chmod(filePath, 0777)
rmFunction(filePath)
def _rmtree(dirname):
import shutil
shutil.rmtree(dirname, 0, _rmtreeOnError)
def _getAllTests(testDir):
"""Return a list of all tests to run."""
testPyFiles = glob.glob(os.path.join(testDir, "test_*.py"))
modules = [f[:-3] for f in testPyFiles if f and f.endswith(".py")]
packages = []
for f in glob.glob(os.path.join(testDir, "test_*")):
if os.path.isdir(f) and "." not in f:
if os.path.isfile(os.path.join(testDir, f, "__init__.py")):
packages.append(f)
return modules + packages
def _setUp(p4d='p4d'):
print "="*50
print "Setting up test workspace."
# Abort if there is a server running at this port or if the tmp
# working directory exists.
if os.path.exists(testsupport.tmp):
raise TestError("Intended test working dir, '%s', already exists. "\
"Perhaps you need to run 'python test.py -c'."\
% testsupport.tmp)
if sys.platform.startswith('win'):
cmd = 'p4 -u andrew -p %s info > nul 2>&1' % testsupport.p4port
else:
cmd = 'p4 -u andrew -p %s info > /dev/null 2>&1' % testsupport.p4port
if not os.system(cmd):
raise TestError("There is currently a Perforce server running at "\
"the intended test server port, '%s'. "\
"Perhaps you need to run 'python test.py -c'."\
% testsupport.p4port)
# Start the server.
os.makedirs(testsupport.tmp)
os.makedirs(testsupport.p4root)
if sys.platform.startswith('win'):
cmd = 'start "test server" /MIN %s -J journal -L log -p %s -r %s'\
% (p4d, testsupport.p4port, os.path.abspath(testsupport.p4root))
else:
cmd = '%s -J journal -L log -p %s -r %s &'\
% (p4d, testsupport.p4port, os.path.abspath(testsupport.p4root))
os.system(cmd)
print "Starting Perforce server with '%s'..." % cmd
time.sleep(1) # Give it a second to start up.
if gVerbosity >= 3:
os.system('p4 -u andrew -p %s info' % testsupport.p4port)
# Setup work spaces for each user. Use the P4CONFIG mechanism to
# setup easy p4 usage in each home directory (use the same P4CONFIG
# value currently in use, if any, else default to '.p4config').
P4CONFIG = os.environ.get("P4CONFIG", None)
if not P4CONFIG and sys.platform.startswith("win"):
o = os.popen('p4 set P4CONFIG')
line = o.read()
o.close()
if line:
P4CONFIG = line.split()[0].split('=')[1]
if not P4CONFIG:
P4CONFIG = '.p4config'
os.environ['P4CONFIG'] = P4CONFIG
for username, user in testsupport.users.items():
os.makedirs(user['home'])
p4config = """\
P4PORT=%s
P4USER=%s
P4CLIENT=%s
""" % (testsupport.p4port, username, user['client'])
open(os.path.join(user['home'], P4CONFIG), 'w').write(p4config)
data = {'client': user['client'],
'user': username,
'abshome': os.path.abspath(user['home'])}
p4client = """\
Client: %(client)s
Owner: %(user)s
Description:
Created by test harness.
Root: %(abshome)s
View:
//depot/... //%(client)s/...
""" % data
tmpfile = tempfile.mktemp()
open(tmpfile, 'w').write(p4client)
cmd = 'p4 -u andrew -p %s client -i < "%s"'\
% (testsupport.p4port, tmpfile)
retval = os.system(cmd)
if retval:
raise TestError("Problem setting up client for '%s' with '%s'."\
% (username, cmd))
# Put p4lib.py on sys.path.
sys.path.insert(0, os.path.abspath(os.pardir))
print "="*50
def _tearDown(p4d='p4d'):
print "="*50
print "Tearing down test workspace."
if sys.platform.startswith('win'):
cmd = 'p4 -u andrew -p %s info > nul 2>&1' % testsupport.p4port
else:
cmd = 'p4 -u andrew -p %s info > /dev/null 2>&1' % testsupport.p4port
if not os.system(cmd):
print "Stopping test server on port %s." % testsupport.p4port
os.system('p4 -u andrew -p %s admin stop' % testsupport.p4port)
time.sleep(1) # Give it a second to shutdown.
if os.path.exists(testsupport.tmp):
print "Removing working dir: '%s'" % testsupport.tmp
_rmtree(testsupport.tmp)
print "="*50
def test(testModules, testDir=os.curdir, exclude=[]):
"""Run the given regression tests and report the results."""
# Determine the test modules to run.
if not testModules:
testModules = _getAllTests(testDir)
testModules = [t for t in testModules if t not in exclude]
# Aggregate the TestSuite's from each module into one big one.
allSuites = []
for moduleFile in testModules:
module = __import__(moduleFile, globals(), locals(), [])
suite = getattr(module, "suite", None)
if suite is not None:
allSuites.append(suite())
else:
if gVerbosity >= 2:
print "WARNING: module '%s' did not have a suite() method."\
% moduleFile
suite = unittest.TestSuite(allSuites)
# Run the suite.
runner = unittest.TextTestRunner(sys.stdout, verbosity=gVerbosity)
result = runner.run(suite)
#---- mainline
def main(argv):
testDir = os.path.dirname(sys.argv[0])
# parse options
global gVerbosity
try:
opts, testModules = getopt.getopt(sys.argv[1:], 'hvqx:cn',
['help', 'verbose', 'quiet', 'exclude=',
'p4d=', 'clean', 'no-clean'])
except getopt.error, ex:
print "%s: ERROR: %s" % (argv[0], ex)
print __doc__
sys.exit(2)
exclude = []
setupOpts = {}
justClean = 0
clean = 1
for opt, optarg in opts:
if opt in ("-h", "--help"):
print __doc__
sys.exit(0)
elif opt in ("-v", "--verbose"):
gVerbosity += 1
elif opt in ("-q", "--quiet"):
gVerbosity -= 1
elif opt in ("-x", "--exclude"):
exclude.append(optarg)
elif opt in ("--p4d",):
setupOpts["p4d"] = optarg
elif opt in ("-c", "--clean"):
justClean = 1
elif opt in ("-n", "--no-clean"):
clean = 0
retval = None
if not justClean:
_setUp(**setupOpts)
try:
if not justClean:
retval = test(testModules, testDir=testDir, exclude=exclude)
finally:
if clean:
_tearDown(**setupOpts)
return retval
if __name__ == '__main__':
sys.exit( main(sys.argv) )