IWF-836: SDK expects stateWaitUntilFailed flag instead stateStartApiSucceeded

Author: samuel27m

iwf-idl

@@ -1,6 +1,6 @@
 import typing
 from dataclasses import dataclass
-from typing import Any, Union
+from typing import Any, Union, Optional
 
 from iwf.errors import WorkflowDefinitionError, NotRegisteredError
 from iwf.iwf_api.models import (
@@ -40,6 +40,7 @@ class CommandResults:
     timer_commands: list[TimerCommandResult]
     internal_channel_commands: list[InternalChannelCommandResult]
     signal_channel_commands: list[SignalChannelCommandResult]
+    wait_until_api_succeeded: Optional[bool] = None
 
 
 def from_idl_command_results(
@@ -48,9 +49,10 @@ def from_idl_command_results(
     signal_channel_types: dict[str, typing.Optional[type]],
     object_encoder: ObjectEncoder,
 ) -> CommandResults:
-    results = CommandResults(list(), list(), list())
+    results = CommandResults(list(), list(), list(), None)
     if isinstance(idl_results, Unset):
         return results
+
     if not isinstance(idl_results.timer_results, Unset):
         for timer in idl_results.timer_results:
             results.timer_commands.append(
@@ -91,4 +93,10 @@ def from_idl_command_results(
                     sig.command_id,
                 )
             )
+
+    if not isinstance(idl_results.state_wait_until_failed, Unset):
+        # The server will set state_wait_until_failed to true if the waitUntil API failed.
+        # Hence, flag inversion is needed here to indicate that the waitUntil API succeeded.
+        results.wait_until_api_succeeded = not idl_results.state_wait_until_failed
+
     return results

iwf/command_results.py

@@ -23,12 +23,14 @@ class CommandResults:
         inter_state_channel_results (Union[Unset, list['InterStateChannelResult']]):
         timer_results (Union[Unset, list['TimerResult']]):
         state_start_api_succeeded (Union[Unset, bool]):
+        state_wait_until_failed (Union[Unset, bool]):
     """
 
     signal_results: Union[Unset, list["SignalResult"]] = UNSET
     inter_state_channel_results: Union[Unset, list["InterStateChannelResult"]] = UNSET
     timer_results: Union[Unset, list["TimerResult"]] = UNSET
     state_start_api_succeeded: Union[Unset, bool] = UNSET
+    state_wait_until_failed: Union[Unset, bool] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -55,6 +57,8 @@ def to_dict(self) -> dict[str, Any]:
 
         state_start_api_succeeded = self.state_start_api_succeeded
 
+        state_wait_until_failed = self.state_wait_until_failed
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
@@ -66,6 +70,8 @@ def to_dict(self) -> dict[str, Any]:
             field_dict["timerResults"] = timer_results
         if state_start_api_succeeded is not UNSET:
             field_dict["stateStartApiSucceeded"] = state_start_api_succeeded
+        if state_wait_until_failed is not UNSET:
+            field_dict["stateWaitUntilFailed"] = state_wait_until_failed
 
         return field_dict
 
@@ -99,11 +105,14 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
 
         state_start_api_succeeded = d.pop("stateStartApiSucceeded", UNSET)
 
+        state_wait_until_failed = d.pop("stateWaitUntilFailed", UNSET)
+
         command_results = cls(
             signal_results=signal_results,
             inter_state_channel_results=inter_state_channel_results,
             timer_results=timer_results,
             state_start_api_succeeded=state_start_api_succeeded,
+            state_wait_until_failed=state_wait_until_failed,
         )
 
         command_results.additional_properties = d