pqcrypto/test/test_duplicate_consistency.py

"""
Checks that files duplicated across schemes/implementations are consistent.
"""

import difflib
import os
import sys

import yaml

import helpers
import pqclean


def pytest_generate_tests(metafunc):
    ids = []
    argvalues = []
    for scheme in pqclean.Scheme.all_schemes():
        for implementation in scheme.implementations:
            if os.path.isfile(
                    os.path.join(
                        'duplicate_consistency',
                        '{}_{}.yml'.format(scheme.name, implementation.name))):
                metafile = os.path.join(
                    'duplicate_consistency',
                    '{}_{}.yml'.format(scheme.name, implementation.name))
                with open(metafile, encoding='utf-8') as f:
                    metadata = yaml.safe_load(f.read())
                    for group in metadata['consistency_checks']:
                        source = pqclean.Implementation.by_name(
                                group['source']['scheme'],
                                group['source']['implementation'])
                        for file in group['files']:
                            argvalues.append((implementation, source, file))
                            ids.append(
                                "{scheme.name} {implementation.name} {source.scheme.name}: {file}"
                                .format(scheme=scheme, source=source,
                                        implementation=implementation,
                                        file=file))
    metafunc.parametrize(('implementation', 'source', 'file'),
                         argvalues,
                         ids=ids)


def file_get_contents(filename):
    with open(filename) as f:
        return f.read()


@helpers.skip_windows()
@helpers.filtered_test
def test_duplicate_consistency(implementation, source, file):
    target_path = os.path.join(source.path(), file)
    this_path = os.path.join(implementation.path(), file)
    target_src = file_get_contents(target_path)
    this_src = file_get_contents(this_path)
    this_transformed_src = this_src.replace(
        implementation.namespace_prefix(), '')
    target_transformed_src = target_src.replace(source.namespace_prefix(), '')

    if not this_transformed_src == target_transformed_src:
        diff = difflib.unified_diff(
                this_transformed_src.splitlines(keepends=True),
                target_transformed_src.splitlines(keepends=True),
                fromfile=this_path,
                tofile=target_path)
        raise AssertionError(
            "Files differed:\n"
            + ''.join(diff))


if __name__ == '__main__':
    import pytest
    pytest.main(sys.argv)