validation: Add test early preparation

4 files changed, 22 insertions, 6 deletions

diff --git a/rtemsspec/tests/spec-validation/action2.yml b/rtemsspec/tests/spec-validation/action2.yml
index 8f7f6907..9a56d818 100644
--- a/rtemsspec/tests/spec-validation/action2.yml
+++ b/rtemsspec/tests/spec-validation/action2.yml
@@ -124,6 +124,8 @@ test-includes:
 test-local-includes:
 - b.h
 test-name: Action 2
+test-prepare: |
+  /* Prepare */
 test-setup:
   brief: |
     Setup brief.
diff --git a/rtemsspec/tests/spec-validation/directive.yml b/rtemsspec/tests/spec-validation/directive.yml
index 71e583f7..47eef2ec 100644
--- a/rtemsspec/tests/spec-validation/directive.yml
+++ b/rtemsspec/tests/spec-validation/directive.yml
@@ -196,6 +196,7 @@ test-includes:
 test-local-includes:
 - x.h
 test-name: Classic Task Identification
+test-prepare: null
 test-setup:
   brief: |
     Setup brief description.
diff --git a/rtemsspec/tests/test_validation.py b/rtemsspec/tests/test_validation.py
index b0a7a3ef..76b9c63e 100644
--- a/rtemsspec/tests/test_validation.py
+++ b/rtemsspec/tests/test_validation.py
@@ -1416,6 +1416,11 @@ static const struct {
   }
 };
 
+static void Action2_Prepare( Action2_Context *ctx )
+{
+  /* Prepare */
+}
+
 static void Action2_Action( Action2_Context *ctx )
 {
   /* Action */
@@ -1462,6 +1467,7 @@ void Action2_Run( int *a, int b, int *c )
         continue;
       }
 
+      Action2_Prepare( ctx );
       Action2_Pre_A_Prepare( ctx, ctx->pcs[ 0 ] );
       Action2_Pre_B_Prepare( ctx, ctx->pcs[ 1 ] );
       Action2_Action( ctx );
diff --git a/rtemsspec/validation.py b/rtemsspec/validation.py
index a6e6172a..20c69114 100644
--- a/rtemsspec/validation.py
+++ b/rtemsspec/validation.py
@@ -471,18 +471,24 @@ class _TestDirectiveItem(_TestItem):
         content.add([f"}} {self.ident}_TransitionInfo[] = {{", "  {"])
         content.append(["\n  }, {\n".join(info_elements), "  }", "};"])
 
-    def _add_action(self, content: CContent) -> None:
-        with content.function("static void", f"{self.ident}_Action",
-                              [f"{self.context} *ctx"]):
-            content.append(self.substitute_code(self["test-action"]))
+    def _add_function(self, content: CContent, key: str, name: str) -> None:
+        if self[key] is not None:
+            with content.function("static void", f"{self.ident}_{name}",
+                                  [f"{self.context} *ctx"]):
+                content.append(self.substitute_code(self[key]))
+
+    def _add_call(self, content: CContent, key: str, name: str) -> None:
+        if self[key] is not None:
+            content.append(f"{self.ident}_{name}( ctx );")
 
     def _add_loop_body(self, content: CContent) -> None:
         with content.condition(f"{self.ident}_TransitionInfo[ index ].Skip"):
             content.append(["++index;", "continue;"])
         content.add_blank_line()
+        self._add_call(content, "test-prepare", "Prepare")
         for index, enum in enumerate(self._pre_index_to_enum):
             content.append(f"{enum[0]}_Prepare( ctx, ctx->pcs[ {index} ] );")
-        content.append(f"{self.ident}_Action( ctx );")
+        self._add_call(content, "test-action", "Action")
         transition_map = f"{self.ident}_TransitionMap"
         for index, enum in enumerate(self._post_index_to_enum):
             content.append([
@@ -642,7 +648,8 @@ class _TestDirectiveItem(_TestItem):
             f"  .initial_context = &{self.ident}_Instance", "};"
         ])
         self._add_transition_map(content)
-        self._add_action(content)
+        self._add_function(content, "test-prepare", "Prepare")
+        self._add_function(content, "test-action", "Action")
         self._add_test_case(content, header)
         content.add("/** @} */")