Update agent_persist_mock to patch _exists method

Ostorlab · Feb 24, 2025 · 6819446 · 6819446
1 parent 6f7af64
commit 6819446
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ostorlab/testing/agent.py b/src/ostorlab/testing/agent.py
@@ -94,7 +94,7 @@ def _get(key):
         if key in storage:
             return storage[key]
 
-    def _add(key: Union[str, bytes], value: bytes):
+    def _add(key: Union[str, bytes], value: bytes) -> None:
         """Check values are present in the storage dict."""
         if isinstance(value, bytes) is False:
             value = str(value).encode()