Test using stores in a context manager

Author: maxrjones

src/zarr/testing/store.py

@@ -100,6 +100,15 @@ async def test_store_open_read_only(self, open_kwargs: dict[str, Any], read_only
         assert store._is_open
         assert store.read_only == read_only
 
+    async def test_store_context_manager(self, open_kwargs: dict[str, Any]) -> None:
+        # Test that the context manager closes the store
+        with await self.store_cls.open(**open_kwargs) as store:
+            assert store._is_open
+            # Test trying to open an already open store
+            with pytest.raises(ValueError):
+                await store._open()
+        assert not store._is_open
+
     async def test_read_only_store_raises(self, open_kwargs: dict[str, Any]) -> None:
         kwargs = {**open_kwargs, "read_only": True}
         store = await self.store_cls.open(**kwargs)