forked from PAWPAW-Mirror/lib_xua
121 lines
4.1 KiB
Python
121 lines
4.1 KiB
Python
# Copyright 2021 XMOS LIMITED.
|
|
# This Software is subject to the terms of the XMOS Public Licence: Version 1.
|
|
import glob
|
|
import os.path
|
|
import subprocess
|
|
import sys
|
|
|
|
UNITY_TEST_DIR = 'src'
|
|
UNITY_TEST_PREFIX = 'test_'
|
|
UNITY_RUNNER_DIR = 'runners'
|
|
UNITY_RUNNER_SUFFIX = '_Runner'
|
|
project_root = os.path.join('..', '..', '..')
|
|
|
|
def get_ruby():
|
|
"""
|
|
Check ruby is avaliable and return the command to invoke it.
|
|
"""
|
|
interpreter_name = 'ruby'
|
|
try:
|
|
dev_null = open(os.devnull, 'w')
|
|
# Call the version command to check the interpreter can be run
|
|
subprocess.check_call([interpreter_name, '--version'],
|
|
stdout=dev_null,
|
|
close_fds=True)
|
|
except OSError as e:
|
|
print("Failed to run Ruby interpreter: {}".format(e), file=sys.stderr)
|
|
exit(1) # TODO: Check this is the correct way to kill xwaf on error
|
|
|
|
return interpreter_name
|
|
|
|
def get_unity_runner_generator(project_root_path):
|
|
"""
|
|
Check the Unity generate_test_runner script is avaliable, and return the
|
|
path to it.
|
|
"""
|
|
unity_runner_generator = os.path.join(
|
|
project_root_path, 'Unity', 'auto', 'generate_test_runner.rb')
|
|
if not os.path.exists(unity_runner_generator):
|
|
print("Unity repo not found in workspace", file=sys.stderr)
|
|
exit(1) # TODO: Check this is the correct way to kill xwaf on error
|
|
return unity_runner_generator
|
|
|
|
|
|
def get_test_name(test_path):
|
|
"""
|
|
Return the test name by removing the extension from the filename.
|
|
"""
|
|
return os.path.splitext(os.path.basename(test_path))[0]
|
|
|
|
|
|
def get_file_type(filename):
|
|
"""
|
|
Return the extension from the filename.
|
|
"""
|
|
return filename.rsplit('.')[-1:][0]
|
|
|
|
|
|
def generate_unity_runner(project_root_path, unity_test_path, unity_runner_dir,
|
|
unity_runner_suffix):
|
|
"""
|
|
Invoke the Unity runner generation script for the given test file, and
|
|
return the path to the generated file. The output directory will be created
|
|
if it does not already exist.
|
|
"""
|
|
runner_path = os.path.join(os.path.join(unity_runner_dir, get_test_name(unity_test_path)))
|
|
if not os.path.exists(runner_path):
|
|
os.makedirs(runner_path)
|
|
|
|
unity_runner_path = os.path.join(
|
|
runner_path, get_test_name(unity_test_path) + unity_runner_suffix
|
|
+ '.' + 'c')
|
|
|
|
try:
|
|
subprocess.check_call([get_ruby(),
|
|
get_unity_runner_generator(project_root_path),
|
|
unity_test_path,
|
|
unity_runner_path])
|
|
except OSError as e:
|
|
print("Ruby generator failed for {}\n\t{}".format(unity_test_path, e),
|
|
file=sys.stderr)
|
|
exit(1) # TODO: Check this is the correct way to kill xwaf on error
|
|
|
|
|
|
def find_unity_test_paths(unity_test_dir, unity_test_prefix):
|
|
"""
|
|
Return a list of all file paths with the unity_test_prefix found in the
|
|
unity_test_dir.
|
|
"""
|
|
return glob.glob(os.path.join(unity_test_dir, unity_test_prefix+'*'))
|
|
|
|
|
|
def find_unity_tests(unity_test_dir, unity_test_prefix):
|
|
"""
|
|
Return a dictionary of all {test names, test language} pairs with the
|
|
unity_test_prefix found in the unity_test_dir.
|
|
"""
|
|
unity_test_paths = find_unity_test_paths(unity_test_dir, unity_test_prefix)
|
|
print('unity_test_paths = ', unity_test_paths)
|
|
return {get_test_name(path): get_file_type(path)
|
|
for path in unity_test_paths}
|
|
|
|
def find_unity_test_paths(unity_test_dir, unity_test_prefix):
|
|
"""
|
|
Return a list of all file paths with the unity_test_prefix found in the
|
|
unity_test_dir.
|
|
"""
|
|
return glob.glob(os.path.join(unity_test_dir, unity_test_prefix+'*'))
|
|
|
|
|
|
def generate_runners():
|
|
UNITY_TESTS = find_unity_tests(UNITY_TEST_DIR, UNITY_TEST_PREFIX)
|
|
print('UNITY_TESTS = ',UNITY_TESTS)
|
|
unity_test_paths = find_unity_test_paths(UNITY_TEST_DIR, UNITY_TEST_PREFIX)
|
|
print('unity_test_paths = ',unity_test_paths)
|
|
for unity_test_path in unity_test_paths:
|
|
generate_unity_runner(project_root, unity_test_path, UNITY_RUNNER_DIR, UNITY_RUNNER_SUFFIX)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
generate_runners()
|