python-dependency-injector/tests/typing/delegate.py
Leonardus Chen 13847c0ce6
Retrofit assert type for some type-stub checks (#943)
Retrofit assert type for:

- configuration.py
- container.py
- declarative_container.py
- delegate.py
- dependencies_container.py
- dependency.py
- dict.py
- dynamic_container.py
2026-01-31 15:34:10 +02:00

25 lines
732 B
Python

from typing import Any, Optional
from typing_extensions import assert_type
from dependency_injector import providers
# Test 1: to check the return type
provider1 = providers.Delegate(providers.Provider())
var1 = provider1()
assert_type(var1, providers.Provider[Any])
# Test 2: to check the return type with await
provider2 = providers.Delegate(providers.Provider())
async def _async2() -> None:
var1 = await provider2() # type: ignore
var2 = await provider2.async_()
assert_type(var2, providers.Provider[Any])
# Test 3: to check class type from provider
provider3 = providers.Delegate(providers.Provider())
provided_provides = provider3.provides
assert_type(provided_provides, Optional[providers.Provider[Any]])