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: