selftest/wscript

   1 #!/usr/bin/env python
   2 # vim: expandtab ft=python
   3
   4 # selftest main code.
   5
   6 import Scripting, os, Options, Utils, Environment, optparse, sys
   7 from samba_utils import *
   8 from samba_autoconf import *
   9 import types
  10
  11 DEFAULT_SELFTEST_PREFIX="./st"
  12
  13 def set_options(opt):
  14
  15     opt.add_option('--enable-selftest',
  16                    help=("enable options necessary for selftest (default=no)"),
  17                    action="store_true", dest='enable_selftest', default=False)
  18     opt.add_option('--enable-coverage',
  19                    help=("enable options necessary for code coverage "
  20                          "reporting on selftest (default=no)"),
  21                    action="store_true", dest='enable_coverage', default=False)
  22     opt.add_option('--with-selftest-prefix',
  23                    help=("specify location of selftest directory "
  24                          "(default=%s)" % DEFAULT_SELFTEST_PREFIX),
  25                    action="store", dest='SELFTEST_PREFIX', default=DEFAULT_SELFTEST_PREFIX)
  26
  27     opt.ADD_COMMAND('test', cmd_test)
  28     opt.ADD_COMMAND('testonly', cmd_testonly)
  29
  30     gr = opt.add_option_group('test options')
  31
  32     gr.add_option('--load-list',
  33                   help=("Load a test id list from a text file"),
  34                   action="store", dest='LOAD_LIST', default=None)
  35     gr.add_option('--list',
  36                   help=("List available tests"),
  37                   action="store_true", dest='LIST', default=False)
  38     gr.add_option('--tests',
  39                   help=("wildcard pattern of tests to run"),
  40                   action="store", dest='TESTS', default='')
  41     gr.add_option('--filtered-subunit',
  42                   help=("output (xfail) filtered subunit"),
  43                   action="store_true", dest='FILTERED_SUBUNIT', default=False)
  44     gr.add_option('--quick',
  45                   help=("enable only quick tests"),
  46                   action="store_true", dest='QUICKTEST', default=False)
  47     gr.add_option('--slow',
  48                   help=("enable the really slow tests"),
  49                   action="store_true", dest='SLOWTEST', default=False)
  50     gr.add_option('--nb-slowest',
  51                   help=("Show the n slowest tests (default=10)"),
  52                   type=int, default=10, dest='NB_SLOWEST')
  53     gr.add_option('--testenv',
  54                   help=("start a terminal with the test environment setup"),
  55                   action="store_true", dest='TESTENV', default=False)
  56     gr.add_option('--valgrind',
  57                   help=("use valgrind on client programs in the tests"),
  58                   action="store_true", dest='VALGRIND', default=False)
  59     gr.add_option('--valgrind-log',
  60                   help=("where to put the valgrind log"),
  61                   action="store", dest='VALGRINDLOG', default=None)
  62     gr.add_option('--valgrind-server',
  63                   help=("use valgrind on the server in the tests (opens an xterm)"),
  64                   action="store_true", dest='VALGRIND_SERVER', default=False)
  65     gr.add_option('--screen',
  66                   help=("run the samba servers in screen sessions"),
  67                   action="store_true", dest='SCREEN', default=False)
  68     gr.add_option('--gdbtest',
  69                   help=("run the servers within a gdb window"),
  70                   action="store_true", dest='GDBTEST', default=False)
  71     gr.add_option('--fail-immediately',
  72                   help=("stop tests on first failure"),
  73                   action="store_true", dest='FAIL_IMMEDIATELY', default=False)
  74     gr.add_option('--socket-wrapper-pcap',
  75                   help=("create a pcap file for each failing test"),
  76                   action="store_true", dest='SOCKET_WRAPPER_PCAP', default=False)
  77     gr.add_option('--socket-wrapper-keep-pcap',
  78                   help=("create a pcap file for all individual test"),
  79                   action="store_true", dest='SOCKET_WRAPPER_KEEP_PCAP', default=False)
  80     gr.add_option('--random-order', dest='RANDOM_ORDER', default=False,
  81                   action="store_true", help="Run testsuites in random order")
  82     gr.add_option('--perf-test', dest='PERF_TEST', default=False,
  83                   action="store_true", help="run performance tests only")
  84     gr.add_option('--test-list', dest='TEST_LIST', default='',
  85                   action="store_true",
  86                   help=("use tests listed here, not defaults "
  87                         "(--test-list='FOO|' will execute FOO; "
  88                         "--test-list='FOO' will read it)"))
  89
  90 def configure(conf):
  91     conf.env.SELFTEST_PREFIX = Options.options.SELFTEST_PREFIX
  92     conf.env.enable_coverage = Options.options.enable_coverage
  93     if conf.env.enable_coverage:
  94         conf.ADD_LDFLAGS('-lgcov', testflags=True)
  95         conf.ADD_CFLAGS('--coverage')
  96
  97     if Options.options.enable_selftest or Options.options.developer:
  98         conf.DEFINE('ENABLE_SELFTEST', 1)
  99
 100
 101 def cmd_testonly(opt):
 102     '''run tests without doing a build first'''
 103     env = LOAD_ENVIRONMENT()
 104     opt.env = env
 105
 106     if Options.options.SELFTEST_PREFIX != DEFAULT_SELFTEST_PREFIX:
 107         env.SELFTEST_PREFIX = Options.options.SELFTEST_PREFIX
 108
 109     if (not CONFIG_SET(opt, 'NSS_WRAPPER') or
 110         not CONFIG_SET(opt, 'UID_WRAPPER') or
 111         not CONFIG_SET(opt, 'SOCKET_WRAPPER')):
 112         print("ERROR: You must use --enable-selftest to enable selftest")
 113         sys.exit(1)
 114
 115     os.environ['SAMBA_SELFTEST'] = '1'
 116
 117     env.TESTS  = Options.options.TESTS
 118
 119     env.SUBUNIT_FORMATTER = os.getenv('SUBUNIT_FORMATTER')
 120     if not env.SUBUNIT_FORMATTER:
 121         if Options.options.PERF_TEST:
 122             env.SUBUNIT_FORMATTER = '${PYTHON} -u ${srcdir}/selftest/format-subunit-json --prefix=${SELFTEST_PREFIX}'
 123         else:
 124             env.SUBUNIT_FORMATTER = '${PYTHON} -u ${srcdir}/selftest/format-subunit --prefix=${SELFTEST_PREFIX} --immediate'
 125     env.FILTER_XFAIL = '${PYTHON} -u ${srcdir}/selftest/filter-subunit --expected-failures=${srcdir}/selftest/knownfail --flapping=${srcdir}/selftest/flapping'
 126
 127     if Options.options.FAIL_IMMEDIATELY:
 128         env.FILTER_XFAIL += ' --fail-immediately'
 129
 130     env.FORMAT_TEST_OUTPUT = '${SUBUNIT_FORMATTER}'
 131
 132     # clean any previous temporary files
 133     os.system("rm -rf %s/tmp" % env.SELFTEST_PREFIX);
 134
 135     # put all command line options in the environment as TESTENV_*=*
 136     for o in dir(Options.options):
 137         if o[0:1] != '_':
 138             val = getattr(Options.options, o, '')
 139             if not issubclass(type(val), types.FunctionType) \
 140                     and not issubclass(type(val), types.MethodType):
 141                 os.environ['TESTENV_%s' % o.upper()] = str(getattr(Options.options, o, ''))
 142
 143     env.OPTIONS = ''
 144     if not Options.options.SLOWTEST:
 145         env.OPTIONS += ' --exclude=${srcdir}/selftest/slow'
 146     if Options.options.QUICKTEST:
 147         env.OPTIONS += ' --quick --include=${srcdir}/selftest/quick'
 148     if Options.options.LOAD_LIST:
 149         env.OPTIONS += ' --load-list=%s' % Options.options.LOAD_LIST
 150     if Options.options.TESTENV:
 151         env.OPTIONS += ' --testenv'
 152     if Options.options.SOCKET_WRAPPER_PCAP:
 153         env.OPTIONS += ' --socket-wrapper-pcap'
 154     if Options.options.SOCKET_WRAPPER_KEEP_PCAP:
 155         env.OPTIONS += ' --socket-wrapper-keep-pcap'
 156     if Options.options.RANDOM_ORDER:
 157         env.OPTIONS += ' --random-order'
 158     if Options.options.PERF_TEST:
 159         env.FILTER_OPTIONS = ('${PYTHON} -u ${srcdir}/selftest/filter-subunit '
 160                               '--perf-test-output')
 161     elif os.environ.get('RUN_FROM_BUILD_FARM') is not None:
 162         env.FILTER_OPTIONS = '${FILTER_XFAIL} --strip-passed-output'
 163     else:
 164         env.FILTER_OPTIONS = '${FILTER_XFAIL}'
 165
 166     if Options.options.VALGRIND:
 167         os.environ['VALGRIND'] = 'valgrind -q --num-callers=30'
 168         if Options.options.VALGRINDLOG is not None:
 169             os.environ['VALGRIND'] += ' --log-file=%s' % Options.options.VALGRINDLOG
 170
 171     server_wrapper=''
 172
 173     if Options.options.VALGRIND_SERVER:
 174         server_wrapper = '${srcdir}/selftest/valgrind_run _DUMMY=X'
 175     elif Options.options.GDBTEST:
 176         server_wrapper = '${srcdir}/selftest/gdb_run _DUMMY=X'
 177
 178     if Options.options.SCREEN:
 179         server_wrapper = '${srcdir}/selftest/in_screen %s' % server_wrapper
 180         os.environ['TERMINAL'] = EXPAND_VARIABLES(opt, '${srcdir}/selftest/in_screen')
 181     elif server_wrapper != '':
 182         server_wrapper = 'xterm -n server -l -e %s' % server_wrapper
 183
 184     if server_wrapper != '':
 185         os.environ['SAMBA_VALGRIND'] = EXPAND_VARIABLES(opt, server_wrapper)
 186         os.environ['NMBD_VALGRIND'] = EXPAND_VARIABLES(opt, server_wrapper)
 187         os.environ['WINBINDD_VALGRIND'] = EXPAND_VARIABLES(opt, server_wrapper)
 188         os.environ['SMBD_VALGRIND'] = EXPAND_VARIABLES(opt, server_wrapper)
 189
 190     # this is needed for systems without rpath, or with rpath disabled
 191     ADD_LD_LIBRARY_PATH('bin/shared')
 192     ADD_LD_LIBRARY_PATH('bin/shared/private')
 193
 194     # if we are using a system version of ldb then we need to tell it to
 195     # load modules from our modules path
 196     if env.USING_SYSTEM_LDB:
 197         os.environ['LDB_MODULES_PATH'] = os.path.abspath(os.path.join(env.cwd, 'bin/modules/ldb'))
 198
 199     # tell build system where to find config.h
 200     os.environ['CONFIG_H'] = 'bin/default/include/config.h'
 201
 202     st_done = os.path.join(env.SELFTEST_PREFIX, 'st_done')
 203     if os.path.exists(st_done):
 204         os.unlink(st_done)
 205
 206     if not os.path.isdir(env.SELFTEST_PREFIX):
 207         os.makedirs(env.SELFTEST_PREFIX, int('755', 8))
 208
 209     if Options.options.TEST_LIST:
 210         env.TESTLISTS = '--testlist=%r' % Options.options.TEST_LIST
 211     elif Options.options.PERF_TEST:
 212         env.TESTLISTS = '--testlist="${PYTHON} ${srcdir}/selftest/perf_tests.py|" '
 213     else:
 214         env.TESTLISTS = ('--testlist="${PYTHON} ${srcdir}/selftest/tests.py|" ' +
 215                          '--testlist="${PYTHON} ${srcdir}/source3/selftest/tests.py|" ' +
 216                          '--testlist="${PYTHON} ${srcdir}/source4/selftest/tests.py|"')
 217
 218     if CONFIG_SET(opt, 'AD_DC_BUILD_IS_ENABLED'):
 219         env.SELFTEST_TARGET = "samba"
 220     else:
 221         env.SELFTEST_TARGET = "samba3"
 222
 223     env.OPTIONS += " --nss_wrapper_so_path=" + CONFIG_GET(opt, 'LIBNSS_WRAPPER_SO_PATH')
 224     env.OPTIONS += " --resolv_wrapper_so_path=" + CONFIG_GET(opt, 'LIBRESOLV_WRAPPER_SO_PATH')
 225     env.OPTIONS += " --socket_wrapper_so_path=" + CONFIG_GET(opt, 'LIBSOCKET_WRAPPER_SO_PATH')
 226     env.OPTIONS += " --uid_wrapper_so_path=" + CONFIG_GET(opt, 'LIBUID_WRAPPER_SO_PATH')
 227
 228     #if unversioned_sys_platform in ('freebsd', 'netbsd', 'openbsd', 'sunos'):
 229     #    env.OPTIONS += " --use-dns-faking"
 230
 231     # FIXME REMOVE ME!
 232     env.OPTIONS += " --use-dns-faking"
 233
 234     if not CONFIG_GET(opt, 'HAVE_GSS_KRB5_CRED_NO_CI_FLAGS_X'):
 235         # older MIT krb5 libraries (< 1.14) don't have
 236         # GSS_KRB5_CRED_NO_CI_FLAGS_X
 237         env.OPTIONS += " --exclude=${srcdir}/selftest/skip.no-GSS_KRB5_CRED_NO_CI_FLAGS_X"
 238
 239     subunit_cache = None
 240     # We use the full path rather than relative path to avoid problems on some platforms (ie. solaris 8).
 241     env.CORE_COMMAND = '${PERL} ${srcdir}/selftest/selftest.pl --target=${SELFTEST_TARGET} --prefix=${SELFTEST_PREFIX} --srcdir=${srcdir} --exclude=${srcdir}/selftest/skip ${TESTLISTS} ${OPTIONS} ${TESTS}'
 242     if Options.options.LIST:
 243         cmd = '${CORE_COMMAND} --list'
 244     else:
 245         env.OPTIONS += ' --socket-wrapper'
 246         cmd = '(${CORE_COMMAND} && touch ${SELFTEST_PREFIX}/st_done) | ${FILTER_OPTIONS}'
 247         if (os.environ.get('RUN_FROM_BUILD_FARM') is None and
 248             not Options.options.FILTERED_SUBUNIT):
 249             subunit_cache = os.path.join(env.SELFTEST_PREFIX, "subunit")
 250             cmd += ' | tee %s | ${FORMAT_TEST_OUTPUT}' % subunit_cache
 251         else:
 252             cmd += ' | ${FILTER_OPTIONS}'
 253     runcmd = EXPAND_VARIABLES(opt, cmd)
 254
 255     print("test: running %s" % runcmd)
 256     ret = RUN_COMMAND(cmd, env=env)
 257
 258     if (os.path.exists(".testrepository") and
 259         not Options.options.LIST and
 260         not Options.options.LOAD_LIST and
 261         subunit_cache is not None):
 262         testrcmd = 'testr load -q < %s > /dev/null' % subunit_cache
 263         runcmd = EXPAND_VARIABLES(opt, testrcmd)
 264         RUN_COMMAND(runcmd, env=env)
 265
 266     if subunit_cache is not None:
 267         nb = Options.options.NB_SLOWEST
 268         cmd = "./script/show_testsuite_time %s %d" % (subunit_cache, nb)
 269         runcmd = EXPAND_VARIABLES(opt, cmd)
 270         RUN_COMMAND(runcmd, env=env)
 271
 272     if ret != 0:
 273         print("ERROR: test failed with exit code %d" % ret)
 274         sys.exit(ret)
 275
 276     if not Options.options.LIST and not os.path.exists(st_done):
 277         print("ERROR: test command failed to complete")
 278         sys.exit(1)
 279
 280
 281 ########################################################################
 282 # main test entry point
 283 def cmd_test(opt):
 284     '''Run the test suite (see test options below)'''
 285
 286     # if running all tests, then force a symbol check
 287     env = LOAD_ENVIRONMENT()
 288     CHECK_MAKEFLAGS(env)
 289     Scripting.commands.append('build')
 290     Scripting.commands.append('testonly')