d33db9fff8
Co-Authored-By: Quentin Torroba <quentin.torroba@mistral.ai> Co-Authored-By: Michel Thomazo <michel.thomazo@mistral.ai> Co-Authored-By: Clément Drouin <clement.drouin@mistral.ai> Co-Authored-By: Vincent Guilloux <vincent.guilloux@mistral.ai> Co-Authored-By: Clément Siriex <clement.sirieix@mistral.ai> Co-Authored-By: Kim-Adeline Miguel <kimadeline.miguel@mistral.ai> Co-Authored-By: Thaddee Tyl <thaddee.tyl@gmail.com> Co-Authored-By: David Brochart <david.brochart@gmail.com> Co-Authored-By: Joseph Guhlin <joseph.guhlin@gmail.com> Co-Authored-By: Thomas Kenbeek <thomaskenbeek@gmail.com> Co-Authored-By: Remenby31 <baptiste.cruvellier31@gmail.com>
22 lines
604 B
Python
22 lines
604 B
Python
from __future__ import annotations
|
|
|
|
from vibe.cli.update_notifier.ports.update_gateway import (
|
|
Update,
|
|
UpdateGateway,
|
|
UpdateGatewayError,
|
|
)
|
|
|
|
|
|
class FakeUpdateGateway(UpdateGateway):
|
|
def __init__(
|
|
self, update: Update | None = None, error: UpdateGatewayError | None = None
|
|
) -> None:
|
|
self._update: Update | None = update
|
|
self._error = error
|
|
self.fetch_update_calls = 0
|
|
|
|
async def fetch_update(self) -> Update | None:
|
|
self.fetch_update_calls += 1
|
|
if self._error is not None:
|
|
raise self._error
|
|
return self._update
|