python-dependency-injector/tests/typing/selector.py

from typing import Any

from dependency_injector import providers


# Test 1: to check the return type
provider1 = providers.Selector(
    lambda: 'a',
    a=providers.Factory(object),
    b=providers.Factory(object),
)
var1: Any = provider1()

# Test 2: to check the provided instance interface
provider2 = providers.Selector(
    lambda: 'a',
    a=providers.Factory(object),
    b=providers.Factory(object),
)
provided2: providers.ProvidedInstance = provider2.provided
attr_getter2: providers.AttributeGetter = provider2.provided.attr
item_getter2: providers.ItemGetter = provider2.provided['item']
method_caller2: providers.MethodCaller = provider2.provided.method.call(123, arg=324)

# Test3 to check the getattr
provider3 = providers.Selector(
    lambda: 'a',
    a=providers.Factory(object),
    b=providers.Factory(object),
)
attr3: providers.Provider = provider3.a

# Test 4: to check the return type with await
provider4 = providers.Selector(
    lambda: 'a',
    a=providers.Factory(object),
    b=providers.Factory(object),
)
async def _async4() -> None:
    var1: Any = await provider4()  # type: ignore
    var2: Any = await provider4.async_()