test(examples): update wait for condition example

Author: Alex Wang

.gitignore

@@ -29,6 +29,7 @@ dist/
 .kiro/
 .idea
 .env
+.env*
 
 .durable_executions
 

examples/src/wait_for_condition/wait_for_condition.py

@@ -4,30 +4,25 @@
 
 from aws_durable_execution_sdk_python.context import DurableContext
 from aws_durable_execution_sdk_python.execution import durable_execution
+from aws_durable_execution_sdk_python.waits import WaitForConditionConfig
 
 
 @durable_execution
 def handler(_event: Any, context: DurableContext) -> int:
     """Handler demonstrating wait-for-condition pattern."""
-    state = 0
-    attempt = 0
-    max_attempts = 5
 
-    while attempt < max_attempts:
-        attempt += 1
+    def condition_function(state: int) -> int:
+        """Increment state by 1."""
+        return state + 1
 
-        # Execute step to update state
-        state = context.step(
-            lambda _, s=state: s + 1,
-            name=f"increment_state_{attempt}",
-        )
-
-        # Check condition
+    def wait_strategy(state: int, attempt: int) -> dict[str, Any]:
+        """Wait strategy that continues until state reaches 3."""
         if state >= 3:
-            # Condition met, stop
-            break
+            return {"shouldContinue": False}
+        return {"shouldContinue": True, "delay_seconds": 1}
+
+    config = WaitForConditionConfig(wait_strategy=wait_strategy, initial_state=0)
 
-        # Wait before next attempt
-        context.wait(seconds=1)
+    result = context.wait_for_condition(check=condition_function, config=config)
 
-    return state
+    return result

examples/test/wait_for_condition/test_wait_for_condition.py

@@ -2,7 +2,6 @@
 
 import pytest
 from aws_durable_execution_sdk_python.execution import InvocationStatus
-from aws_durable_execution_sdk_python.lambda_service import OperationType
 
 from src.wait_for_condition import wait_for_condition
 from test.conftest import deserialize_operation_payload
@@ -21,13 +20,3 @@ def test_wait_for_condition(durable_runner):
     assert result.status is InvocationStatus.SUCCEEDED
     # Should reach state 3 after 3 increments
     assert deserialize_operation_payload(result.result) == 3
-
-    # Verify step operations exist (should have 3 increment steps)
-    step_ops = [
-        op for op in result.operations if op.operation_type == OperationType.STEP
-    ]
-    assert len(step_ops) == 3
-
-    # Verify wait operations exist (should have 2 waits before final state)
-    wait_ops = [op for op in result.operations if op.operation_type.value == "WAIT"]
-    assert len(wait_ops) == 2