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 dataclasses import dataclass
65from datetime import timedelta
76from enum import Enum
8- from typing import Any , Iterator
9- from unittest .mock import patch
7+ from typing import Any
108
119import pytest
1210
@@ -319,19 +317,6 @@ async def _do_start_update_test(
319317 )
320318 assert await update_handle .result () == "update-result-1"
321319
322- @contextmanager
323- def assert_network_call (
324- self ,
325- expect_network_call : bool ,
326- ) -> Iterator [None ]:
327- with patch .object (
328- self .client .workflow_service ,
329- "poll_workflow_execution_update" ,
330- wraps = self .client .workflow_service .poll_workflow_execution_update ,
331- ) as _wrapped_poll :
332- yield
333- assert _wrapped_poll .called == expect_network_call
334-
335320
336321async def test_update_with_start_sets_first_execution_run_id (
337322 client : Client , env : WorkflowEnvironment
You can’t perform that action at this time.
0 commit comments