Add get and set methods to test class

Author: maxrjones

src/zarr/testing/store.py

@@ -23,15 +23,15 @@ class StoreTests(Generic[S, B]):
     async def set(self, store: S, key: str, value: Buffer) -> None:
         """
         Insert a value into a storage backend, with a specific key.
-        This should not not use any store methods. Bypassing the store methods allows them to be
+        This should not use any store methods. Bypassing the store methods allows them to be
         tested.
         """
         raise NotImplementedError
 
     async def get(self, store: S, key: str) -> Buffer:
         """
         Retrieve a value from a storage backend, by key.
-        This should not not use any store methods. Bypassing the store methods allows them to be
+        This should not use any store methods. Bypassing the store methods allows them to be
         tested.
         """
 

tests/test_store/test_object.py

@@ -3,10 +3,14 @@
 
 obstore = pytest.importorskip("obstore")
 
-from zarr.core.buffer import cpu
+import re
+
+from zarr.core.buffer import Buffer, cpu
 from zarr.storage.object_store import ObjectStore
 from zarr.testing.store import StoreTests
 
+PATTERN = r"file://(/[\w/.-]+)"
+
 
 class TestObjectStore(StoreTests[ObjectStore, cpu.Buffer]):
     store_cls = ObjectStore
@@ -21,6 +25,18 @@ def store_kwargs(self, tmpdir) -> dict[str, str | bool]:
     def store(self, store_kwargs: dict[str, str | bool]) -> ObjectStore:
         return self.store_cls(**store_kwargs)
 
+    async def get(self, store: ObjectStore, key: str) -> Buffer:
+        # TODO: There must be a better way to get the path to the store
+        store_path = re.search(PATTERN, str(store)).group(1)
+        new_local_store = obstore.store.LocalStore(prefix=store_path)
+        return self.buffer_cls.from_bytes(obstore.get(new_local_store, key))
+
+    async def set(self, store: ObjectStore, key: str, value: Buffer) -> None:
+        # TODO: There must be a better way to get the path to the store
+        store_path = re.search(PATTERN, str(store)).group(1)
+        new_local_store = obstore.store.LocalStore(prefix=store_path)
+        obstore.put(new_local_store, key, value.to_bytes())
+
     def test_store_repr(self, store: ObjectStore) -> None:
         from fnmatch import fnmatch