Merge pull request #37 from Enapter/test-generator-coverage-9721216632657680690

🧪 Add tests for async_.generator

Author: rnovatorov

tests/unit/test_async/test_generator.py

@@ -16,3 +16,46 @@ async def agen():
 
     with pytest.raises(StopAsyncIteration):
         await g.__anext__()
+
+
+async def test_generator_closes_on_exception_in_context() -> None:
+    closed = False
+
+    @enapter.async_.generator
+    async def agen():
+        nonlocal closed
+        try:
+            yield 1
+            yield 2  # pragma: no cover
+        finally:
+            closed = True
+
+    with pytest.raises(ValueError, match="Test error"):
+        async with agen() as g:
+            assert await g.__anext__() == 1
+            raise ValueError("Test error")
+
+    assert closed
+
+
+async def test_generator_raises_exception() -> None:
+    @enapter.async_.generator
+    async def agen():
+        yield 1
+        raise ValueError("Generator error")
+
+    async with agen() as g:
+        assert await g.__anext__() == 1
+        with pytest.raises(ValueError, match="Generator error"):
+            await g.__anext__()
+
+
+async def test_generator_with_args() -> None:
+    @enapter.async_.generator
+    async def agen(a: int, b: int = 0):
+        yield a
+        yield b
+
+    async with agen(10, b=20) as g:
+        assert await g.__anext__() == 10
+        assert await g.__anext__() == 20