mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2024-12-02 06:24:03 +03:00
ae3024588c
* Bump version to 4.3.9: FastAPI example * Reengineer wiring * Add @inject decorator * Add .workspace dir to gitignore * Add generic typing for @inject * Add type cast for @inject * Update movie lister example * Update cli application tutorial * Update demo example * Update wiring docs and examples * Update aiohttp example and tutorial * Update multiple containers example * Update single container example * Update decoupled packages example * Update django example * Update asyncio daemon example and tutorial * Update FastAPI example * Update flask example and tutorial * Update sanic example * Add wiring registry * Add new line to .gitignore * Add @inject to the test samples * Fix flake8 errors
94 lines
2.3 KiB
Python
94 lines
2.3 KiB
Python
"""Test module for wiring."""
|
|
|
|
from decimal import Decimal
|
|
from typing import Callable
|
|
|
|
from dependency_injector.wiring import inject, Provide, Provider
|
|
|
|
from .container import Container, SubContainer
|
|
from .service import Service
|
|
|
|
|
|
class TestClass:
|
|
|
|
@inject
|
|
def __init__(self, service: Service = Provide[Container.service]):
|
|
self.service = service
|
|
|
|
@inject
|
|
def method(self, service: Service = Provide[Container.service]):
|
|
return service
|
|
|
|
@classmethod
|
|
@inject
|
|
def class_method(cls, service: Service = Provide[Container.service]):
|
|
return service
|
|
|
|
@staticmethod
|
|
@inject
|
|
def static_method(service: Service = Provide[Container.service]):
|
|
return service
|
|
|
|
|
|
@inject
|
|
def test_function(service: Service = Provide[Container.service]):
|
|
return service
|
|
|
|
|
|
@inject
|
|
def test_function_provider(service_provider: Callable[..., Service] = Provider[Container.service]):
|
|
service = service_provider()
|
|
return service
|
|
|
|
|
|
@inject
|
|
def test_config_value(
|
|
some_value_int: int = Provide[Container.config.a.b.c.as_int()],
|
|
some_value_str: str = Provide[Container.config.a.b.c.as_(str)],
|
|
some_value_decimal: Decimal = Provide[Container.config.a.b.c.as_(Decimal)],
|
|
):
|
|
return some_value_int, some_value_str, some_value_decimal
|
|
|
|
|
|
@inject
|
|
def test_provide_provider(service_provider: Callable[..., Service] = Provider[Container.service.provider]):
|
|
service = service_provider()
|
|
return service
|
|
|
|
|
|
@inject
|
|
def test_provided_instance(some_value: int = Provide[Container.service.provided.foo['bar'].call()]):
|
|
return some_value
|
|
|
|
|
|
@inject
|
|
def test_subcontainer_provider(some_value: int = Provide[Container.sub.int_object]):
|
|
return some_value
|
|
|
|
|
|
@inject
|
|
def test_config_invariant(some_value: int = Provide[Container.config.option[Container.config.switch]]):
|
|
return some_value
|
|
|
|
|
|
@inject
|
|
def test_provide_from_different_containers(
|
|
service: Service = Provide[Container.service],
|
|
some_value: int = Provide[SubContainer.int_object],
|
|
):
|
|
return service, some_value
|
|
|
|
|
|
class ClassDecorator:
|
|
def __init__(self, fn):
|
|
self._fn = fn
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
return self._fn(*args, **kwargs)
|
|
|
|
|
|
@ClassDecorator
|
|
@inject
|
|
def test_class_decorator(service: Service = Provide[Container.service]):
|
|
return service
|