# file    : test/test_general.py
# project : Salis-VM
# author  : Paul Oliver <contact@pauloliver.dev>

# index:
# [section] setup/teardown
# [section] file lists
# [section] assertions
# [section] subprocess

import os
import subprocess
import unittest

from tempfile import TemporaryDirectory

class SalisTest(unittest.TestCase):
    # --------------------------------------------------------------------------
    # [section] setup/teardown
    # --------------------------------------------------------------------------
    def setUp(self):
        self.tempdir = TemporaryDirectory(prefix="salis_test_")

    def tearDown(self):
        self.tempdir.cleanup()

    # --------------------------------------------------------------------------
    # [section] file lists
    # --------------------------------------------------------------------------
    def files_on_new(self, name):
        return [
            f"{self.tempdir.name}/{name}/evas/evas-{0:016x}",
            f"{self.tempdir.name}/{name}/opts.json",
            f"{self.tempdir.name}/{name}/{name}",
            f"{self.tempdir.name}/{name}/{name}-{0:016x}",
            f"{self.tempdir.name}/{name}/{name}.sqlite3",
        ]

    # --------------------------------------------------------------------------
    # [section] assertions
    # --------------------------------------------------------------------------
    def assert_files_exist(self, files):
        for file in files:
            assert os.path.isfile(file), f"{file} does not exist"

    # --------------------------------------------------------------------------
    # [section] subprocess
    # --------------------------------------------------------------------------
    def run_subprocess(self, command):
        subprocess.run(command.split(), check=True, stdout=subprocess.DEVNULL)

    def run_pipe(self, command):
        return subprocess.Popen(command.split(), stdout=subprocess.PIPE, text=True)

    def run_ui_test(self, anc, anc_path, ui, ui_path, vm_arch, clones=1, cores=1, mvec_pow=20, name="def.sim", seed=0, data_push_pow=28, sync_pow=20, auto_save_pow=36, **kwargs):
        command = f"./salis.py new -a{anc} -A{anc_path} -C{clones} -c{cores} -d{data_push_pow} -f -H{self.tempdir.name} -m{mvec_pow} -n{name} -s{seed} -u{ui} -U{ui_path} -v{vm_arch} -y{sync_pow} -z{auto_save_pow}"
        self.run_subprocess(command)
        self.assert_files_exist(self.files_on_new(name))