Bopytex/test/worker/test_generate.py

89 lines
2.1 KiB
Python
Raw Normal View History

2022-04-10 13:50:58 +00:00
import os
2022-07-28 07:39:51 +00:00
2022-04-10 13:50:58 +00:00
import jinja2
import pytest
2022-07-28 07:39:51 +00:00
from bopytex.worker.generate import generate
2022-04-10 13:50:58 +00:00
@pytest.fixture
def jinja2_env(tmp_path):
templateEnv = jinja2.Environment(loader=jinja2.FileSystemLoader(tmp_path))
return templateEnv
2022-04-10 13:50:58 +00:00
@pytest.fixture
def template_path(tmp_path):
template = tmp_path / "template.j2"
with open(template, "w") as tpl:
tpl.write("Plop {{ a }}")
return template
def test_generate(template_path, jinja2_env):
tmp_path = template_path.parent
os.chdir(tmp_path)
assert template_path.exists
template = str(template_path.name)
output = "output"
2022-04-13 10:26:04 +00:00
message = generate(
args={
"options": {
"direct_access": {"a": 2},
"jinja2": {"environment": jinja2_env},
},
2022-07-28 07:39:51 +00:00
"subject": {},
},
2022-04-10 13:50:58 +00:00
deps=[template],
output=output,
2022-04-13 10:26:04 +00:00
)
print(message.err)
assert message.status == 0
assert message.out == ["GENERATE - template.j2 to output"]
2022-04-10 13:50:58 +00:00
with open(output, "r") as out:
lines = out.readlines()
assert lines == ["Plop 2"]
@pytest.fixture
def template_path_with_random(tmp_path):
template = tmp_path / "template.j2"
with open(template, "w") as tpl:
tpl.write("Plop {{ random.randint(0, 10) }}")
return template
def test_generate_with_random(template_path_with_random, jinja2_env):
tmp_path = template_path_with_random.parent
os.chdir(tmp_path)
assert template_path_with_random.exists
template = str(template_path_with_random.name)
output = "output"
import random
message = generate(
args={
"options": {
"jinja2": {"environment": jinja2_env},
"direct_access": {
"random": random,
},
},
2022-07-28 07:39:51 +00:00
"subject": {},
},
deps=[template],
output=output,
)
print(message.err)
assert message.status == 0
with open(output, "r") as out:
lines = out.readlines()
assert int(lines[0][-1]) >= 0
assert int(lines[0][-1]) < 10