2 # vim: expandtab ft=python
6 import Scripting, os, Options, Utils, Environment, optparse, sys
7 from samba_utils import *
8 from samba_autoconf import *
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')
19 opt.ADD_COMMAND('test', cmd_test)
20 opt.ADD_COMMAND('testonly', cmd_testonly)
22 gr = opt.add_option_group('test options')
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('--list',
28 help=("List available tests"),
29 action="store_true", dest='LIST', default=False)
30 gr.add_option('--tests',
31 help=("wildcard pattern of tests to run"),
32 action="store", dest='TESTS', default='')
33 gr.add_option('--target',
34 help=("target of samba3 or samba4"),
35 action="store", dest='TARGET', default=None)
36 gr.add_option('--filtered-subunit',
37 help=("output (xfail) filtered subunit"),
38 action="store_true", dest='FILTERED_SUBUNIT', default=False)
39 gr.add_option('--quick',
40 help=("enable only quick tests"),
41 action="store_true", dest='QUICKTEST', default=False)
42 gr.add_option('--slow',
43 help=("enable the really slow tests"),
44 action="store_true", dest='SLOWTEST', default=False)
45 gr.add_option('--testenv',
46 help=("start a terminal with the test environment setup"),
47 action="store_true", dest='TESTENV', default=False)
48 gr.add_option('--valgrind',
49 help=("use valgrind on client programs in the tests"),
50 action="store_true", dest='VALGRIND', default=False)
51 gr.add_option('--valgrind-log',
52 help=("where to put the valgrind log"),
53 action="store", dest='VALGRINDLOG', default=None)
54 gr.add_option('--valgrind-server',
55 help=("use valgrind on the server in the tests (opens an xterm)"),
56 action="store_true", dest='VALGRIND_SERVER', default=False)
57 gr.add_option('--screen',
58 help=("run the samba servers in screen sessions"),
59 action="store_true", dest='SCREEN', default=False)
60 gr.add_option('--gdbtest',
61 help=("run the servers within a gdb window"),
62 action="store_true", dest='GDBTEST', default=False)
63 gr.add_option('--fail-immediately',
64 help=("stop tests on first failure"),
65 action="store_true", dest='FAIL_IMMEDIATELY', default=False)
66 gr.add_option('--socket-wrapper-pcap',
67 help=("create a pcap file for each failing test"),
68 action="store_true", dest='SOCKET_WRAPPER_PCAP', default=False)
69 gr.add_option('--socket-wrapper-keep-pcap',
70 help=("create a pcap file for all individual test"),
71 action="store_true", dest='SOCKET_WRAPPER_KEEP_PCAP', default=False)
74 conf.env.SELFTEST_PREFIX = Options.options.SELFTEST_PREFIX
76 def cmd_testonly(opt):
77 '''run tests without doing a build first'''
78 env = LOAD_ENVIRONMENT()
81 if (not CONFIG_SET(opt, 'NSS_WRAPPER') or
82 not CONFIG_SET(opt, 'UID_WRAPPER') or
83 not CONFIG_SET(opt, 'SOCKET_WRAPPER')):
84 print("ERROR: You must use --enable-selftest to enable selftest")
87 os.environ['SAMBA_SELFTEST'] = '1'
89 env.TESTS = Options.options.TESTS
91 env.SUBUNIT_FORMATTER = os.getenv('SUBUNIT_FORMATTER')
92 if not env.SUBUNIT_FORMATTER:
93 env.SUBUNIT_FORMATTER = '${PYTHON} -u ${srcdir}/selftest/format-subunit --prefix=${SELFTEST_PREFIX} --immediate'
94 env.FILTER_XFAIL = '${PYTHON} -u ${srcdir}/selftest/filter-subunit --expected-failures=${SELFTEST_DIR}/knownfail'
96 if Options.options.FAIL_IMMEDIATELY:
97 env.FILTER_XFAIL += ' --fail-immediately'
99 env.FORMAT_TEST_OUTPUT = '${SUBUNIT_FORMATTER}'
101 # clean any previous temporary files
102 os.system("rm -rf %s/tmp" % env.SELFTEST_PREFIX);
104 # put all command line options in the environment as TESTENV_*=*
105 for o in dir(Options.options):
107 os.environ['TESTENV_%s' % o.upper()] = str(getattr(Options.options, o, ''))
109 binary_mapping = ('nmblookup3:nmblookup3,' +
110 'smbclient3:smbclient3,' +
111 'smbtorture4:smbtorture,' +
112 'ntlm_auth3:ntlm_auth3')
114 env.OPTIONS = '--binary-mapping=%s' % binary_mapping
115 if not Options.options.SLOWTEST:
116 env.OPTIONS += ' --exclude=${SELFTEST_DIR}/slow'
117 if Options.options.QUICKTEST:
118 env.OPTIONS += ' --quick --include=${SELFTEST_DIR}/quick'
119 if Options.options.LOAD_LIST:
120 env.OPTIONS += ' --load-list=%s' % Options.options.LOAD_LIST
121 if Options.options.TESTENV:
122 env.OPTIONS += ' --testenv'
123 if Options.options.SOCKET_WRAPPER_PCAP:
124 env.OPTIONS += ' --socket-wrapper-pcap'
125 if Options.options.SOCKET_WRAPPER_KEEP_PCAP:
126 env.OPTIONS += ' --socket-wrapper-keep-pcap'
127 if Options.options.LIST:
128 env.OPTIONS += ' --list'
129 if os.environ.get('RUN_FROM_BUILD_FARM') is not None:
130 env.FILTER_OPTIONS = '${FILTER_XFAIL} --strip-passed-output'
132 env.FILTER_OPTIONS = '${FILTER_XFAIL}'
134 if Options.options.VALGRIND:
135 os.environ['VALGRIND'] = 'valgrind -q --num-callers=30'
136 if Options.options.VALGRINDLOG is not None:
137 os.environ['VALGRIND'] += ' --log-file=%s' % Options.options.VALGRINDLOG
141 if Options.options.VALGRIND_SERVER:
142 server_wrapper = '${srcdir}/selftest/valgrind_run _DUMMY=X'
143 elif Options.options.GDBTEST:
144 server_wrapper = '${srcdir}/selftest/gdb_run _DUMMY=X'
146 if Options.options.SCREEN:
147 server_wrapper = '${srcdir}/selftest/in_screen %s' % server_wrapper
148 os.environ['TERMINAL'] = EXPAND_VARIABLES(opt, '${srcdir}/selftest/in_screen')
149 elif server_wrapper != '':
150 server_wrapper = 'xterm -n server -l -e %s' % server_wrapper
152 if server_wrapper != '':
153 os.environ['SAMBA_VALGRIND'] = EXPAND_VARIABLES(opt, server_wrapper)
155 # this is needed for systems without rpath, or with rpath disabled
156 ADD_LD_LIBRARY_PATH('bin/shared')
157 ADD_LD_LIBRARY_PATH('bin/shared/private')
159 # if we are using a system version of ldb then we need to tell it to
160 # load modules from our modules path
161 if env.USING_SYSTEM_LDB:
162 os.environ['LDB_MODULES_PATH'] = 'bin/modules/ldb'
164 # tell build system where to find config.h
165 os.environ['VFSLIBDIR'] = os.path.abspath('bin/modules/vfs')
166 os.environ['CONFIG_H'] = 'bin/default/include/config.h'
168 st_done = os.path.join(env.SELFTEST_PREFIX, 'st_done')
169 if os.path.exists(st_done):
172 if not os.path.isdir(env.SELFTEST_PREFIX):
173 os.makedirs(env.SELFTEST_PREFIX, int('755', 8))
175 if Options.options.TARGET:
176 env.SELFTEST_TARGET = Options.options.TARGET
178 env.SELFTEST_TARGET = "samba4"
180 if env.SELFTEST_TARGET == "samba4":
181 env.SELFTEST_DIR = "${srcdir}/source4/selftest"
182 elif env.SELFTEST_TARGET == "samba3":
183 env.SELFTEST_DIR = "${srcdir}/source3/selftest"
185 # We use the full path rather than relative path because it cause problems on some plateforms (ie. solaris 8).
186 cmd = '(${PERL} ${srcdir}/selftest/selftest.pl --target=${SELFTEST_TARGET} --prefix=${SELFTEST_PREFIX} --srcdir=${srcdir} --exclude=${SELFTEST_DIR}/skip --testlist="${PYTHON} ${SELFTEST_DIR}/tests.py|" ${OPTIONS} --socket-wrapper ${TESTS} && touch ${SELFTEST_PREFIX}/st_done) | ${FILTER_OPTIONS} | tee ${SELFTEST_PREFIX}/subunit'
187 if os.environ.get('RUN_FROM_BUILD_FARM') is None and not Options.options.FILTERED_SUBUNIT:
188 cmd += ' | ${FORMAT_TEST_OUTPUT}'
190 cmd += ' | ${PYTHON} -u ${srcdir}/selftest/filter-subunit'
191 runcmd = EXPAND_VARIABLES(opt, cmd)
193 print("test: running %s" % runcmd)
194 ret = RUN_COMMAND(cmd, env=env)
195 if os.path.exists(".testrepository"):
196 # "testr load -q" isn't
197 testrcmd = 'testr load -q < ${SELFTEST_PREFIX}/subunit > /dev/null'
198 runcmd = EXPAND_VARIABLES(opt, testrcmd)
199 RUN_COMMAND(runcmd, env=env)
202 print("ERROR: test failed with exit code %d" % ret)
205 if not os.path.exists(st_done):
206 print("ERROR: test command failed to complete")
210 ########################################################################
211 # main test entry point
213 '''Run the test suite (see test options below)'''
214 Scripting.commands.append('build')
215 Scripting.commands.append('testonly')