Fix windows tests

Author: strawgate

key-value/key-value-aio/tests/stores/valkey/test_valkey.py

@@ -9,7 +9,6 @@
 from typing_extensions import override
 
 from key_value.aio.stores.base import BaseStore
-from key_value.aio.stores.valkey import ValkeyStore
 from tests.conftest import detect_on_windows, docker_container, should_skip_docker_tests
 from tests.stores.base import (
     BaseStoreTests,
@@ -35,10 +34,6 @@ class ValkeyFailedToStartError(Exception):
     pass
 
 
-def get_valkey_client_from_store(store: ValkeyStore):
-    return store._connected_client  # pyright: ignore[reportPrivateUsage, reportReturnType]
-
-
 @pytest.mark.skipif(should_skip_docker_tests(), reason="Docker is not running")
 @pytest.mark.skipif(detect_on_windows(), reason="Valkey is not supported on Windows")
 class TestValkeyStore(ContextManagerStoreTestMixin, BaseStoreTests):
@@ -93,9 +88,13 @@ async def store(self, setup_valkey: None):
     @override
     async def test_not_unbounded(self, store: BaseStore): ...
 
-    async def test_value_stored(self, store: ValkeyStore):
+    async def test_value_stored(self, store: BaseStore):
+        from key_value.aio.stores.valkey import ValkeyStore
+
         await store.put(collection="test", key="test_key", value={"name": "Alice", "age": 30})
 
+        assert isinstance(store, ValkeyStore)
+
         valkey_client = store._connected_client  # pyright: ignore[reportPrivateUsage]
         assert valkey_client is not None
         value = await valkey_client.get(key="test::test_key")

key-value/key-value-sync/tests/code_gen/stores/valkey/test_valkey.py

@@ -12,7 +12,6 @@
 from typing_extensions import override
 
 from key_value.sync.code_gen.stores.base import BaseStore
-from key_value.sync.code_gen.stores.valkey import ValkeyStore
 from tests.code_gen.conftest import detect_on_windows, docker_container, should_skip_docker_tests
 from tests.code_gen.stores.base import BaseStoreTests, ContextManagerStoreTestMixin
 
@@ -33,10 +32,6 @@ class ValkeyFailedToStartError(Exception):
     pass
 
 
-def get_valkey_client_from_store(store: ValkeyStore):
-    return store._connected_client  # pyright: ignore[reportPrivateUsage, reportReturnType]
-
-
 @pytest.mark.skipif(should_skip_docker_tests(), reason="Docker is not running")
 @pytest.mark.skipif(detect_on_windows(), reason="Valkey is not supported on Windows")
 class TestValkeyStore(ContextManagerStoreTestMixin, BaseStoreTests):
@@ -91,9 +86,13 @@ def store(self, setup_valkey: None):
     @override
     def test_not_unbounded(self, store: BaseStore): ...
 
-    def test_value_stored(self, store: ValkeyStore):
+    def test_value_stored(self, store: BaseStore):
+        from key_value.sync.code_gen.stores.valkey import ValkeyStore
+
         store.put(collection="test", key="test_key", value={"name": "Alice", "age": 30})
 
+        assert isinstance(store, ValkeyStore)
+
         valkey_client = store._connected_client  # pyright: ignore[reportPrivateUsage]
         assert valkey_client is not None
         value = valkey_client.get(key="test::test_key")