v1.1.3

Co-Authored-By: Quentin Torroba <quentin.torroba@mistral.ai>
Co-Authored-By: Michel Thomazo <michel.thomazo@mistral.ai>
Co-Authored-By: Vincent Guilloux <vincent.guilloux@mistral.ai>

tests/update_notifier/adapters/fake_update_cache_repository.py

@@ -0,0 +1,17 @@
+from __future__ import annotations
+
+from vibe.cli.update_notifier.ports.update_cache_repository import (
+    UpdateCache,
+    UpdateCacheRepository,
+)
+
+
+class FakeUpdateCacheRepository(UpdateCacheRepository):
+    def __init__(self, update_cache: UpdateCache | None = None) -> None:
+        self.update_cache: UpdateCache | None = update_cache
+
+    async def get(self) -> UpdateCache | None:
+        return self.update_cache
+
+    async def set(self, update_cache: UpdateCache) -> None:
+        self.update_cache = update_cache

tests/update_notifier/adapters/fake_version_update_gateway.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+from vibe.cli.update_notifier.ports.version_update_gateway import (
+    VersionUpdate,
+    VersionUpdateGateway,
+    VersionUpdateGatewayError,
+)
+
+
+class FakeVersionUpdateGateway(VersionUpdateGateway):
+    def __init__(
+        self,
+        update: VersionUpdate | None = None,
+        error: VersionUpdateGatewayError | None = None,
+    ) -> None:
+        self._update: VersionUpdate | None = update
+        self._error = error
+        self.fetch_update_calls = 0
+
+    async def fetch_update(self) -> VersionUpdate | None:
+        self.fetch_update_calls += 1
+        if self._error is not None:
+            raise self._error
+        return self._update