fix coverage

Author: qianl15

tests/test_dbos.py

@@ -1201,11 +1201,12 @@ async def hitl_main_loop(prompt: str) -> AgentRunResult[str | DeferredToolReques
         if status.status == 'SUCCESS':
             break
 
+        assert status.status == 'PENDING'
         # Wait and check if the workflow has set a deferred tool request event.
         deferred_tool_requests = await DBOS.get_event_async(
             wf_handle.workflow_id, 'deferred_tool_requests', timeout_seconds=1
         )
-        if deferred_tool_requests is not None:
+        if deferred_tool_requests is not None:  # pragma: no branch
             results = DeferredToolResults()
             # Approve all calls
             for tool_call in deferred_tool_requests.approvals:
@@ -1328,7 +1329,7 @@ def hitl_main_loop_sync(prompt: str) -> AgentRunResult[str | DeferredToolRequest
 
         # Wait and check if the workflow has set a deferred tool request event.
         deferred_tool_requests = DBOS.get_event(wf_handle.workflow_id, 'deferred_tool_requests', timeout_seconds=1)
-        if deferred_tool_requests is not None:
+        if deferred_tool_requests is not None:  # pragma: no branch
             results = DeferredToolResults()
             # Approve all calls
             for tool_call in deferred_tool_requests.approvals: