mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2024-11-26 03:23:58 +03:00
8dd8446d39
* Add container injections to wiring * Add example * Update docs * Update changelog * Improve typing
98 lines
2.4 KiB
Python
98 lines
2.4 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
|
|
|
|
|
|
def test_container(container: Container = Provide[Container]):
|
|
return container.service()
|