1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-22 13:34:15 +03:00
samba-mirror/selftest/selftesthelpers.py

228 lines
7.6 KiB
Python
Raw Normal View History

#!/usr/bin/python
# This script generates a list of testsuites that should be run as part of
# the Samba 4 test suite.
# The output of this script is parsed by selftest.pl, which then decides
# which of the tests to actually run. It will, for example, skip all tests
# listed in selftest/skip or only run a subset during "make quicktest".
# The idea is that this script outputs all of the tests of Samba 4, not
# just those that are known to pass, and list those that should be skipped
# or are known to fail in selftest/skip or selftest/knownfail. This makes it
# very easy to see what functionality is still missing in Samba 4 and makes
# it possible to run the testsuite against other servers, such as Samba 3 or
# Windows that have a different set of features.
# The syntax for a testsuite is "-- TEST --" on a single line, followed
# by the name of the test, the environment it needs and the command to run, all
# three separated by newlines. All other lines in the output are considered
# comments.
import os
import subprocess
import sys
def srcdir():
return os.path.normpath(os.getenv("SRCDIR", os.path.join(os.path.dirname(os.path.abspath(__file__)), "..")))
def source4dir():
return os.path.normpath(os.path.join(srcdir(), "source4"))
def bindir():
return os.path.normpath(os.getenv("BINDIR", "./bin"))
binary_mapping = {}
def binpath(name):
if name in binary_mapping:
name = binary_mapping[name]
return os.path.join(bindir(), name)
binary_mapping_string = os.getenv("BINARY_MAPPING", None)
if binary_mapping_string is not None:
for binmapping_entry in binary_mapping_string.split(','):
try:
(from_path, to_path) = binmapping_entry.split(':', 1)
except ValueError:
continue
binary_mapping[from_path] = to_path
# Split perl variable to allow $PERL to be set to e.g. "perl -W"
perl = os.getenv("PERL", "perl").split()
if subprocess.call(perl + ["-e", "eval require Test::More;"]) == 0:
has_perl_test_more = True
else:
has_perl_test_more = False
try:
from subunit.run import TestProgram
except ImportError:
has_system_subunit_run = False
else:
has_system_subunit_run = True
python = os.getenv("PYTHON", "python")
#Set a default value, overridden if we find a working one on the system
tap2subunit = "PYTHONPATH=%s/lib/subunit/python:%s/lib/testtools %s %s/lib/subunit/filters/tap2subunit" % (srcdir(), srcdir(), python, srcdir())
sub = subprocess.Popen("tap2subunit 2> /dev/null", stdout=subprocess.PIPE, stdin=subprocess.PIPE, shell=True)
sub.communicate("")
if sub.returncode == 0:
cmd = "echo -ne \"1..1\nok 1 # skip doesn't seem to work yet\n\" | tap2subunit 2> /dev/null | grep skip"
sub = subprocess.Popen(cmd, stdout=subprocess.PIPE, stdin=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
if sub.returncode == 0:
tap2subunit = "tap2subunit"
def valgrindify(cmdline):
"""Run a command under valgrind, if $VALGRIND was set."""
valgrind = os.getenv("VALGRIND")
if valgrind is None:
return cmdline
return valgrind + " " + cmdline
def plantestsuite(name, env, cmdline, allow_empty_output=False):
"""Plan a test suite.
:param name: Testsuite name
:param env: Environment to run the testsuite in
:param cmdline: Command line to run
"""
print "-- TEST --"
print name
print env
if isinstance(cmdline, list):
cmdline = " ".join(cmdline)
filter_subunit_args = []
if not allow_empty_output:
filter_subunit_args.append("--fail-on-empty")
if "$LISTOPT" in cmdline:
filter_subunit_args.append("$LISTOPT")
print "%s 2>&1 | %s/selftest/filter-subunit %s --prefix=\"%s.\" --suffix=\"(%s)\"" % (cmdline,
srcdir(),
" ".join(filter_subunit_args),
name, env)
if allow_empty_output:
print >>sys.stderr, "WARNING: allowing empty subunit output from %s" % name
def add_prefix(prefix, env, support_list=False):
if support_list:
listopt = "$LISTOPT "
else:
listopt = ""
return "%s/selftest/filter-subunit %s--fail-on-empty --prefix=\"%s.\" --suffix=\"(%s)\"" % (srcdir(), listopt, prefix, env)
def plantestsuite_loadlist(name, env, cmdline):
print "-- TEST-LOADLIST --"
if env == "none":
fullname = name
else:
fullname = "%s(%s)" % (name, env)
print fullname
print env
if isinstance(cmdline, list):
cmdline = " ".join(cmdline)
support_list = ("$LISTOPT" in cmdline)
print "%s $LOADLIST 2>&1 | %s" % (cmdline, add_prefix(name, env, support_list))
def plantestsuite_idlist(name, env, cmdline):
print "-- TEST-IDLIST --"
if env == "none":
fullname = name
else:
fullname = "%s(%s)" % (name, env)
print fullname
print env
if isinstance(cmdline, list):
cmdline = " ".join(cmdline)
print cmdline
def skiptestsuite(name, reason):
"""Indicate that a testsuite was skipped.
:param name: Test suite name
:param reason: Reason the test suite was skipped
"""
# FIXME: Report this using subunit, but re-adjust the testsuite count somehow
print >>sys.stderr, "skipping %s (%s)" % (name, reason)
def planperltestsuite(name, path):
"""Run a perl test suite.
:param name: Name of the test suite
:param path: Path to the test runner
"""
if has_perl_test_more:
plantestsuite(name, "none", "%s %s | %s" % (" ".join(perl), path, tap2subunit))
else:
skiptestsuite(name, "Test::More not available")
def planpythontestsuite(env, module, name=None, extra_path=[]):
if name is None:
name = module
pypath = list(extra_path)
if not has_system_subunit_run:
pypath.extend(["%s/lib/subunit/python" % srcdir(),
"%s/lib/testtools" % srcdir()])
args = [python, "-m", "subunit.run", "$LISTOPT", module]
if pypath:
2011-11-10 23:44:59 +04:00
args.insert(0, "PYTHONPATH=%s" % ":".join(["$PYTHONPATH"] + pypath))
plantestsuite_idlist(name, env, args)
def get_env_torture_options():
ret = []
if not os.getenv("SELFTEST_VERBOSE"):
ret.append("--option=torture:progress=no")
if os.getenv("SELFTEST_QUICK"):
ret.append("--option=torture:quick=yes")
return ret
samba4srcdir = source4dir()
bbdir = os.path.join(srcdir(), "testprogs/blackbox")
configuration = "--configfile=$SMB_CONF_PATH"
smbtorture4 = binpath("smbtorture4")
smbtorture4_testsuite_list = subprocess.Popen([smbtorture4, "--list-suites"], stdout=subprocess.PIPE, stderr=subprocess.PIPE).communicate("")[0].splitlines()
smbtorture4_options = [
configuration,
"--maximum-runtime=$SELFTEST_MAXTIME",
"--basedir=$SELFTEST_TMPDIR",
"--format=subunit"
] + get_env_torture_options()
def print_smbtorture4_version():
"""Print the version of Samba smbtorture4 comes from.
:return: Whether smbtorture4 was successfully run
"""
sub = subprocess.Popen([smbtorture4, "-V"], stdout=sys.stderr)
sub.communicate("")
return (sub.returncode == 0)
def plansmbtorture4testsuite(name, env, options, target, modname=None):
if modname is None:
modname = "samba4.%s" % name
if isinstance(options, list):
options = " ".join(options)
options = " ".join(smbtorture4_options + ["--target=%s" % target]) + " " + options
cmdline = "%s $LISTOPT %s %s" % (valgrindify(smbtorture4), options, name)
plantestsuite_loadlist(modname, env, cmdline)
def smbtorture4_testsuites(prefix):
return filter(lambda x: x.startswith(prefix), smbtorture4_testsuite_list)