28 lines
957 B
Python
28 lines
957 B
Python
import sys
|
|
import os
|
|
import argparse
|
|
import json
|
|
|
|
|
|
def get_test_paths_for_shard(shard_name):
|
|
unit_tests_json = f'{os.getcwd()}/.github/workflows/unit-test-shards.json'
|
|
with open(unit_tests_json) as file:
|
|
unit_test_workflow_shards = json.loads(file.read())
|
|
|
|
if shard_name not in unit_test_workflow_shards:
|
|
sys.stdout.write("Error, invalid shard name provided. please provide a valid shard name as specified in unit-test-shards.json")
|
|
|
|
return unit_test_workflow_shards.get(shard_name)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--shard-name", action="store", default="")
|
|
argument = parser.parse_args()
|
|
|
|
if not argument.shard_name:
|
|
sys.stdout.write("Error, no shard name provided. please provide a valid shard name as specified in unit-test-shards.json")
|
|
|
|
unit_test_paths = get_test_paths_for_shard(argument.shard_name)
|
|
sys.stdout.write(unit_test_paths)
|