import os
import jinja2
from bopytex.worker.generate import generate
import pytest


@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"
    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"

    message = generate(
        args={"a": 2, "jinja2": {"environment": jinja2_env}},
        deps=[template],
        output=output,
    )
    print(message.err)
    assert message.status == 0
    assert message.out == ["GENERATE - template.j2 to output"]

    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