Skip to content

Commit 42135f0

Browse files
authored
Delete unused helper (#710)
1 parent c643e8a commit 42135f0

File tree

1 file changed

+1
-15
lines changed

1 file changed

+1
-15
lines changed

tests/worker/test_update_with_start.py

Lines changed: 1 addition & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,10 @@
22

33
import asyncio
44
import uuid
5-
from contextlib import contextmanager
65
from dataclasses import dataclass
76
from datetime import timedelta
87
from enum import Enum, IntEnum
9-
from typing import Any, Iterator, Mapping, Optional
8+
from typing import Any, Mapping, Optional
109
from unittest.mock import patch
1110

1211
import temporalio.api.common.v1
@@ -328,19 +327,6 @@ async def _do_start_update_test(
328327
)
329328
assert await update_handle.result() == "update-result-1"
330329

331-
@contextmanager
332-
def assert_network_call(
333-
self,
334-
expect_network_call: bool,
335-
) -> Iterator[None]:
336-
with patch.object(
337-
self.client.workflow_service,
338-
"poll_workflow_execution_update",
339-
wraps=self.client.workflow_service.poll_workflow_execution_update,
340-
) as _wrapped_poll:
341-
yield
342-
assert _wrapped_poll.called == expect_network_call
343-
344330

345331
async def test_update_with_start_sets_first_execution_run_id(
346332
client: Client, env: WorkflowEnvironment

0 commit comments

Comments
 (0)