rougail/tests/test_1_flattener.py

from lxml import etree
from os import getcwd
from os.path import isfile, join, isdir
from pytest import fixture, raises
from os import listdir, environ
from json import load

environ['TIRAMISU_LOCALE'] = 'en'

from rougail import RougailConvert, RougailConfig
from rougail.error import DictConsistencyError



dico_dirs = 'tests/dictionaries'


test_ok = set()
test_raise = set()

for test in listdir(dico_dirs):
    if isdir(join(dico_dirs, test)):
        if isdir(join(dico_dirs, test, 'tiramisu')):
            test_ok.add(test)
        elif test != '__pycache__':
            test_raise.add(test)

excludes = set([])
#excludes = set(['70services_ip_extra'])
test_ok -= excludes
test_raise -= excludes
#test_ok = ['01base_provider_extra']
#test_ok = []
#test_raise = ['80auto_autofreeze']
#test_raise = []

ORI_DIR = getcwd()

debug = False
#debug = True

test_ok = list(test_ok)
test_raise = list(test_raise)
test_ok.sort()
test_raise.sort()


@fixture(scope="module", params=test_ok)
def test_dir(request):
    return request.param


@fixture(scope="module", params=test_raise)
def test_dir_error(request):
    return request.param


def load_rougail_object(test_dir):
    rougailconfig = RougailConfig.copy()
    rougailconfig['functions_file'] = join(dico_dirs, '../eosfunc/test.py')
    dirs = [test_dir]
    subfolder = join(test_dir, 'subfolder')
    if isdir(subfolder):
        dirs.append(subfolder)
    rougailconfig['dictionaries_dir'] = dirs
    rougailconfig['extra_dictionaries'] = {}
    if isdir(join(test_dir, 'extra_dirs')):
        extras = listdir(join(test_dir, 'extra_dirs'))
        extras.sort()
        for extra in extras:
            subfolder = join(test_dir, 'extra_dirs', extra)
            if isdir(subfolder):
                rougailconfig['extra_dictionaries'][extra] = [subfolder]
    return RougailConvert(rougailconfig)


def launch_flattener(eolobj, path_prefix=None):
    eolobj.load_dictionaries(path_prefix=path_prefix)


def save(test_dir, eolobj, multi=False):
    tiramisu_objects = eolobj.save(None)
    if 'children=[]' in tiramisu_objects.split('\n')[-2]:
        raise Exception('empty tiramisu object?')
    tiramisu_dir = join(test_dir, 'tiramisu')
    if isdir(tiramisu_dir):
        if not multi:
            filename = 'base.py'
        else:
            filename = 'multi.py'
        tiramisu_file = join(tiramisu_dir, filename)
        if not isfile(tiramisu_file) or debug:
            with open(tiramisu_file, 'w') as fh:
                fh.write(tiramisu_objects)
        with open(tiramisu_file, 'r') as fh:
            tiramisu_objects_ori = fh.read()
        assert tiramisu_objects == tiramisu_objects_ori


def test_dictionary(test_dir):
    assert getcwd() == ORI_DIR
    test_dir = join(dico_dirs, test_dir)
    eolobj = load_rougail_object(test_dir)
    launch_flattener(eolobj)
    save(test_dir, eolobj)
    assert getcwd() == ORI_DIR


def test_dictionary_multi(test_dir):
    assert getcwd() == ORI_DIR
    test_dir = join(dico_dirs, test_dir)
    eolobj = load_rougail_object(test_dir)
    launch_flattener(eolobj, path_prefix='1')
    launch_flattener(eolobj, path_prefix='2')
    save(test_dir, eolobj, multi=True)
    assert getcwd() == ORI_DIR


def test_error_dictionary(test_dir_error):
    assert getcwd() == ORI_DIR
    test_dir = join(dico_dirs, test_dir_error)
    errno = 0
    eolobj = load_rougail_object(test_dir)
    for i in listdir(test_dir):
        if i.startswith('errno_'):
            if errno:
                raise Exception('multiple errno')
            errno = int(i.split('_')[1])
    with raises(DictConsistencyError) as err:
        launch_flattener(eolobj)
        save(test_dir, eolobj)
    if err.value.errno != errno:
        print(f'expected errno: {errno}, errno: {err.value.errno}')
        launch_flattener(eolobj)
        save(test_dir, eolobj)
    assert getcwd() == ORI_DIR