From 9f329956930b66bb4a92390713c4c8136ab89fe4 Mon Sep 17 00:00:00 2001
From: protolambda <proto@protolambda.com>
Date: Sun, 7 Apr 2019 16:32:48 +1000
Subject: [PATCH] enable generator to locate configurations

---
 test_libs/gen_helpers/gen_base/gen_runner.py | 10 +++++++++-
 test_libs/gen_helpers/gen_base/gen_typing.py |  3 ++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/test_libs/gen_helpers/gen_base/gen_runner.py b/test_libs/gen_helpers/gen_base/gen_runner.py
index ad729449a..4c3f69bb1 100644
--- a/test_libs/gen_helpers/gen_base/gen_runner.py
+++ b/test_libs/gen_helpers/gen_base/gen_runner.py
@@ -55,6 +55,13 @@ def run_generator(generator_name, suite_creators: List[TestSuiteCreator]):
         default=False,
         help="if set overwrite test files if they exist",
     )
+    parser.add_argument(
+        "-c",
+        "--configs-path",
+        dest="configs_path",
+        default=True,
+        help="specify the path of the configs directory (containing constants_presets and fork_timelines)",
+    )
 
     args = parser.parse_args()
     output_dir = args.output_dir
@@ -66,8 +73,9 @@ def run_generator(generator_name, suite_creators: List[TestSuiteCreator]):
     yaml = YAML(pure=True)
 
     print(f"Generating tests for {generator_name}, creating {len(suite_creators)} test suite files...")
+    print(f"Reading config presets and fork timelines from {args.configs_path}")
     for suite_creator in suite_creators:
-        suite = suite_creator()
+        suite = suite_creator(args.configs_path)
 
         filename = make_filename_for_test(suite)
         path = output_dir / filename
diff --git a/test_libs/gen_helpers/gen_base/gen_typing.py b/test_libs/gen_helpers/gen_base/gen_typing.py
index 1384c870f..d6bd679af 100644
--- a/test_libs/gen_helpers/gen_base/gen_typing.py
+++ b/test_libs/gen_helpers/gen_base/gen_typing.py
@@ -2,4 +2,5 @@ from typing import Callable, Dict, Any
 
 TestCase = Dict[str, Any]
 TestSuite = Dict[str, Any]
-TestSuiteCreator = Callable[[], TestSuite]
+# Args: <presets path>
+TestSuiteCreator = Callable[[str], TestSuite]