File tree Expand file tree Collapse file tree 1 file changed +1
-16
lines changed Expand file tree Collapse file tree 1 file changed +1
-16
lines changed Original file line number Diff line number Diff line change 11from __future__ import annotations
22
33import uuid
4- from contextlib import contextmanager
54from datetime import timedelta
65from enum import Enum
7- from typing import Any , Iterator
8- from unittest .mock import patch
6+ from typing import Any
97
108import pytest
119
@@ -294,19 +292,6 @@ async def _do_start_update_test(
294292 )
295293 assert await update_handle .result () == "update-result-1"
296294
297- @contextmanager
298- def assert_network_call (
299- self ,
300- expect_network_call : bool ,
301- ) -> Iterator [None ]:
302- with patch .object (
303- self .client .workflow_service ,
304- "poll_workflow_execution_update" ,
305- wraps = self .client .workflow_service .poll_workflow_execution_update ,
306- ) as _wrapped_poll :
307- yield
308- assert _wrapped_poll .called == expect_network_call
309-
310295
311296async def test_update_with_start_sets_first_execution_run_id (
312297 client : Client , env : WorkflowEnvironment
You can’t perform that action at this time.
0 commit comments