diff options
Diffstat (limited to 'test/file.py')
| -rw-r--r-- | test/file.py | 53 |
1 files changed, 27 insertions, 26 deletions
diff --git a/test/file.py b/test/file.py index 3f07aaa..a918984 100644 --- a/test/file.py +++ b/test/file.py @@ -3,6 +3,7 @@ # See LICENSE.txt for details. from datetime import datetime +from enum import Enum from glob import iglob as glob import filecmp import logging @@ -13,8 +14,8 @@ from tempfile import TemporaryDirectory from toolkit import * -class _TestExitCode: - SUCCESS, FAILURE, ERROR, SKIPPED = range(4) +class TestExitCode(Enum): + SUCCESS, FAILURE, ERROR, SKIPPED = range(1, 5) _KEY_EXT = 'key' _IV_EXT = 'iv' @@ -31,12 +32,12 @@ def _run_encryption_test(tools, tmp_dir, algorithm, mode, key, plain_path, ciphe if force: logging.warn('Overwriting expected ciphertext file') shutil.copy(tmp_path, cipher_path) - return _TestExitCode.SKIPPED + return TestExitCode.SKIPPED if filecmp.cmp(cipher_path, tmp_path): - return _TestExitCode.SUCCESS + return TestExitCode.SUCCESS else: logging.error('The encrypted file doesn\'t match the ciphertext file') - return _TestExitCode.FAILURE + return TestExitCode.FAILURE def _run_decryption_test(tools, tmp_dir, algorithm, mode, key, cipher_path, plain_path, iv=None): logging.info('Running decryption test...') @@ -46,10 +47,10 @@ def _run_decryption_test(tools, tmp_dir, algorithm, mode, key, cipher_path, plai logging.info('\tDecrypted file path: ' + tmp_path) tools.run_decrypt_file(algorithm, mode, key, cipher_path, tmp_path, iv) if filecmp.cmp(tmp_path, plain_path): - return _TestExitCode.SUCCESS + return TestExitCode.SUCCESS else: logging.error('The decrypted file doesn\'t match the plaintext file') - return _TestExitCode.FAILURE + return TestExitCode.FAILURE def _list_dirs(root_path): xs = map(lambda x: os.path.join(root_path, x), os.listdir(root_path)) @@ -97,7 +98,7 @@ def _run_tests(tools, suite_dir, force=False): maybe_algorithm = Algorithm.try_parse(algorithm) if maybe_algorithm is None: logging.warn('Unknown or unsupported algorithm: ' + algorithm) - exit_codes.append(_TestExitCode.SKIPPED) + exit_codes.append(TestExitCode.SKIPPED) continue algorithm = maybe_algorithm logging.info('Algorithm: {}'.format(algorithm)) @@ -106,7 +107,7 @@ def _run_tests(tools, suite_dir, force=False): maybe_mode = Mode.try_parse(mode) if maybe_mode is None: logging.warn('Unknown or unsupported mode: ' + mode) - exit_codes.append(_TestExitCode.SKIPPED) + exit_codes.append(TestExitCode.SKIPPED) continue mode = maybe_mode logging.info('Mode: {}'.format(mode)) @@ -129,7 +130,7 @@ def _run_tests(tools, suite_dir, force=False): except Exception as e: logging.error('Encountered an exception!') logging.exception(e) - exit_codes.append(_TestExitCode.ERROR) + exit_codes.append(TestExitCode.ERROR) if not force: try: exit_codes.append(_run_decryption_test( @@ -138,18 +139,22 @@ def _run_tests(tools, suite_dir, force=False): except Exception as e: logging.error('Encountered an exception!') logging.exception(e) - exit_codes.append(_TestExitCode.ERROR) + exit_codes.append(TestExitCode.ERROR) logging.info('Test exit codes:') - logging.info('\tSkipped: {0}'.format(exit_codes.count(_TestExitCode.SKIPPED))) - logging.info('\tError(s): {0}'.format(exit_codes.count(_TestExitCode.ERROR))) - logging.info('\tSucceeded: {0}'.format(exit_codes.count(_TestExitCode.SUCCESS))) - logging.info('\tFailed: {0}'.format(exit_codes.count(_TestExitCode.FAILURE))) - if (exit_codes.count(_TestExitCode.ERROR) == 0 and - exit_codes.count(_TestExitCode.FAILURE) == 0): + logging.info('\tSkipped: {}'.format(exit_codes.count(TestExitCode.SKIPPED))) + logging.info('\tError(s): {}'.format(exit_codes.count(TestExitCode.ERROR))) + logging.info('\tSucceeded: {}'.format(exit_codes.count(TestExitCode.SUCCESS))) + logging.info('\tFailed: {}'.format(exit_codes.count(TestExitCode.FAILURE))) + if (exit_codes.count(TestExitCode.ERROR) == 0 and + exit_codes.count(TestExitCode.FAILURE) == 0): sys.exit() else: sys.exit(1) +def _build_default_log_path(): + return datetime.now().strftime('{}_%Y-%m-%d_%H-%M-%S.log').format( + os.path.splitext(os.path.basename(__file__))[0]) + if __name__ == '__main__': import argparse parser = argparse.ArgumentParser() @@ -157,21 +162,17 @@ if __name__ == '__main__': help='set path to file encryption utilities') parser.add_argument('--sde', '-e', action='store_true', help='use Intel SDE to run *.exe files') - parser.add_argument('--log', '-l', help='set log file path') + parser.add_argument('--log', '-l', default=_build_default_log_path(), + help='set log file path') parser.add_argument('--force', '-f', action='store_true', help='overwrite ciphertext files') parser.add_argument('--suite', '-s', default='file', help='set test suite directory path') args = parser.parse_args() - logging_options = { - 'format': '%(asctime)s | %(module)s | %(levelname)s | %(message)s', - 'level': logging.DEBUG } - if args.log is None: - logging_options['filename'] = datetime.now().strftime('file_%Y-%m-%d_%H-%M-%S.log') - else: - logging_options['filename'] = args.log - logging.basicConfig(**logging_options) + logging.basicConfig(filename=args.log, + format='%(asctime)s | %(module)s | %(levelname)s | %(message)s', + level=logging.DEBUG) tools = Tools(args.path, use_sde=args.sde) _run_tests(tools, args.suite, args.force) |