diff --git a/test/worker/test_generate.py b/test/worker/test_generate.py index fbd3baf..015f322 100644 --- a/test/worker/test_generate.py +++ b/test/worker/test_generate.py @@ -1,7 +1,6 @@ import os import jinja2 -from pathlib import Path -from bopytex.worker.generate import tpl2tex, generate +from bopytex.worker.generate import generate, tpl2tex import pytest @@ -12,6 +11,12 @@ def test_tpl2tex(): assert fed == "Plop 1" +@pytest.fixture +def jinja2_env(tmp_path): + templateEnv = jinja2.Environment(loader=jinja2.FileSystemLoader(tmp_path)) + return templateEnv + + @pytest.fixture def template_path(tmp_path): template = tmp_path / "template.j2" @@ -20,12 +25,6 @@ def template_path(tmp_path): return template -@pytest.fixture -def jinja2_env(tmp_path): - templateEnv = jinja2.Environment(loader=jinja2.FileSystemLoader(tmp_path)) - return templateEnv - - def test_generate(template_path, jinja2_env): tmp_path = template_path.parent os.chdir(tmp_path) @@ -46,3 +45,35 @@ def test_generate(template_path, jinja2_env): 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={"random": random, "jinja2": {"environment": jinja2_env}}, + 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