File tree 1 file changed +1
-16
lines changed
1 file changed +1
-16
lines changed Original file line number Diff line number Diff line change 1
1
from __future__ import annotations
2
2
3
3
import uuid
4
- from contextlib import contextmanager
5
4
from datetime import timedelta
6
5
from enum import Enum
7
- from typing import Any , Iterator
8
- from unittest .mock import patch
6
+ from typing import Any
9
7
10
8
import pytest
11
9
@@ -294,19 +292,6 @@ async def _do_start_update_test(
294
292
)
295
293
assert await update_handle .result () == "update-result-1"
296
294
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
-
310
295
311
296
async def test_update_with_start_sets_first_execution_run_id (
312
297
client : Client , env : WorkflowEnvironment
You can’t perform that action at this time.
0 commit comments