from bopytex.planner.generate_compile_join_planner import tasks_builder as gcj_task_builder
from bopytex.planner.activate_corr_compile_join_planner import tasks_builder as accj_task_builder
from bopytex.tasks import Task


def test_tasks_builder_generate():
    tasks = gcj_task_builder(
        options={
            "template": "tpl_source.tex",
            "subjects": [{"number": "01"}, {"number": "02"}],
            "no_pdf": True,
        }
    )
    assert tasks == [
        Task(
            action="GENERATE",
            args={
                "number": "01",
                "template": "tpl_source.tex",
                "subjects": [{"number": "01"}, {"number": "02"}],
                "no_pdf": True,
            },
            deps=["tpl_source.tex"],
            output="01_source.tex",
        ),
        Task(
            action="GENERATE",
            args={
                "number": "02",
                "template": "tpl_source.tex",
                "subjects": [{"number": "01"}, {"number": "02"}],
                "no_pdf": True,
            },
            deps=["tpl_source.tex"],
            output="02_source.tex",
        ),
    ]


def test_tasks_builder_generate_compile():
    tasks = gcj_task_builder(
        options={
            "template": "tpl_source.tex",
            "subjects": [{"number": "01"}, {"number": "02"}],
            "no_join": True,
        }
    )
    assert tasks == [
        Task(
            action="GENERATE",
            args={
                "number": "01",
                "template": "tpl_source.tex",
                "subjects": [{"number": "01"}, {"number": "02"}],
                "no_join": True,
            },
            deps=["tpl_source.tex"],
            output="01_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["01_source.tex"],
            output="01_source.pdf",
        ),
        Task(
            action="GENERATE",
            args={
                "number": "02",
                "template": "tpl_source.tex",
                "subjects": [{"number": "01"}, {"number": "02"}],
                "no_join": True,
            },
            deps=["tpl_source.tex"],
            output="02_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["02_source.tex"],
            output="02_source.pdf",
        ),
    ]


def test_tasks_builder_generate_compile_join():
    tasks = gcj_task_builder(
        options={
            "template": "tpl_source.tex",
            "subjects": [{"number": "01"}, {"number": "02"}],
        }
    )
    assert tasks == [
        Task(
            action="GENERATE",
            args={
                "number": "01",
                "template": "tpl_source.tex",
                "subjects": [{"number": "01"}, {"number": "02"}],
            },
            deps=["tpl_source.tex"],
            output="01_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["01_source.tex"],
            output="01_source.pdf",
        ),
        Task(
            action="GENERATE",
            args={
                "number": "02",
                "template": "tpl_source.tex",
                "subjects": [{"number": "01"}, {"number": "02"}],
            },
            deps=["tpl_source.tex"],
            output="02_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["02_source.tex"],
            output="02_source.pdf",
        ),
        Task(
            action="JOIN",
            args={},
            deps=["01_source.pdf", "02_source.pdf"],
            output="joined_source.pdf",
        ),
    ]


def test_tasks_builder_generate_compile_corr():
    tasks = gcj_task_builder(
        options={
            "template": "tpl_source.tex",
            "subjects": [{"number": "01"}, {"number": "02"}],
            "corr": True,
            "no_join": True,
        }
    )
    assert tasks == [
        Task(
            action="GENERATE",
            args={
                "number": "01",
                "template": "tpl_source.tex",
                "subjects": [{"number": "01"}, {"number": "02"}],
                "corr": True,
                "no_join": True,
            },
            deps=["tpl_source.tex"],
            output="01_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["01_source.tex"],
            output="01_source.pdf",
        ),
        Task(
            action="ACTIVATE_CORR",
            args={},
            deps=["01_source.tex"],
            output="corr_01_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["corr_01_source.tex"],
            output="corr_01_source.pdf",
        ),
        Task(
            action="GENERATE",
            args={
                "number": "02",
                "template": "tpl_source.tex",
                "subjects": [{"number": "01"}, {"number": "02"}],
                "corr": True,
                "no_join": True,
            },
            deps=["tpl_source.tex"],
            output="02_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["02_source.tex"],
            output="02_source.pdf",
        ),
        Task(
            action="ACTIVATE_CORR",
            args={},
            deps=["02_source.tex"],
            output="corr_02_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["corr_02_source.tex"],
            output="corr_02_source.pdf",
        ),
    ]


def test_tasks_builder_generate_compile_corr_joined():
    tasks = gcj_task_builder(
        options={
            "template": "tpl_source.tex",
            "subjects": [{"number": "01"}, {"number": "02"}],
            "corr": True,
            "no_join": False,
        }
    )
    assert tasks == [
        Task(
            action="GENERATE",
            args={
                "number": "01",
                "corr": True,
                "no_join": False,
                "template": "tpl_source.tex",
                "subjects": [{"number": "01"}, {"number": "02"}],
            },
            deps=["tpl_source.tex"],
            output="01_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["01_source.tex"],
            output="01_source.pdf",
        ),
        Task(
            action="ACTIVATE_CORR",
            args={},
            deps=["01_source.tex"],
            output="corr_01_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["corr_01_source.tex"],
            output="corr_01_source.pdf",
        ),
        Task(
            action="GENERATE",
            args={
                "number": "02",
                "template": "tpl_source.tex",
                "subjects": [{"number": "01"}, {"number": "02"}],
                "corr": True,
                "no_join": False,
            },
            deps=["tpl_source.tex"],
            output="02_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["02_source.tex"],
            output="02_source.pdf",
        ),
        Task(
            action="ACTIVATE_CORR",
            args={},
            deps=["02_source.tex"],
            output="corr_02_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["corr_02_source.tex"],
            output="corr_02_source.pdf",
        ),
        Task(
            action="JOIN",
            args={},
            deps=["01_source.pdf", "02_source.pdf"],
            output="joined_source.pdf",
        ),
        Task(
            action="JOIN",
            args={},
            deps=["corr_01_source.pdf", "corr_02_source.pdf"],
            output="corr_joined_source.pdf",
        ),
    ]


def test_only_corr_tasks_builder():
    tasks = accj_task_builder(
        options={
            "sources": ["01_source.tex", "02_source.tex"],
        }
    )
    assert tasks == [
        Task(
            action="ACTIVATE_CORR",
            args={},
            deps=["01_source.tex"],
            output="corr_01_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["corr_01_source.tex"],
            output="corr_01_source.pdf",
        ),
        Task(
            action="ACTIVATE_CORR",
            args={},
            deps=["02_source.tex"],
            output="corr_02_source.tex",
        ),
        Task(
            action="COMPILE",
            args={},
            deps=["corr_02_source.tex"],
            output="corr_02_source.pdf",
        ),
        Task(
            action="JOIN",
            args={},
            deps=["corr_01_source.pdf", "corr_02_source.pdf"],
            output="corr_joined.pdf",
        ),
    ]