Skip to content

Commit d7fbbdc

Browse files
committed
make store test class get and set async
1 parent b5dd50c commit d7fbbdc

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

tests/v3/test_store/test_local.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,10 @@ class TestLocalStore(StoreTests[LocalStore, cpu.Buffer]):
1818
store_cls = LocalStore
1919
buffer_cls = cpu.Buffer
2020

21-
def get(self, store: LocalStore, key: str) -> Buffer:
21+
async def get(self, store: LocalStore, key: str) -> Buffer:
2222
return self.buffer_cls.from_bytes((Path(store.path) / key).read_bytes())
2323

24-
def set(self, store: LocalStore, key: str, value: Buffer) -> None:
24+
async def set(self, store: LocalStore, key: str, value: Buffer) -> None:
2525
target = Path(store.path) / key
2626
parent = target.parent
2727
if not parent.exists():

tests/v3/test_store/test_memory.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,10 @@ class TestMemoryStore(StoreTests[MemoryStore, cpu.Buffer]):
1212
store_cls = MemoryStore
1313
buffer_cls = cpu.Buffer
1414

15-
def set(self, store: MemoryStore, key: str, value: Buffer) -> None:
15+
async def set(self, store: MemoryStore, key: str, value: Buffer) -> None:
1616
store._store_dict[store.resolve_key(key)] = value
1717

18-
def get(self, store: MemoryStore, key: str) -> Buffer:
18+
async def get(self, store: MemoryStore, key: str) -> Buffer:
1919
return store._store_dict[store.resolve_key(key)]
2020

2121
@pytest.fixture(params=[None, True])

tests/v3/test_store/test_zip.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def store_kwargs(self, request) -> dict[str, str | bool]:
3232
async def get(self, store: ZipStore, key: str) -> Buffer:
3333
return store._get(key, prototype=default_buffer_prototype())
3434

35-
def set(self, store: ZipStore, key: str, value: Buffer) -> None:
35+
async def set(self, store: ZipStore, key: str, value: Buffer) -> None:
3636
return store._set(store.resolve_key(key), value)
3737

3838
def test_store_mode(self, store: ZipStore, store_kwargs: dict[str, Any]) -> None:

0 commit comments

Comments
 (0)