Adding a test script timeout argument

Author: VitjanZ

module_renderer.py

@@ -176,6 +176,7 @@ def _build_render_context_for_module(
             verbose=self.args.verbose,
             run_state=self.run_state,
             event_bus=self.event_bus,
+            test_script_timeout=self.args.test_script_timeout,
         )
 
     def _render_module(

plain2code.py

@@ -47,8 +47,6 @@
 from system_config import system_config
 from tui.plain2code_tui import Plain2CodeTUI
 
-TEST_SCRIPT_EXECUTION_TIMEOUT = 120  # 120 seconds
-
 DEFAULT_TEMPLATE_DIRS = importlib.resources.files("standard_template_library")
 
 MAX_UNITTEST_FIX_ATTEMPTS = 20

plain2code_arguments.py

@@ -202,6 +202,13 @@ def create_parser():
         help="Path to a shell script that prepares the testing environment. The script should accept the source code folder path as its first argument.",
     )
 
+    parser.add_argument(
+        "--test-script-timeout",
+        type=int,
+        default=None,
+        help="Timeout for test scripts in seconds. If not provided, the default timeout of 120 seconds is used.",
+    )
+
     parser.add_argument(
         "--api",
         type=str,

render_machine/actions/prepare_testing_environment.py

@@ -21,6 +21,7 @@ def execute(self, render_context: RenderContext, _previous_action_payload: Any |
             [render_context.build_folder],
             render_context.verbose,
             "Testing Environment Preparation",
+            timeout=render_context.test_script_timeout
         )
 
         render_context.conformance_tests_running_context.should_prepare_testing_environment = False

render_machine/actions/run_conformance_tests.py

@@ -43,7 +43,8 @@ def execute(self, render_context: RenderContext, _previous_action_payload: Any |
             [render_context.build_folder, conformance_tests_folder_name],
             render_context.verbose,
             "Conformance Tests",
-            render_context.conformance_tests_running_context.current_testing_frid,
+            frid=render_context.conformance_tests_running_context.current_testing_frid,
+            timeout=render_context.test_script_timeout
         )
         render_context.script_execution_history.latest_conformance_test_output_path = conformance_tests_temp_file_path
         render_context.script_execution_history.should_update_script_outputs = True

render_machine/actions/run_unit_tests.py

@@ -24,6 +24,7 @@ def execute(self, render_context: RenderContext, _previous_action_payload: Any |
             [render_context.build_folder],
             render_context.verbose,
             "Unit Tests",
+            timeout=render_context.test_script_timeout
         )
 
         render_context.script_execution_history.latest_unit_test_output_path = unittests_temp_file_path

render_machine/render_context.py

@@ -51,6 +51,7 @@ def __init__(
         verbose: bool,
         run_state: RunState,
         event_bus: EventBus,
+        test_script_timeout: Optional[int] = None,
     ):
         self.codeplain_api: CodeplainAPI = codeplain_api
         self.memory_manager = memory_manager
@@ -75,6 +76,7 @@ def __init__(
         self.event_bus = event_bus
         self.script_execution_history = ScriptExecutionHistory()
         self.starting_frid = None
+        self.test_script_timeout = test_script_timeout
 
         resources_list = []
         plain_spec.collect_linked_resources(plain_source_tree, resources_list, None, True)

render_machine/render_utils.py

@@ -42,7 +42,7 @@ def print_inputs(render_context, existing_files_content, message):
 
 
 def execute_script(
-    script: str, scripts_args: list[str], verbose: bool, script_type: str, frid: Optional[str] = None
+    script: str, scripts_args: list[str], verbose: bool, script_type: str, frid: Optional[str] = None, timeout: Optional[int] = None
 ) -> tuple[int, str, Optional[str]]:
     temp_file_path = None
     try:
@@ -52,7 +52,7 @@ def execute_script(
             stdout=subprocess.PIPE,
             stderr=subprocess.STDOUT,
             text=True,
-            timeout=SCRIPT_EXECUTION_TIMEOUT,
+            timeout=timeout if timeout is not None else SCRIPT_EXECUTION_TIMEOUT,
         )
         elapsed_time = time.time() - start_time
         # Log the info about the script execution