tempgen.tests.test_tempgen

import pytest
import os
import os
from collections import defaultdict
from syrupy.extensions.single_file import SingleFileSnapshotExtension
from tempgen.module import Tempgen
from tempgen.parsers import Parsers
from tempgen.transforms import Transforms
from tempgen.tests.helpers import ext_serializer_map

tests_dir = os.path.dirname(os.path.abspath(__file__))
fixture_dir = os.path.join(tests_dir, 'fixtures')
fixture_name = 'test_template'
generated_name = 'generated'

transforms = Transforms().name_transform_map.keys()
extensions = Parsers().ext_parser_map.keys()
serializers = ext_serializer_map

@pytest.fixture(autouse=True)
def tempgen_instance():
    return Tempgen()

@pytest.fixture
def tempgen_instances(request):
    return [Tempgen() for _ in range(request.param)]

@pytest.mark.parametrize('extension', extensions)
def test_load_template(extension, tempgen_instance, snapshot):
    template = os.path.join(fixture_dir, fixture_name + extension)
    tempgen_instance.load_template(template)
    assert template in tempgen_instance.get_templates()
    assert tempgen_instance.get_fields() == snapshot

@pytest.mark.parametrize('extension', extensions)
def test_save_result(extension, tempgen_instance, snapshot):
    template = os.path.join(fixture_dir, fixture_name + extension)
    tempgen_instance.load_template(template)
    replacements = { key: value['value'] for key, value in tempgen_instance.get_fields().items() }
    replacements['doer'] = 'Петров П.П.'
    replacements['itn'] = '987654321098'
    save_path = os.path.join(fixture_dir, generated_name)
    tempgen_instance.save_result(template, save_path, replacements)
    assert ext_serializer_map[extension](save_path + extension) == snapshot
    os.remove(save_path + extension)

@pytest.mark.parametrize('extension', extensions)
@pytest.mark.parametrize('transform', transforms)
@pytest.mark.parametrize('tempgen_instances', [2], indirect=['tempgen_instances'])
def test_independence(extension, transform, tempgen_instances):
    instance_0, instance_1 = tempgen_instances
    assert instance_0.parsers != instance_1.parsers
    assert instance_0.transforms != instance_1.transforms
    instance_0.parsers[extension].parse = lambda *args, **kwargs: ({})
    instance_0.transforms[transform] = lambda x: x
    assert instance_0.parsers != instance_1.parsers
    assert instance_0.transforms != instance_1.transforms

@pytest.fixture(autouse=True)
def tempgen_instance():
    return Tempgen()

@pytest.fixture
def tempgen_instances(request):
    return [Tempgen() for _ in range(request.param)]

@pytest.mark.parametrize('extension', extensions)
def test_load_template(extension, tempgen_instance, snapshot):
    template = os.path.join(fixture_dir, fixture_name + extension)
    tempgen_instance.load_template(template)
    assert template in tempgen_instance.get_templates()
    assert tempgen_instance.get_fields() == snapshot

@pytest.mark.parametrize('extension', extensions)
def test_save_result(extension, tempgen_instance, snapshot):
    template = os.path.join(fixture_dir, fixture_name + extension)
    tempgen_instance.load_template(template)
    replacements = { key: value['value'] for key, value in tempgen_instance.get_fields().items() }
    replacements['doer'] = 'Петров П.П.'
    replacements['itn'] = '987654321098'
    save_path = os.path.join(fixture_dir, generated_name)
    tempgen_instance.save_result(template, save_path, replacements)
    assert ext_serializer_map[extension](save_path + extension) == snapshot
    os.remove(save_path + extension)

@pytest.mark.parametrize('extension', extensions)
@pytest.mark.parametrize('transform', transforms)
@pytest.mark.parametrize('tempgen_instances', [2], indirect=['tempgen_instances'])
def test_independence(extension, transform, tempgen_instances):
    instance_0, instance_1 = tempgen_instances
    assert instance_0.parsers != instance_1.parsers
    assert instance_0.transforms != instance_1.transforms
    instance_0.parsers[extension].parse = lambda *args, **kwargs: ({})
    instance_0.transforms[transform] = lambda x: x
    assert instance_0.parsers != instance_1.parsers
    assert instance_0.transforms != instance_1.transforms