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 dataclasses import dataclass
6
5
from datetime import timedelta
7
6
from enum import Enum
8
- from typing import Any , Iterator
9
- from unittest .mock import patch
7
+ from typing import Any
10
8
11
9
import pytest
12
10
@@ -319,19 +317,6 @@ async def _do_start_update_test(
319
317
)
320
318
assert await update_handle .result () == "update-result-1"
321
319
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
-
335
320
336
321
async def test_update_with_start_sets_first_execution_run_id (
337
322
client : Client , env : WorkflowEnvironment
You can’t perform that action at this time.
0 commit comments