Files
edx-platform/scripts/gha_unit_tests_collector.py

41 lines
1.2 KiB
Python

import sys
import os
import yaml
import argparse
def get_all_unit_test_modules():
unit_tests_yml = f'{os.getcwd()}/.github/workflows/unit-tests.yml'
with open(unit_tests_yml) as file:
unit_test_workflow_yaml = yaml.safe_load(file)
return unit_test_workflow_yaml['jobs']['run-tests']['strategy']['matrix']['test_module']
def get_modules_except_cms():
all_unit_test_modules = get_all_unit_test_modules()
return [module for module in all_unit_test_modules if not module.startswith('cms')]
def get_cms_modules():
all_unit_test_modules = get_all_unit_test_modules()
return [module for module in all_unit_test_modules if module.startswith('cms')]
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--cms-only", action="store_true", default="")
parser.add_argument("--lms-only", action="store_true", default="")
argument = parser.parse_args()
if argument.lms_only:
modules = get_modules_except_cms()
elif argument.cms_only:
modules = get_cms_modules()
else:
modules = get_all_unit_test_modules()
unit_test_paths = ' '.join(modules)
sys.stdout.write(unit_test_paths)