from test.libregrtest.findtests import findtests

from concurrent.futures import ThreadPoolExecutor

import _xxsubinterpreters as interpreters
import _xxinterpchannels as channels


# Get a list of tests
test_names = findtests()
skip_tests = [
    # "test_builtin" # Crashes
    "test_httpservers", # hangs?
    "test_imaplib", # hangs?
    "test.test_concurrent_futures.test_process_pool", # hangs on test.test_concurrent_futures.test_process_pool.ProcessPoolSpawnProcessPoolExecutorTest.test_traceback
    "test.test_multiprocessing_forkserver.test_manager", # hangs in multiple places
    "test.test_multiprocessing_forkserver.test_processes", # hangs in multiple places
    "test.test_multiprocessing_spawn.test_manager", # hangs in multiple places
]


def run_test():
    # Args- test_name: str
    import sys

    # Use a patched faulthandler because the stdlib can't be imported
    from fake_faulthandler import FaultHandler
    sys.modules['faulthandler'] = FaultHandler()

    # Force ModuleNotFoundError for datetime imports instead of a segfault
    # See GH 102995
    # sys.modules['datetime'] = None

    from test.libregrtest.main import Regrtest, Namespace

    ns = Namespace()
    ns.single = True
    ns.verbose = 1
    # Namespace attributes that are assumed to be set
    ns.fast_ci = True
    ns.args = ()
    ns.xmlpath = None
    ns.use_resources = 'all'
    ns.python = None

    r = Regrtest(ns)
    runtests = r.create_run_tests((test_name, ))
    r.run_test(test_name, runtests, None)
    # TODO : Save test result somewhere.

results = {}

# If argument then use that as the only test
import sys
if len(sys.argv) > 1:
    test_names = [sys.argv[1]]

def run_test_thread(test):
    if test in skip_tests:
        print(f"Skipping test {test}")
        return
    print(f"Running test {test}")
    interp = interpreters.create()
    channel_id = channels.create()
    try:
        result = interpreters.run_func(
                interp,
                run_test,
                shared={
                    "test_name": test,
                })
    except Exception as e:
        print(f"Test {test} failed with exception {e}")
        results[test] = {
            "result": "failed",
            "reason": f"Test {test} failed with exception {e}"
        }
    finally:
        interpreters.destroy(interp)
        channels.destroy(channel_id)

with ThreadPoolExecutor(max_workers=4) as executor:
    for test in test_names:
        executor.submit(run_test_thread, test)

import os
import pathlib
import re

print("| Test | Result | Detail |")

# Read all txt files in testresults/
test_checks = []
for test_result_fname in pathlib.Path("testresults").glob("test*.txt"):
    test_name = test_result_fname.name.replace('.txt', '')
    test_checks.append((test_name, test_result_fname))

test_checks = sorted(test_checks, key=lambda x: x[0])

for test_name, test_result_fname in test_checks:
    print(f"| {test_name} |", end='')
    result = reason = ""
    with open(test_result_fname, encoding='utf-8', errors='ignore') as test_result:
        contents = test_result.read()

        for l in contents.splitlines():
            check = f"{test_name} skipped"
            if check in l:
                result = "🟠 skipped"
                reason = l.replace(check, '').replace(" -- ", "")

            if test_summary := re.search(r"Ran \d* test(s?) in \d*\.\d*s", l):
                if "\nOK" in contents:
                    result = "🟢 passed"
                elif "FAILED (" in contents:
                    error = re.search(r"\wError: ([\w \']*)", contents)
                    result = "🔴 failed"
                    if error:
                        reason = error.groups()[0]
                elif "NO TESTS RAN" in contents:
                    result = "🟠 skipped"
                else:
                    result = "🟠 ??? "
        if result == "" and f"Skipping test {test_name}" in contents:
            result = "🟠 skipped"
        if result == "":
            error = re.search(r"\wError: ([\w \']*)", contents)
            result = "🔴 failed"
            if error:
                reason = error.groups()[0]

        if "Debug memory block at address" in contents:
            result += " [1]"


    print(f" {result} | {reason} |")