Add async context manager resource cleanup tests

Author: maldoinc

test/unit/test_container_generator_factory.py

@@ -12,6 +12,21 @@
 from test.conftest import Container
 
 
+class AsyncContextManagerResource:
+    def __init__(self, cleanup_events: list[str]) -> None:
+        self.cleanup_events = cleanup_events
+        self.entered = False
+        self.exited = False
+
+    async def __aenter__(self) -> "AsyncContextManagerResource":
+        self.entered = True
+        return self
+
+    async def __aexit__(self, *_: object) -> None:
+        self.exited = True
+        self.cleanup_events.append("resource")
+
+
 def test_cleans_up_on_exit(container: Container) -> None:
     _cleanup_performed = False
     Something = NewType("Something", str)
@@ -50,6 +65,26 @@ async def target(smth: Injected[Something]):
     assert _cleanup_performed
 
 
+async def test_async_generator_factory_closes_async_context_manager_on_container_close() -> None:
+    cleanup_events: list[str] = []
+
+    @injectable
+    async def some_factory() -> AsyncIterator[AsyncContextManagerResource]:
+        async with AsyncContextManagerResource(cleanup_events) as resource:
+            yield resource
+
+    container = wireup.create_async_container(injectables=[some_factory])
+    resource = await container.get(AsyncContextManagerResource)
+
+    assert resource.entered
+    assert not resource.exited
+
+    await container.close()
+
+    assert resource.exited
+    assert cleanup_events == ["resource"]
+
+
 def test_injects_transient() -> None:
     _cleanups: list[str] = []
     Something = NewType("Something", str)
@@ -103,6 +138,28 @@ def f2(something: Something) -> Iterator[SomethingElse]:
     assert _cleanups == ["f2", "f1"]
 
 
+@pytest.mark.parametrize("lifetime", ["scoped", "transient"])
+async def test_async_generator_factory_closes_async_context_manager_on_scope_exit(
+    lifetime: InjectableLifetime,
+) -> None:
+    cleanup_events: list[str] = []
+
+    @injectable(lifetime=lifetime)
+    async def some_factory() -> AsyncIterator[AsyncContextManagerResource]:
+        async with AsyncContextManagerResource(cleanup_events) as resource:
+            yield resource
+
+    container = wireup.create_async_container(injectables=[some_factory])
+
+    async with container.enter_scope() as scoped:
+        resource = await scoped.get(AsyncContextManagerResource)
+        assert resource.entered
+        assert not resource.exited
+
+    assert resource.exited
+    assert cleanup_events == ["resource"]
+
+
 def test_cleans_up_in_order() -> None:
     _cleanups: list[str] = []
     Something = NewType("Something", str)