import os

from test_general import SalisTest

SIM_NAME = "test-run.sim"
PUSH_POW = 8

class TestRun(SalisTest):
    def cmds(self, anc, arch):
        return (
            (
                f"./salis.py new -a{anc} -d{PUSH_POW} -f -g{compiler} -H{self.tempdir.name} -n{SIM_NAME} {optimized} -s{seed} -upush -Utest/ui/ -v{arch} -y{PUSH_POW}",
                f"./salis.py load -g{compiler} -H{self.tempdir.name} -n{SIM_NAME} {optimized} -upush -Utest/ui/",
            )
            for compiler in ["clang", "gcc"]
            for optimized in ["", "-o"]
            for seed in [-1, 0, 1234]
        )

    def cmds_null(self):
        return self.cmds("0123", "null")

    def cmds_v1(self):
        return self.cmds("55a", "v1")

    def files_on_first_push(self):
        return [
            *self.files_on_new(SIM_NAME),
            os.path.join(self.tempdir.name, SIM_NAME, "evas", f"evas-{2 ** PUSH_POW:016x}"),
        ]

    def files_on_second_push(self):
        return [
            *self.files_on_first_push(),
            os.path.join(self.tempdir.name, SIM_NAME, "evas", f"evas-{2 * (2 ** PUSH_POW):016x}"),
        ]

    def test_null(self):
        for cmd in self.cmds_null():
            SalisTest.run_subprocess(cmd[0])
            self.assert_file_list_exists(self.files_on_first_push())
            SalisTest.run_subprocess(cmd[1])
            self.assert_file_list_exists(self.files_on_second_push())

    def test_v1(self):
        for cmd in self.cmds_v1():
            SalisTest.run_subprocess(cmd[0])
            self.assert_file_list_exists(self.files_on_first_push())
            SalisTest.run_subprocess(cmd[1])
            self.assert_file_list_exists(self.files_on_second_push())