source4/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
  10 def set_options(opt):
  11
  12     opt.add_option('--enable-selftest',
  13                    help=("enable options necessary for selftest (default=no)"),
  14                    action="store_true", dest='enable_selftest', default=False)
  15     opt.add_option('--with-selftest-prefix',
  16                    help=("specify location of selftest directory (default=./st)"),
  17                    action="store", dest='SELFTEST_PREFIX', default='./st')
  18
  19     opt.ADD_COMMAND('test', cmd_test)
  20     opt.ADD_COMMAND('testonly', cmd_testonly)
  21
  22     gr = opt.add_option_group('test options')
  23
  24     gr.add_option('--load-list',
  25                   help=("Load a test id list from a text file"),
  26                   action="store", dest='LOAD_LIST', default=None)
  27     gr.add_option('--tests',
  28                   help=("wildcard pattern of tests to run"),
  29                   action="store", dest='TESTS', default='')
  30     gr.add_option('--filtered-subunit',
  31                   help=("output (xfail) filtered subunit"),
  32                   action="store_true", dest='FILTERED_SUBUNIT', default=False)
  33     gr.add_option('--quick',
  34                   help=("enable only quick tests"),
  35                   action="store_true", dest='QUICKTEST', default=False)
  36     gr.add_option('--slow',
  37                   help=("enable the really slow tests"),
  38                   action="store_true", dest='SLOWTEST', default=False)
  39     gr.add_option('--testenv',
  40                   help=("start a terminal with the test environment setup"),
  41                   action="store_true", dest='TESTENV', default=False)
  42     gr.add_option('--valgrind',
  43                   help=("use valgrind on client programs in the tests"),
  44                   action="store_true", dest='VALGRIND', default=False)
  45     gr.add_option('--valgrind-log',
  46                   help=("where to put the valgrind log"),
  47                   action="store", dest='VALGRINDLOG', default=None)
  48     gr.add_option('--valgrind-server',
  49                   help=("use valgrind on the server in the tests (opens an xterm)"),
  50                   action="store_true", dest='VALGRIND_SERVER', default=False)
  51     gr.add_option('--screen',
  52                   help=("run the samba servers in screen sessions"),
  53                   action="store_true", dest='SCREEN', default=False)
  54     gr.add_option('--gdbtest',
  55                   help=("run the servers within a gdb window"),
  56                   action="store_true", dest='GDBTEST', default=False)
  57     gr.add_option('--fail-immediately',
  58                   help=("stop tests on first failure"),
  59                   action="store_true", dest='FAIL_IMMEDIATELY', default=False)
  60     gr.add_option('--socket-wrapper-pcap',
  61                   help=("create a pcap file for each failing test"),
  62                   action="store_true", dest='SOCKET_WRAPPER_PCAP', default=False)
  63     gr.add_option('--socket-wrapper-keep-pcap',
  64                   help=("create a pcap file for all individual test"),
  65                   action="store_true", dest='SOCKET_WRAPPER_KEEP_PCAP', default=False)
  66
  67 def configure(conf):
  68     conf.env.SELFTEST_PREFIX = Options.options.SELFTEST_PREFIX
  69
  70 def cmd_testonly(opt):
  71     '''run tests without doing a build first'''
  72     env = LOAD_ENVIRONMENT()
  73     opt.env = env
  74
  75     if (not CONFIG_SET(opt, 'NSS_WRAPPER') or
  76         not CONFIG_SET(opt, 'UID_WRAPPER') or
  77         not CONFIG_SET(opt, 'SOCKET_WRAPPER')):
  78         print("ERROR: You must use --enable-selftest to enable selftest")
  79         sys.exit(1)
  80
  81     env.TESTS  = Options.options.TESTS
  82
  83     env.SUBUNIT_FORMATTER = '${PYTHON} -u ../selftest/format-subunit --prefix=${SELFTEST_PREFIX} --immediate'
  84     env.FILTER_XFAIL = '${PYTHON} -u ../selftest/filter-subunit --expected-failures=./selftest/knownfail'
  85
  86     if Options.options.FAIL_IMMEDIATELY:
  87         env.FILTER_XFAIL += ' --fail-immediately'
  88
  89     env.FORMAT_TEST_OUTPUT = '${SUBUNIT_FORMATTER}'
  90
  91     # put all command line options in the environment as TESTENV_*=*
  92     for o in dir(Options.options):
  93         if o[0:1] != '_':
  94             os.environ['TESTENV_%s' % o.upper()] = str(getattr(Options.options, o, ''))
  95
  96     env.OPTIONS = ''
  97     if not Options.options.SLOWTEST:
  98         env.OPTIONS += ' --exclude=./selftest/slow'
  99     if Options.options.QUICKTEST:
 100         env.OPTIONS += ' --quick --include=./selftest/quick'
 101     if Options.options.LOAD_LIST:
 102         env.OPTIONS += ' --load-list=%s' % Options.options.LOAD_LIST
 103     if Options.options.TESTENV:
 104         env.OPTIONS += ' --testenv'
 105     if Options.options.SOCKET_WRAPPER_PCAP:
 106         env.OPTIONS += ' --socket-wrapper-pcap'
 107     if Options.options.SOCKET_WRAPPER_KEEP_PCAP:
 108         env.OPTIONS += ' --socket-wrapper-keep-pcap'
 109
 110     if os.environ.get('RUN_FROM_BUILD_FARM') is not None:
 111         env.FILTER_OPTIONS = '${FILTER_XFAIL} --strip-passed-output'
 112     else:
 113         env.FILTER_OPTIONS = '${FILTER_XFAIL}'
 114
 115     if Options.options.VALGRIND:
 116         os.environ['VALGRIND'] = 'valgrind -q --num-callers=30'
 117         if Options.options.VALGRINDLOG is not None:
 118             os.environ['VALGRIND'] += ' --log-file=%s' % Options.options.VALGRINDLOG
 119
 120     server_wrapper=''
 121
 122     if Options.options.VALGRIND_SERVER:
 123         server_wrapper = '../selftest/valgrind_run _DUMMY=X'
 124     elif Options.options.GDBTEST:
 125         server_wrapper = '../selftest/gdb_run _DUMMY=X'
 126
 127     if Options.options.SCREEN:
 128         server_wrapper = '../selftest/in_screen %s' % server_wrapper
 129     elif server_wrapper != '':
 130         server_wrapper = 'xterm -n server -l -e %s' % server_wrapper
 131
 132     if server_wrapper != '':
 133         os.environ['SAMBA_VALGRIND'] = server_wrapper
 134
 135     # this is needed for systems without rpath, or with rpath disabled
 136     ADD_LD_LIBRARY_PATH('bin/shared')
 137
 138     # tell build system where to find config.h
 139     os.environ['CONFIG_H'] = 'bin/default/source4/include/config.h'
 140
 141     st_done = os.path.join(env.SELFTEST_PREFIX, 'st_done')
 142     if os.path.exists(st_done):
 143         os.unlink(st_done)
 144
 145     cmd = '(${PERL} ../selftest/selftest.pl --prefix=${SELFTEST_PREFIX} --builddir=. --srcdir=. --exclude=./selftest/skip --testlist="${PYTHON} ./selftest/tests.py|" ${OPTIONS} --socket-wrapper ${TESTS} && touch ${SELFTEST_PREFIX}/st_done) | ${FILTER_OPTIONS} | tee ${SELFTEST_PREFIX}/subunit'
 146     if os.environ.get('RUN_FROM_BUILD_FARM') is None and not Options.options.FILTERED_SUBUNIT:
 147         cmd += ' | ${FORMAT_TEST_OUTPUT}'
 148     cmd = EXPAND_VARIABLES(opt, cmd)
 149
 150     print("test: running %s" % cmd)
 151     ret = RUN_COMMAND(cmd, env=env)
 152     if os.path.exists(".testrepository"):
 153         # "testr load -q" isn't
 154         cmd = 'testr load -q < ${SELFTEST_PREFIX}/subunit > /dev/null'
 155         cmd = EXPAND_VARIABLES(opt, cmd)
 156         RUN_COMMAND(cmd, env=env)
 157
 158     if ret != 0:
 159         print("ERROR: test failed with exit code %d" % ret)
 160         sys.exit(ret)
 161
 162     if not os.path.exists(st_done):
 163         print("ERROR: test command failed to complete")
 164         sys.exit(1)
 165
 166
 167 ########################################################################
 168 # main test entry point
 169 def cmd_test(opt):
 170     '''Run the test suite (see test options below)'''
 171     Scripting.commands.append('build')
 172     Scripting.commands.append('testonly')