mirror of
https://github.com/henrydcase/pqc.git
synced 2024-11-26 17:31:38 +00:00
75be658074
Modify the tests to no longer construct the classes multiple times and to have them make use of the new helpers.make() function.
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
"""
|
|
Checks that the functional test program (functest) can be successfully built
|
|
and executed for every scheme/implementation.
|
|
"""
|
|
|
|
import os
|
|
import platform
|
|
import unittest
|
|
|
|
import pqclean
|
|
import helpers
|
|
|
|
|
|
def test_functest():
|
|
for scheme in pqclean.Scheme.all_schemes():
|
|
for implementation in scheme.implementations:
|
|
yield check_functest, implementation
|
|
|
|
|
|
def test_functest_sanitizers():
|
|
for scheme in pqclean.Scheme.all_schemes():
|
|
for implementation in scheme.implementations:
|
|
yield check_functest_sanitizers, implementation
|
|
|
|
|
|
def check_functest(implementation):
|
|
helpers.make(TYPE=implementation.scheme.type,
|
|
SCHEME=implementation.scheme.name,
|
|
IMPLEMENTATION=implementation.name,
|
|
working_dir=os.path.join('..', 'test'))
|
|
helpers.run_subprocess(
|
|
['./functest_{}_{}'.format(implementation.scheme.name,
|
|
implementation.name)],
|
|
os.path.join('..', 'bin'),
|
|
)
|
|
|
|
|
|
def check_functest_sanitizers(implementation):
|
|
env = None
|
|
if platform.machine() == 'ppc' and os.environ.get('CC', 'gcc') == 'clang':
|
|
raise unittest.SkipTest()
|
|
elif platform.machine() in ['armv7l', 'aarch64']:
|
|
env = {'ASAN_OPTIONS': 'detect_leaks=0'}
|
|
else:
|
|
print("Supported platform: {}".format(platform.machine()))
|
|
helpers.make('clean-scheme', 'functest',
|
|
TYPE=implementation.scheme.type,
|
|
SCHEME=implementation.scheme.name,
|
|
IMPLEMENTATION=implementation.name,
|
|
EXTRAFLAGS='-fsanitize=address,undefined',
|
|
working_dir=os.path.join('..', 'test'),
|
|
env=env)
|
|
helpers.run_subprocess(
|
|
['./functest_{}_{}'.format(implementation.scheme.name,
|
|
implementation.name)],
|
|
os.path.join('..', 'bin'),
|
|
env=env,
|
|
)
|
|
# Remove files with ASAN library compiled in
|
|
helpers.make('clean-scheme',
|
|
TYPE=implementation.scheme.type,
|
|
SCHEME=implementation.scheme.name,
|
|
IMPLEMENTATION=implementation.name,
|
|
working_dir=os.path.join('..', 'test'))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
import nose2
|
|
nose2.main()
|
|
except ImportError:
|
|
import nose
|
|
nose.runmodule()
|