#!/usr/bin/env python3

from pathlib import Path
from sys import path, argv, exit
from glob import glob
from os import _exit
from os.path import isfile
from traceback import format_exc
from json import loads
from subprocess import run

path.extend((f"{Path.cwd()}/utils", f"{Path.cwd()}/tests"))

from DockerTest import DockerTest
from AutoconfTest import AutoconfTest
from SwarmTest import SwarmTest
from KubernetesTest import KubernetesTest
from LinuxTest import LinuxTest
from logger import log

if len(argv) <= 1:
    log("TESTS", "❌", "Missing type argument")
    exit(1)

test_type = argv[1]
if test_type not in ("linux", "docker", "autoconf", "swarm", "kubernetes", "ansible"):
    log("TESTS", "❌", "Wrong type argument " + test_type)
    exit(1)

run("docker system prune", shell=True)

log("TESTS", "ℹ️", "Starting tests for " + test_type + " ...")
ret = False
end_fun = None
if test_type == "docker":
    ret = DockerTest.init()
    end_fun = DockerTest.end
elif test_type == "autoconf":
    ret = AutoconfTest.init()
    end_fun = AutoconfTest.end
elif test_type == "swarm":
    ret = SwarmTest.init()
    end_fun = SwarmTest.end
elif test_type == "kubernetes":
    ret = KubernetesTest.init()
    end_fun = KubernetesTest.end
elif test_type == "linux":
    distro = argv[2]
    ret = LinuxTest.init(distro)
    end_fun = LinuxTest.end
if not ret:
    log("TESTS", "❌", "Test.init() failed")
    exit(1)

for example in glob("./examples/*"):
    if isfile(f"{example}/tests.json"):
        try:
            with open(f"{example}/tests.json") as f:
                tests = loads(f.read())
            if test_type not in tests["kinds"]:
                log(
                    "TESTS",
                    "ℹ️",
                    "Skipping tests for " + tests["name"] + " (not in kinds)",
                )
                continue
            test_obj = None
            no_copy_container = False
            delay = 0
            if "no_copy_container" in tests:
                no_copy_container = tests["no_copy_container"]
            if "delay" in tests:
                delay = tests["delay"]
            if test_type == "docker":
                test_obj = DockerTest(
                    tests["name"],
                    tests["timeout"],
                    tests["tests"],
                    no_copy_container=no_copy_container,
                    delay=delay,
                )
            elif test_type == "autoconf":
                test_obj = AutoconfTest(
                    tests["name"],
                    tests["timeout"],
                    tests["tests"],
                    no_copy_container=no_copy_container,
                    delay=delay,
                )
            elif test_type == "swarm":
                test_obj = SwarmTest(tests["name"], tests["timeout"], tests["tests"], delay=delay)
            elif test_type == "kubernetes":
                test_obj = KubernetesTest(tests["name"], tests["timeout"], tests["tests"], delay=delay)
            elif test_type == "linux":
                test_obj = LinuxTest(tests["name"], tests["timeout"], tests["tests"], distro)
            if not test_obj.run_tests():
                log("TESTS", "❌", "Tests failed for " + tests["name"])
                if test_type == "linux":
                    ret = end_fun(distro)
                else:
                    ret = end_fun()
                _exit(1)
        except:
            log(
                "TESTS",
                "❌",
                "Exception while executing test for example " + example + " : " + format_exc(),
            )
            if test_type == "linux":
                ret = end_fun(distro)
            else:
                ret = end_fun()
            exit(1)

if test_type == "linux":
    ret = end_fun(distro)
else:
    ret = end_fun()
if not ret:
    log("TESTS", "❌", "Test.end() failed")
    exit(1)

log("TESTS", "ℹ️", "All tests finished for " + test_type + " !")

run("docker system prune", shell=True)