Add stub for the Selector provider

This commit is contained in:
Roman Mogylatov 2020-08-26 17:58:56 -04:00
parent ce412980d2
commit 60e614ba31
2 changed files with 39 additions and 0 deletions

View File

@ -268,6 +268,16 @@ class Container(Provider):
def __getattr__(self, name: str) -> Provider: ...
class Selector(Provider):
def __init__(self, selector: _Callable[..., Any], **providers: Provider): ...
def __call__(self, *args: Injection, **kwargs: Injection) -> Any: ...
def __getattr__(self, name: str) -> Provider: ...
@property
def provided(self) -> ProvidedInstance: ...
@property
def providers(self) -> Dict[str, Provider]: ...
class ProvidedInstanceFluentInterface:
def __getattr__(self, item: str) -> AttributeGetter: ...
def __getitem__(self, item: str) -> ItemGetter: ...

29
tests/typing/selector.py Normal file
View File

@ -0,0 +1,29 @@
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: int = 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