fix: TestContainer.override created cache generated during override

Author: notypecheck

aioinject/testing/testcontainer.py

@@ -95,7 +95,9 @@ def _exit(self) -> None:
             self.registry.providers[self.info.interface] = self.prev
 
         if self.prev_cache is not None:
+            new_cache = self.container.root.cache
             self.container.root.cache = self.prev_cache
+            self.container.root.cache.update(new_cache)
 
     def _clear_provider(
         self, providers: list[ProviderRecord[object]] | None

tests/testing/test_testcontainer.py

@@ -92,3 +92,22 @@ class B:
         b_scoped = await ctx.resolve(B)
         assert b_scoped is not b
         assert b.a is a
+
+
+async def test_should_not_remove_unrelated_objects_from_cache() -> None:
+    class A:
+        pass
+
+    class B:
+        pass
+
+    container = Container()
+    container.register(Singleton(A), Singleton(B))
+    testcontainer = TestContainer(container)
+
+    async with testcontainer.override(Singleton(B)):
+        a = await container.root.resolve(A)
+        b = await container.root.resolve(B)
+
+    assert await container.root.resolve(A) is a
+    assert await container.root.resolve(B) is not b