mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2025-11-24 03:45:48 +03:00
45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
from dependency_injector import providers
|
|
from typing_extensions import assert_type, Any
|
|
|
|
|
|
class Animal: ...
|
|
|
|
|
|
class Cat(Animal): ...
|
|
|
|
|
|
# Test 1: to check Aggregate provider
|
|
provider1 = providers.Aggregate(
|
|
a=providers.Object("str1"),
|
|
b=providers.Object("str2"),
|
|
)
|
|
provider_a_1 = provider1.a
|
|
provider_b_1: providers.Provider[str] = provider1.b
|
|
val1 = provider1("a")
|
|
assert_type(provider1, providers.Aggregate[str])
|
|
assert_type(provider_a_1, providers.Provider[str])
|
|
assert_type(provider_b_1, providers.Provider[str])
|
|
assert_type(val1, str)
|
|
|
|
provider1_set_non_string_keys = providers.Aggregate[str]()
|
|
provider1_set_non_string_keys.set_providers({Cat: providers.Object("str")})
|
|
provider_set_non_string_1 = provider1_set_non_string_keys.providers[Cat]
|
|
assert_type(provider_set_non_string_1, providers.Provider[str])
|
|
|
|
|
|
# TODO: Change providers.Aggregate to accept Mapping? Then remove explicit typing here
|
|
provider1_new_non_string_keys = providers.Aggregate[str](
|
|
{Cat: providers.Object("str")},
|
|
)
|
|
factory_new_non_string_1 = provider1_new_non_string_keys.providers[Cat]
|
|
assert_type(provider1_new_non_string_keys, providers.Aggregate[str])
|
|
assert_type(factory_new_non_string_1, providers.Provider[str])
|
|
|
|
|
|
provider1_no_explicit_typing = providers.Aggregate(a=providers.Object("str"))
|
|
provider1_no_explicit_typing_factory = provider1_no_explicit_typing.providers["a"]
|
|
provider1_no_explicit_typing_object = provider1_no_explicit_typing("a")
|
|
|
|
assert_type(provider1_no_explicit_typing_factory, providers.Provider[str])
|
|
assert_type(provider1_no_explicit_typing_object, str)
|