diff --git a/tests/test_default_args.py b/tests/test_default_args.py index ed5b3bc..58d5243 100644 --- a/tests/test_default_args.py +++ b/tests/test_default_args.py @@ -43,3 +43,28 @@ def test_default_consumer_args(): assert dep == (123, '42') +def test_default_inj_args(): + 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)) + + @repo.mark_injector() + def int_to_a(i: int, opt_dep: optC = '42') -> A: + return A(i + int(opt_dep)) + + 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 + + fn2 = repo.get_conversion((int,), consumer, force_commutative=True, force_async=False, allow_async=False) + dep = fn2(123) + assert dep == 123 + 42