rougail/tests/test_3_template.py

98 lines
3.2 KiB
Python
Raw Permalink Normal View History

2021-05-13 22:30:58 +02:00
from os import listdir, mkdir, readlink
2021-02-21 19:48:30 +01:00
from os.path import join, isdir, isfile, islink
2019-12-02 10:33:46 +01:00
from shutil import rmtree
2020-02-14 17:59:39 +01:00
from pytest import fixture, mark
2019-12-02 10:33:46 +01:00
from lxml.etree import parse
from tiramisu import Config
from rougail import RougailConfig, RougailSystemdTemplate
2019-12-02 10:33:46 +01:00
template_dirs = 'tests/dictionaries'
excludes = set([])
#excludes = set(['01base_file_utfchar'])
test_ok = {f for f in listdir(template_dirs) if not f.startswith('_') and isdir(join(template_dirs, f, 'tmpl'))}
test_ok -= excludes
test_ok = list(test_ok)
test_ok.sort()
#test_ok = ['20override']
2019-12-02 10:33:46 +01:00
@fixture(scope="module", params=test_ok)
def test_dir(request):
return request.param
2020-07-06 19:47:45 +02:00
def find_files(dirname: str,
root: list,
files: set,
) -> None:
2019-12-22 11:04:39 +01:00
for filename in listdir(dirname):
abs_dirname = join(dirname, filename)
root_file = root + [filename]
if isdir(join(dirname, filename)):
2020-07-06 19:47:45 +02:00
find_files(abs_dirname,
root_file,
files,
)
2019-12-22 11:04:39 +01:00
else:
files.add(join(*root_file))
2020-02-14 17:59:39 +01:00
@mark.asyncio
async def test_dictionary(test_dir):
2019-12-02 10:33:46 +01:00
test_dir = join(template_dirs, test_dir)
tmp_dir = join(test_dir, 'tmp')
2020-07-29 08:59:40 +02:00
modulepath = test_dir.replace('/', '.') + '.tiramisu.base'
2020-07-29 08:59:40 +02:00
mod = __import__(modulepath)
for token in modulepath.split(".")[1:]:
mod = getattr(mod, token)
config = await Config(mod.option_0)
await config.property.read_only()
2019-12-02 10:33:46 +01:00
funcs_file = join(template_dirs, '../eosfunc/test.py')
distrib_dir = join(test_dir, 'tmpl')
2019-12-02 10:33:46 +01:00
if isdir(tmp_dir):
rmtree(tmp_dir)
mkdir(tmp_dir)
dest_dir = join(test_dir, 'dest')
if isdir(dest_dir):
rmtree(dest_dir)
mkdir(dest_dir)
2021-02-14 10:10:48 +01:00
RougailConfig['patches_dir'] = join(test_dir, 'patches')
RougailConfig['templates_dir'] = distrib_dir
RougailConfig['tmp_dir'] = tmp_dir
RougailConfig['functions_file'] = funcs_file
RougailConfig['destinations_dir'] = dest_dir
engine = RougailSystemdTemplate(config)
2021-02-14 10:10:48 +01:00
await engine.instance_files()
2020-02-14 17:59:39 +01:00
list_templates = set()
if isdir(dest_dir):
2020-07-06 19:47:45 +02:00
find_files(dest_dir,
[],
list_templates,
)
2019-12-22 11:04:39 +01:00
list_results = set()
2020-02-14 17:59:39 +01:00
if isdir(join(test_dir, 'result')):
2020-07-06 19:47:45 +02:00
find_files(join(test_dir, 'result'),
[],
list_results,
)
2019-12-02 10:33:46 +01:00
assert list_templates == list_results
2019-12-22 11:04:39 +01:00
for result in list_results:
template_file = join(dest_dir, result)
2021-05-13 22:30:58 +02:00
assert islink(template_file) == islink(join(test_dir, 'result', result))
if islink(template_file):
assert readlink(template_file) == readlink(join(test_dir, 'result', result))
2021-02-21 19:48:30 +01:00
continue
2019-12-02 10:33:46 +01:00
if not isfile(template_file):
raise Exception(f'{template_file} is not generated')
with open(join(test_dir, 'result', result), 'r') as fh:
result_file = fh.read()
with open(template_file, 'r') as fh:
generated_file = fh.read()
assert result_file == generated_file, f'{template_file} content : \n{generated_file}\nexpected: \nresult_file\n'
2019-12-22 11:29:57 +01:00
rmtree(dest_dir)
rmtree(tmp_dir)