Skip to content

Commit fdec5cf

Browse files
committed
Delete unused helper
1 parent 540faeb commit fdec5cf

File tree

1 file changed

+1
-16
lines changed

1 file changed

+1
-16
lines changed

tests/worker/test_update_with_start.py

+1-16
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,9 @@
11
from __future__ import annotations
22

33
import uuid
4-
from contextlib import contextmanager
54
from datetime import timedelta
65
from enum import Enum
7-
from typing import Any, Iterator
8-
from unittest.mock import patch
6+
from typing import Any
97

108
import 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

311296
async def test_update_with_start_sets_first_execution_run_id(
312297
client: Client, env: WorkflowEnvironment

0 commit comments

Comments
 (0)