From d2a5164028ecc06b460c6444b4467353ea1949c9 Mon Sep 17 00:00:00 2001 From: petechd Date: Fri, 2 Aug 2024 09:13:07 +0100 Subject: [PATCH] Remove test duplication and no schema name error --- scripts/generate_integration_test.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/scripts/generate_integration_test.py b/scripts/generate_integration_test.py index 5be85ec637..551e129960 100644 --- a/scripts/generate_integration_test.py +++ b/scripts/generate_integration_test.py @@ -91,7 +91,7 @@ def process_launcher_request(request: Request) -> None: with open(output["file_name"], "a", encoding="utf-8") as file: path = f'"{urlparse(request.url).path}"' file.write(generate_method_request(method="get", data=path)) - else: + elif "schema_name" in parse_qs(request.url): # start of journey, so create a skeleton file using the schema name survey_journey["schema_name"] = parse_qs(request.url)["schema_name"][0] output["file_name"] = f"./scripts/{survey_journey['schema_name']}.py" @@ -99,10 +99,14 @@ def process_launcher_request(request: Request) -> None: with open(output["file_name"], "w", encoding="utf-8") as file: # Type ignore: schema_name is taken as string from query string class_name = survey_journey["schema_name"].title().replace("_", "") # type: ignore + function_name = survey_journey["schema_name"] + if not class_name.lower().startswith("test"): + class_name = f"Test{class_name}" + function_name = f"test_{survey_journey['schema_name']}" file.write( TEST_TEMPLATE.format( class_name=class_name, - function_name=survey_journey["schema_name"], + function_name=function_name, schema_name=survey_journey["schema_name"], ) )