From ec78d03b9b378509268bf680dcf8ab55de0544af Mon Sep 17 00:00:00 2001 From: Roman Mogylatov Date: Sat, 8 Jan 2022 19:39:51 -0500 Subject: [PATCH] Add typing tests --- tests/typing/aggregate.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 tests/typing/aggregate.py diff --git a/tests/typing/aggregate.py b/tests/typing/aggregate.py new file mode 100644 index 00000000..993634f7 --- /dev/null +++ b/tests/typing/aggregate.py @@ -0,0 +1,32 @@ +from dependency_injector import providers + + +class Animal: + ... + + +class Cat(Animal): + ... + + +# Test 1: to check Aggregate provider +provider1: providers.Aggregate[str] = providers.Aggregate( + a=providers.Object("str1"), + b=providers.Object("str2"), +) +provider_a_1: providers.Provider[str] = provider1.a +provider_b_1: providers.Provider[str] = provider1.b +val1: str = provider1("a") + +provider1_set_non_string_keys: providers.Aggregate[str] = providers.Aggregate() +provider1_set_non_string_keys.set_providers({Cat: providers.Object("str")}) +provider_set_non_string_1: providers.Provider[str] = provider1_set_non_string_keys.providers[Cat] + +provider1_new_non_string_keys: providers.Aggregate[str] = providers.Aggregate( + {Cat: providers.Object("str")}, +) +factory_new_non_string_1: providers.Provider[str] = provider1_new_non_string_keys.providers[Cat] + +provider1_no_explicit_typing = providers.Aggregate(a=providers.Object("str")) +provider1_no_explicit_typing_factory: providers.Provider[str] = provider1_no_explicit_typing.providers["a"] +provider1_no_explicit_typing_object: str = provider1_no_explicit_typing("a")