test: Override MongoDB sanitization tests to handle default passthrough behavior

With sanitization now off by default (PassthroughStrategy), MongoDB tests
need to handle two scenarios:
1. Default store (no sanitization) - should fail on long/special chars
2. Sanitizing store (with V1 strategy) - should succeed

Changes:
- Add sanitizing_store fixture with MongoDBV1CollectionSanitizationStrategy
- Override test_long_collection_name to test both behaviors
- Override test_special_characters_in_collection_name to test both behaviors
- Update test_mongodb_collection_name_sanitization to use sanitizing_store
- Apply to both TestMongoDBStoreNativeMode and TestMongoDBStoreNonNativeMode

Co-authored-by: William Easton <strawgate@users.noreply.github.com>

Author: github-actions[bot]

key-value/key-value-aio/tests/stores/mongodb/test_mongodb.py

@@ -14,7 +14,10 @@
 
 from key_value.aio.stores.base import BaseStore
 from key_value.aio.stores.mongodb import MongoDBStore
-from key_value.aio.stores.mongodb.store import MongoDBSerializationAdapter
+from key_value.aio.stores.mongodb.store import (
+    MongoDBSerializationAdapter,
+    MongoDBV1CollectionSanitizationStrategy,
+)
 from tests.conftest import docker_container, should_skip_docker_tests
 from tests.stores.base import BaseStoreTests, ContextManagerStoreTestMixin
 
@@ -118,12 +121,30 @@ async def setup_mongodb(self, request: pytest.FixtureRequest) -> AsyncGenerator[
     @override
     async def test_not_unbounded(self, store: BaseStore): ...
 
-    async def test_mongodb_collection_name_sanitization(self, store: MongoDBStore):
-        """Tests that a special characters in the collection name will not raise an error."""
-        await store.put(collection="test_collection!@#$%^&*()", key="test_key", value={"test": "test"})
-        assert await store.get(collection="test_collection!@#$%^&*()", key="test_key") == {"test": "test"}
+    @override
+    async def test_long_collection_name(self, store: MongoDBStore, sanitizing_store: MongoDBStore):  # pyright: ignore[reportIncompatibleMethodOverride]
+        """Tests that a long collection name will not raise an error."""
+        with pytest.raises(Exception):  # noqa: B017, PT011
+            await store.put(collection="test_collection" * 100, key="test_key", value={"test": "test"})
+
+        await sanitizing_store.put(collection="test_collection" * 100, key="test_key", value={"test": "test"})
+        assert await sanitizing_store.get(collection="test_collection" * 100, key="test_key") == {"test": "test"}
+
+    @override
+    async def test_special_characters_in_collection_name(self, store: MongoDBStore, sanitizing_store: MongoDBStore):  # pyright: ignore[reportIncompatibleMethodOverride]
+        """Tests that special characters in the collection name will not raise an error."""
+        with pytest.raises(Exception):  # noqa: B017, PT011
+            await store.put(collection="test_collection!@#$%^&*()", key="test_key", value={"test": "test"})
+
+        await sanitizing_store.put(collection="test_collection!@#$%^&*()", key="test_key", value={"test": "test"})
+        assert await sanitizing_store.get(collection="test_collection!@#$%^&*()", key="test_key") == {"test": "test"}
 
-        collections = store._collections_by_name.values()
+    async def test_mongodb_collection_name_sanitization(self, sanitizing_store: MongoDBStore):
+        """Tests that the V1 sanitization strategy produces the expected collection names."""
+        await sanitizing_store.put(collection="test_collection!@#$%^&*()", key="test_key", value={"test": "test"})
+        assert await sanitizing_store.get(collection="test_collection!@#$%^&*()", key="test_key") == {"test": "test"}
+
+        collections = sanitizing_store._collections_by_name.values()
         collection_names = [collection.name for collection in collections]
         assert collection_names == snapshot(["S_test_collection_-daf4a2ec"])
 
@@ -141,6 +162,19 @@ async def store(self, setup_mongodb: None) -> MongoDBStore:
 
         return store
 
+    @pytest.fixture
+    async def sanitizing_store(self, setup_mongodb: None) -> MongoDBStore:
+        store = MongoDBStore(
+            url=f"mongodb://{MONGODB_HOST}:{MONGODB_HOST_PORT}",
+            db_name=f"{MONGODB_TEST_DB}-native-sanitizing",
+            native_storage=True,
+            collection_sanitization_strategy=MongoDBV1CollectionSanitizationStrategy(),
+        )
+
+        await clean_mongodb_database(store=store)
+
+        return store
+
     async def test_value_stored_as_bson_dict(self, store: MongoDBStore):
         """Verify values are stored as BSON dicts, not JSON strings."""
         await store.put(collection="test", key="test_key", value={"name": "Alice", "age": 30})
@@ -190,6 +224,19 @@ async def store(self, setup_mongodb: None) -> MongoDBStore:
 
         return store
 
+    @pytest.fixture
+    async def sanitizing_store(self, setup_mongodb: None) -> MongoDBStore:
+        store = MongoDBStore(
+            url=f"mongodb://{MONGODB_HOST}:{MONGODB_HOST_PORT}",
+            db_name=f"{MONGODB_TEST_DB}-sanitizing",
+            native_storage=False,
+            collection_sanitization_strategy=MongoDBV1CollectionSanitizationStrategy(),
+        )
+
+        await clean_mongodb_database(store=store)
+
+        return store
+
     async def test_value_stored_as_json(self, store: MongoDBStore):
         """Verify values are stored as JSON strings."""
         await store.put(collection="test", key="test_key", value={"name": "Alice", "age": 30})

key-value/key-value-sync/tests/code_gen/stores/mongodb/test_mongodb.py

@@ -17,7 +17,7 @@
 
 from key_value.sync.code_gen.stores.base import BaseStore
 from key_value.sync.code_gen.stores.mongodb import MongoDBStore
-from key_value.sync.code_gen.stores.mongodb.store import MongoDBSerializationAdapter
+from key_value.sync.code_gen.stores.mongodb.store import MongoDBSerializationAdapter, MongoDBV1CollectionSanitizationStrategy
 from tests.code_gen.conftest import docker_container, should_skip_docker_tests
 from tests.code_gen.stores.base import BaseStoreTests, ContextManagerStoreTestMixin
 
@@ -119,12 +119,30 @@ def setup_mongodb(self, request: pytest.FixtureRequest) -> Generator[None, None,
     @override
     def test_not_unbounded(self, store: BaseStore): ...
 
-    def test_mongodb_collection_name_sanitization(self, store: MongoDBStore):
-        """Tests that a special characters in the collection name will not raise an error."""
-        store.put(collection="test_collection!@#$%^&*()", key="test_key", value={"test": "test"})
-        assert store.get(collection="test_collection!@#$%^&*()", key="test_key") == {"test": "test"}
+    @override
+    def test_long_collection_name(self, store: MongoDBStore, sanitizing_store: MongoDBStore):  # pyright: ignore[reportIncompatibleMethodOverride]
+        "Tests that a long collection name will not raise an error."
+        with pytest.raises(Exception):  # noqa: B017, PT011
+            store.put(collection="test_collection" * 100, key="test_key", value={"test": "test"})
+
+        sanitizing_store.put(collection="test_collection" * 100, key="test_key", value={"test": "test"})
+        assert sanitizing_store.get(collection="test_collection" * 100, key="test_key") == {"test": "test"}
+
+    @override
+    def test_special_characters_in_collection_name(self, store: MongoDBStore, sanitizing_store: MongoDBStore):  # pyright: ignore[reportIncompatibleMethodOverride]
+        "Tests that special characters in the collection name will not raise an error."
+        with pytest.raises(Exception):  # noqa: B017, PT011
+            store.put(collection="test_collection!@#$%^&*()", key="test_key", value={"test": "test"})
+
+        sanitizing_store.put(collection="test_collection!@#$%^&*()", key="test_key", value={"test": "test"})
+        assert sanitizing_store.get(collection="test_collection!@#$%^&*()", key="test_key") == {"test": "test"}
 
-        collections = store._collections_by_name.values()
+    def test_mongodb_collection_name_sanitization(self, sanitizing_store: MongoDBStore):
+        """Tests that the V1 sanitization strategy produces the expected collection names."""
+        sanitizing_store.put(collection="test_collection!@#$%^&*()", key="test_key", value={"test": "test"})
+        assert sanitizing_store.get(collection="test_collection!@#$%^&*()", key="test_key") == {"test": "test"}
+
+        collections = sanitizing_store._collections_by_name.values()
         collection_names = [collection.name for collection in collections]
         assert collection_names == snapshot(["S_test_collection_-daf4a2ec"])
 
@@ -142,6 +160,19 @@ def store(self, setup_mongodb: None) -> MongoDBStore:
 
         return store
 
+    @pytest.fixture
+    def sanitizing_store(self, setup_mongodb: None) -> MongoDBStore:
+        store = MongoDBStore(
+            url=f"mongodb://{MONGODB_HOST}:{MONGODB_HOST_PORT}",
+            db_name=f"{MONGODB_TEST_DB}-native-sanitizing",
+            native_storage=True,
+            collection_sanitization_strategy=MongoDBV1CollectionSanitizationStrategy(),
+        )
+
+        clean_mongodb_database(store=store)
+
+        return store
+
     def test_value_stored_as_bson_dict(self, store: MongoDBStore):
         """Verify values are stored as BSON dicts, not JSON strings."""
         store.put(collection="test", key="test_key", value={"name": "Alice", "age": 30})
@@ -186,6 +217,19 @@ def store(self, setup_mongodb: None) -> MongoDBStore:
 
         return store
 
+    @pytest.fixture
+    def sanitizing_store(self, setup_mongodb: None) -> MongoDBStore:
+        store = MongoDBStore(
+            url=f"mongodb://{MONGODB_HOST}:{MONGODB_HOST_PORT}",
+            db_name=f"{MONGODB_TEST_DB}-sanitizing",
+            native_storage=False,
+            collection_sanitization_strategy=MongoDBV1CollectionSanitizationStrategy(),
+        )
+
+        clean_mongodb_database(store=store)
+
+        return store
+
     def test_value_stored_as_json(self, store: MongoDBStore):
         """Verify values are stored as JSON strings."""
         store.put(collection="test", key="test_key", value={"name": "Alice", "age": 30})