Skip to content

Commit 6ce297a

Browse files
committed
Delete unused helper
1 parent 1a68b58 commit 6ce297a

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,12 +1,10 @@
11
from __future__ import annotations
22

33
import uuid
4-
from contextlib import contextmanager
54
from dataclasses import dataclass
65
from datetime import timedelta
76
from enum import Enum
8-
from typing import Any, Iterator
9-
from unittest.mock import patch
7+
from typing import Any
108

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

336321
async def test_update_with_start_sets_first_execution_run_id(
337322
client: Client, env: WorkflowEnvironment

0 commit comments

Comments
 (0)