mirror of
https://github.com/mongodb/mongo.git
synced 2024-12-01 09:32:32 +01:00
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
'''Pseudo-builders for building and registering unit tests.'''
|
|
from SCons.Script import Action
|
|
|
|
|
|
def exists(env):
|
|
return True
|
|
|
|
|
|
_unittests = []
|
|
def register_unit_test(env, test):
|
|
_unittests.append(test.path)
|
|
env.Alias('$UNITTEST_ALIAS', test)
|
|
|
|
def unit_test_list_builder_action(env, target, source):
|
|
ofile = open(str(target[0]), 'w')
|
|
try:
|
|
for s in _unittests:
|
|
print('\t' + str(s))
|
|
ofile.write('%s\n' % s)
|
|
finally:
|
|
ofile.close()
|
|
|
|
def build_cpp_unit_test(env, target, source, **kwargs):
|
|
libdeps = kwargs.get('LIBDEPS', [])
|
|
libdeps.append( '$BUILD_DIR/mongo/unittest/unittest_main' )
|
|
|
|
kwargs['LIBDEPS'] = libdeps
|
|
unit_test_components = {'tests', 'unittests'}
|
|
if (
|
|
'AIB_COMPONENT' in kwargs
|
|
and not kwargs['AIB_COMPONENT'].endswith('-test')
|
|
):
|
|
kwargs['AIB_COMPONENT'] += '-test'
|
|
|
|
if 'AIB_COMPONENTS_EXTRA' in kwargs:
|
|
kwargs['AIB_COMPONENTS_EXTRA'] = set(kwargs['AIB_COMPONENTS_EXTRA']).union(unit_test_components)
|
|
else:
|
|
kwargs['AIB_COMPONENTS_EXTRA'] = unit_test_components
|
|
|
|
result = env.Program(target, source, **kwargs)
|
|
env.RegisterUnitTest(result[0])
|
|
hygienic = env.GetOption('install-mode') == 'hygienic'
|
|
if not hygienic:
|
|
env.Install('#/build/unittests/', result[0])
|
|
return result
|
|
|
|
|
|
def generate(env):
|
|
env.Command('$UNITTEST_LIST', env.Value(_unittests),
|
|
Action(unit_test_list_builder_action, 'Generating $TARGET'))
|
|
env.AddMethod(register_unit_test, 'RegisterUnitTest')
|
|
env.AddMethod(build_cpp_unit_test, 'CppUnitTest')
|
|
env.Alias('$UNITTEST_ALIAS', '$UNITTEST_LIST')
|