[KERNEL] Add test framework

git-svn-id: svn://kolibrios.org@9249 a494cfbc-eb01-0410-851d-a64ba20cac60

kernel/trunk/runtests.py

@@ -0,0 +1,95 @@
+#!/usr/bin/python3
+# Copyright 2021 KolibriOS Team
+# Copyright 2021 Nekos Team
+# Published under MIT License
+
+import os
+import sys
+from importlib.machinery import SourceFileLoader
+import timeit
+import urllib.request
+import subprocess
+
+sys.path.append('test')
+import common
+
+root_dir = os.getcwd()
+tests = []
+
+def log(s, end = "\n"):
+    print(s, end = end, flush = True)
+
+def execute(s, mute = False):
+    mute = ">/dev/null" if mute else ""
+    code = os.system(f"{s}{mute}")
+    if code:
+        print(f"Command returned {code}: \"{s}\"")
+        exit(-1)
+
+def stage(name, command, mute = False):
+    print(f"{name}... ", end = "")
+    execute(command, mute = mute)
+    print("Done.")
+
+if not os.path.exists("kolibri_test.img"):
+    if len(sys.argv) == 1:
+        execute("wget -q --show-progress http://builds.kolibrios.org/eng/data/data/kolibri.img -O kolibri_test.img")
+    else:
+        builds = sys.argv[1]
+        execute(f"cp {builds}/data/data/kolibri.img kolibri_test.img")
+
+# Remove old kernel (may fail if we removed it before so no check here)
+os.system("mdel -i kolibri_test.img ::kernel.mnt > /dev/null")
+
+# Check free space after kernel remove
+free_clusters = int(subprocess.check_output("mdu -i kolibri_test.img :: -s", shell=True).split()[-1])
+floppy_image_clusters = 2880
+if floppy_image_clusters - free_clusters < 500:
+    # Remove unuseful files from IMG if lesser than 500 sectors
+    execute("mdeltree -i kolibri_test.img ::GAMES", mute = True)
+    execute("mdeltree -i kolibri_test.img ::DEMOS", mute = True)
+    execute("mdeltree -i kolibri_test.img ::3D", mute = True)
+
+# Build kernel with debug output
+stage("Building bootbios.bin.pretest",
+      "fasm -m 65536 -dpretest_build=1 bootbios.asm bootbios.bin.pretest", mute = True)
+
+stage("Building kernel.mnt.pretest",
+      "fasm -m 65536 -dpretest_build=1 -ddebug_com_base=0xe9 kernel.asm kernel.mnt.pretest", mute = True)
+
+# Put the kernel into IMG
+execute("mcopy -D o -i kolibri_test.img kernel.mnt.pretest ::kernel.mnt", mute = True)
+
+# Collect tests from test folder (not recursively yet)
+for test_folder in os.listdir("test"):
+    test_folder_path = f"test/{test_folder}"
+    test_file = f"{test_folder_path}/test.py"
+
+    if not os.path.isdir(test_folder_path):
+        continue
+
+    if os.path.exists(test_file):
+        tests.append(test_folder_path)
+
+# Execute each test
+test_number = 1
+for test in tests:
+    test_dir = f"{root_dir}/{test}"
+
+    os.chdir(test_dir)
+    print(f"[{test_number}/{len(tests)}] {test}... ", end = "", flush=True)
+    start = timeit.default_timer()
+    try:
+        SourceFileLoader("test", "test.py").load_module().run()
+    except common.TestTimeoutException:
+        result = "TIMEOUT"
+    except common.TestFailureException:
+        result = "FAILURE"
+    else:
+        result = "SUCCESS"
+    finish = timeit.default_timer()
+    print(f"{result} ({finish - start:.2f} seconds)")
+    os.chdir(root_dir)
+
+    test_number += 1
+