rougail/tests/test_3_template.py

119 lines
3.8 KiB
Python
Raw 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()
2022-01-09 13:25:22 +01:00
#test_ok = ['01base_file_include_content']
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))
2022-10-01 22:27:22 +02:00
async def template(test_dir, filename, root):
2019-12-02 10:33:46 +01:00
test_dir = join(template_dirs, test_dir)
2022-10-01 22:27:22 +02:00
tmp_dir = join(test_dir, '..', 'tmp')
2020-07-29 08:59:40 +02:00
2022-10-01 22:27:22 +02:00
modulepath = test_dir.replace('/', '.') + f'.tiramisu.{filename}'
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()
2022-10-01 22:27:22 +02:00
if filename == 'multi':
sconfig = config
config = config.option(root)
2020-07-29 08:59:40 +02:00
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
2022-03-06 18:54:57 +01:00
if isfile(join(test_dir, 'file_tmpfile_dest_dir')):
RougailConfig['tmpfile_dest_dir'] = '/test/new/file'
else:
RougailConfig['tmpfile_dest_dir'] = '/usr/local/lib'
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,
)
2022-01-09 13:25:22 +01:00
assert list_templates == list_results, f'error with {test_dir}:'
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)
2022-10-01 22:27:22 +02:00
@mark.asyncio
async def test_template(test_dir):
await template(test_dir, 'base', '1')
@mark.asyncio
async def test_template_multi_1(test_dir):
await template(test_dir, 'multi', '1')
@mark.asyncio
async def test_template_multi_2(test_dir):
await template(test_dir, 'multi', '1')