Add remaining orchestration surface

Author: andystaples

azure-functions-durable/CHANGELOG.md

@@ -31,6 +31,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
   Two-argument (durabletask-native) orchestrators continue to work unchanged.
   `DurableOrchestrationContext.call_http` raises `NotImplementedError` pending a
   durabletask durable-HTTP implementation.
+- `DurableOrchestrationContext` also exposes `custom_status` (reflecting the
+  value set via `set_custom_status`) and `will_continue_as_new` (True once
+  `continue_as_new` has been called). `parent_instance_id`, `function_context`,
+  and `histories` raise `NotImplementedError` because durabletask does not
+  surface that information on the orchestration context.
 
 - Backwards-compatible, deprecated aliases on `DurableFunctionsClient` for the
   v1 `DurableOrchestrationClient` method names: `start_new`, `get_status`,

azure-functions-durable/azure/durable_functions/internal/compat/orchestration_context.py

@@ -27,6 +27,8 @@ class DurableOrchestrationContext:
     def __init__(self, ctx: OrchestrationContext, orchestration_input: Any = None):
         self._ctx = ctx
         self._input = orchestration_input
+        self._custom_status: Any = None
+        self._will_continue_as_new = False
 
     # -- input ---------------------------------------------------------------
     def get_input(self) -> Any:
@@ -49,6 +51,47 @@ def current_utc_datetime(self) -> datetime:
         """Get the replay-safe current UTC date/time."""
         return self._ctx.current_utc_datetime
 
+    @property
+    def custom_status(self) -> Any:
+        """Get the custom status set during this execution (or ``None``)."""
+        return self._custom_status
+
+    @property
+    def will_continue_as_new(self) -> bool:
+        """Whether :meth:`continue_as_new` has been called in this execution."""
+        return self._will_continue_as_new
+
+    @property
+    def parent_instance_id(self) -> str:
+        """Get the ID of the parent orchestration.
+
+        Not available: durabletask does not currently surface the parent
+        instance ID on the orchestration context.
+        """
+        raise NotImplementedError(
+            "parent_instance_id is not currently exposed by durabletask.")
+
+    @property
+    def function_context(self) -> Any:
+        """Get the Azure Functions-level context.
+
+        Not available: durabletask does not provide the v1 ``FunctionContext``
+        binding metadata.
+        """
+        raise NotImplementedError(
+            "function_context is not available in this SDK.")
+
+    @property
+    def histories(self) -> Any:
+        """Get the running history of scheduled tasks.
+
+        Not available: durabletask manages orchestration history internally and
+        does not expose it on the context.
+        """
+        raise NotImplementedError(
+            "histories is not exposed by durabletask; use the client's "
+            "get_orchestration_history instead.")
+
     # -- activities ----------------------------------------------------------
     def call_activity(self, name: Callable[..., Any] | str, input_: Any = None) -> Task[Any]:
         """Schedule an activity function for execution."""
@@ -92,10 +135,12 @@ def wait_for_external_event(self,
     # -- control -------------------------------------------------------------
     def continue_as_new(self, input_: Any) -> None:
         """Restart the orchestration with a new input."""
+        self._will_continue_as_new = True
         self._ctx.continue_as_new(input_)
 
     def set_custom_status(self, status: Any) -> None:
         """Set the orchestration's custom status payload."""
+        self._custom_status = status
         self._ctx.set_custom_status(status)
 
     # -- deterministic IDs ---------------------------------------------------

tests/azure-functions-durable/test_orchestration_context_compat.py

@@ -116,6 +116,44 @@ def test_call_http_raises_not_implemented():
         adapter.call_http("GET", "http://example.com")
 
 
+# ---------------------------------------------------------------------------
+# Additional context members
+# ---------------------------------------------------------------------------
+
+def test_custom_status_tracks_set_custom_status():
+    adapter, fake = _adapter()
+    assert adapter.custom_status is None
+    adapter.set_custom_status({"progress": 50})
+    assert adapter.custom_status == {"progress": 50}
+    fake.set_custom_status.assert_called_once_with({"progress": 50})
+
+
+def test_will_continue_as_new_tracks_continue_as_new():
+    adapter, fake = _adapter()
+    assert adapter.will_continue_as_new is False
+    adapter.continue_as_new({"next": 1})
+    assert adapter.will_continue_as_new is True
+    fake.continue_as_new.assert_called_once_with({"next": 1})
+
+
+def test_parent_instance_id_raises_not_implemented():
+    adapter, _ = _adapter()
+    with pytest.raises(NotImplementedError):
+        _ = adapter.parent_instance_id
+
+
+def test_function_context_raises_not_implemented():
+    adapter, _ = _adapter()
+    with pytest.raises(NotImplementedError):
+        _ = adapter.function_context
+
+
+def test_histories_raises_not_implemented():
+    adapter, _ = _adapter()
+    with pytest.raises(NotImplementedError):
+        _ = adapter.histories
+
+
 # ---------------------------------------------------------------------------
 # Arity detection and wrapping
 # ---------------------------------------------------------------------------