Add context manager tests

tests/test_ctxmanager.py

@@ -0,0 +1,86 @@
+from contextlib import contextmanager, asynccontextmanager
+from dataclasses import dataclass
+from typing import Any, Generator, AsyncGenerator
+
+import pytest
+
+from src.breakshaft.convertor import ConvRepo
+
+pytest_plugins = ('pytest_asyncio',)
+
+
+@dataclass
+class A:
+    a: int
+
+
+@dataclass
+class B:
+    b: float
+
+
+def test_sync_ctxmanager():
+    repo = ConvRepo()
+
+    @repo.mark_injector()
+    def b_to_a(b: B) -> A:
+        return A(int(b.b))
+
+    @repo.mark_injector()
+    def a_to_b(a: A) -> B:
+        return B(float(a.a))
+
+    int_to_a_finalized = [False]
+
+    @repo.mark_injector()
+    @contextmanager
+    def int_to_a(i: int) -> Generator[A, Any, None]:
+        yield A(i)
+        int_to_a_finalized[0] = True
+
+    def consumer(dep: A) -> int:
+        return dep.a
+
+    fn1 = repo.get_conversion((B,), consumer, force_commutative=True, force_async=False, allow_async=False)
+    dep = fn1(B(42.1))
+    assert dep == 42
+    assert not int_to_a_finalized[0]
+
+    fn2 = repo.get_conversion((int,), consumer, force_commutative=True, force_async=False, allow_async=False)
+    dep = fn2(123)
+    assert dep == 123
+    assert int_to_a_finalized[0]
+
+
+@pytest.mark.asyncio
+async def test_async_ctxmanager():
+    repo = ConvRepo()
+
+    @repo.mark_injector()
+    def b_to_a(b: B) -> A:
+        return A(int(b.b))
+
+    @repo.mark_injector()
+    def a_to_b(a: A) -> B:
+        return B(float(a.a))
+
+    int_to_a_finalized = [False]
+
+    @repo.mark_injector()
+    @asynccontextmanager
+    async def int_to_a(i: int) -> AsyncGenerator[A, Any]:
+        yield A(i)
+        int_to_a_finalized[0] = True
+
+    def consumer(dep: A) -> int:
+        return dep.a
+
+    fn1 = repo.get_conversion((B,), consumer, force_commutative=True, force_async=False, allow_async=True)
+    dep = fn1(B(42.1))
+    assert dep == 42
+    assert not int_to_a_finalized[0]
+
+    fn2 = repo.get_conversion((int,), consumer, force_commutative=True, force_async=False, allow_async=True)
+    dep = await fn2(123)
+    assert dep == 123
+    assert int_to_a_finalized[0]