Class FakeWork

tests/lib/environments/test_poller_method.py:44–65

Sync resource fake. `poll_script` is consumed in order; values are either `BetaSelfHostedWork`-shaped stubs, `None` (no work available), `Exception` instances (raised from poll), or callables producing one of those (so a script step can also move a fake clock).

class FakeWork:
    """Sync resource fake. `poll_script` is consumed in order; values are
    either `BetaSelfHostedWork`-shaped stubs, `None` (no work available),
    `Exception` instances (raised from poll), or callables producing one of
    those (so a script step can also move a fake clock).
    """

    def __init__(self, poll_script: list[Any]) -> None:
        self._poll_script = list(poll_script)
        self.poll_calls: list[dict[str, Any]] = []
        self.ack_calls: list[tuple[str, dict[str, Any]]] = []
        self.stop_calls: list[tuple[str, dict[str, Any]]] = []

    def poll(self, environment_id: str, **kwargs: Any) -> Any:
        self.poll_calls.append({"environment_id": environment_id, **kwargs})
        return _next_poll_result(self._poll_script)

    def ack(self, work_id: str, **kwargs: Any) -> None:
        self.ack_calls.append((work_id, kwargs))

    def stop(self, work_id: str, **kwargs: Any) -> None:
        self.stop_calls.append((work_id, kwargs))


class FakeAsyncWork: