Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 1 addition & 15 deletions tests/worker/test_update_with_start.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,10 @@

import asyncio
import uuid
from contextlib import contextmanager
from dataclasses import dataclass
from datetime import timedelta
from enum import Enum, IntEnum
from typing import Any, Iterator, Mapping, Optional
from typing import Any, Mapping, Optional
from unittest.mock import patch

import temporalio.api.common.v1
Expand Down Expand Up @@ -328,19 +327,6 @@ async def _do_start_update_test(
)
assert await update_handle.result() == "update-result-1"

@contextmanager
def assert_network_call(
self,
expect_network_call: bool,
) -> Iterator[None]:
with patch.object(
self.client.workflow_service,
"poll_workflow_execution_update",
wraps=self.client.workflow_service.poll_workflow_execution_update,
) as _wrapped_poll:
yield
assert _wrapped_poll.called == expect_network_call


async def test_update_with_start_sets_first_execution_run_id(
client: Client, env: WorkflowEnvironment
Expand Down
Loading