Fix tests

Author: Peter-Sh

src/tests/test_state.py

@@ -290,24 +290,28 @@ def test_ephemeral_field_exists(self) -> None:
         workflow = Workflow(workflow_file="test.yml")
 
         assert hasattr(workflow, "ephemeral")
-        assert workflow.ephemeral.trigger_failed is False
-        assert workflow.ephemeral.timed_out is False
+        assert workflow.ephemeral.trigger_workflow is None
+        assert workflow.ephemeral.wait_for_completion is None
 
     def test_ephemeral_field_can_be_modified(self) -> None:
         """Test that ephemeral field values can be modified."""
+        from py_trees.common import Status
+
         workflow = Workflow(workflow_file="test.yml")
 
-        workflow.ephemeral.trigger_failed = True
-        workflow.ephemeral.timed_out = True
+        workflow.ephemeral.trigger_workflow = Status.FAILURE
+        workflow.ephemeral.wait_for_completion = Status.RUNNING
 
-        assert workflow.ephemeral.trigger_failed is True
-        assert workflow.ephemeral.timed_out is True
+        assert workflow.ephemeral.trigger_workflow == Status.FAILURE
+        assert workflow.ephemeral.wait_for_completion == Status.RUNNING
 
     def test_ephemeral_field_not_serialized_to_json(self) -> None:
         """Test that ephemeral field is serialized but log_once_flags are excluded."""
+        from py_trees.common import Status
+
         workflow = Workflow(workflow_file="test.yml")
-        workflow.ephemeral.trigger_failed = True
-        workflow.ephemeral.timed_out = True
+        workflow.ephemeral.trigger_workflow = Status.FAILURE
+        workflow.ephemeral.wait_for_completion = Status.SUCCESS
         workflow.ephemeral.log_once_flags["test_flag"] = True
 
         # Serialize to JSON
@@ -316,8 +320,8 @@ def test_ephemeral_field_not_serialized_to_json(self) -> None:
 
         # Verify ephemeral field IS in JSON (except log_once_flags)
         assert "ephemeral" in json_data
-        assert json_data["ephemeral"]["trigger_failed"] is True
-        assert json_data["ephemeral"]["timed_out"] is True
+        assert json_data["ephemeral"]["trigger_workflow"] == "FAILURE"
+        assert json_data["ephemeral"]["wait_for_completion"] == "SUCCESS"
         assert "log_once_flags" not in json_data["ephemeral"]
 
         # Verify other fields are present
@@ -326,16 +330,18 @@ def test_ephemeral_field_not_serialized_to_json(self) -> None:
 
     def test_ephemeral_field_not_in_model_dump(self) -> None:
         """Test that ephemeral field is in model_dump but log_once_flags are excluded."""
+        from py_trees.common import Status
+
         workflow = Workflow(workflow_file="test.yml")
-        workflow.ephemeral.trigger_failed = True
+        workflow.ephemeral.trigger_workflow = Status.SUCCESS
         workflow.ephemeral.log_once_flags["test_flag"] = True
 
         # Get dict representation
         data = workflow.model_dump()
 
         # Verify ephemeral field IS in dict (except log_once_flags)
         assert "ephemeral" in data
-        assert data["ephemeral"]["trigger_failed"] is True
+        assert data["ephemeral"]["trigger_workflow"] == Status.SUCCESS
         assert "log_once_flags" not in data["ephemeral"]
 
     def test_ephemeral_field_initialized_on_deserialization(self) -> None:
@@ -346,11 +352,13 @@ def test_ephemeral_field_initialized_on_deserialization(self) -> None:
 
         # Ephemeral field should be initialized with defaults
         assert hasattr(workflow, "ephemeral")
-        assert workflow.ephemeral.trigger_failed is False
-        assert workflow.ephemeral.timed_out is False
+        assert workflow.ephemeral.trigger_workflow is None
+        assert workflow.ephemeral.wait_for_completion is None
 
     def test_release_state_ephemeral_not_serialized(self) -> None:
         """Test that ephemeral fields are serialized but log_once_flags are excluded."""
+        from py_trees.common import Status
+
         config = Config(
             version=1,
             packages={
@@ -366,8 +374,10 @@ def test_release_state_ephemeral_not_serialized(self) -> None:
         state = ReleaseState.from_config(config)
 
         # Modify ephemeral fields
-        state.packages["test-package"].build.ephemeral.trigger_failed = True
-        state.packages["test-package"].publish.ephemeral.timed_out = True
+        state.packages["test-package"].build.ephemeral.trigger_workflow = Status.FAILURE
+        state.packages["test-package"].publish.ephemeral.wait_for_completion = (
+            Status.SUCCESS
+        )
         state.packages["test-package"].build.ephemeral.log_once_flags["test"] = True
 
         # Serialize to JSON
@@ -379,10 +389,10 @@ def test_release_state_ephemeral_not_serialized(self) -> None:
         publish_workflow = json_data["packages"]["test-package"]["publish"]
 
         assert "ephemeral" in build_workflow
-        assert build_workflow["ephemeral"]["trigger_failed"] is True
+        assert build_workflow["ephemeral"]["trigger_workflow"] == "FAILURE"
         assert "log_once_flags" not in build_workflow["ephemeral"]
         assert "ephemeral" in publish_workflow
-        assert publish_workflow["ephemeral"]["timed_out"] is True
+        assert publish_workflow["ephemeral"]["wait_for_completion"] == "SUCCESS"
         assert "log_once_flags" not in publish_workflow["ephemeral"]