Subversion Repositories Kolibri OS

#!/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