python-dependency-injector/tests/unit/samples/wiringstringidssamples/module.py

131 lines
3.1 KiB
Python
Raw Normal View History

"""Test module for wiring."""
from decimal import Decimal
from typing import Callable
from dependency_injector.wiring import (
inject,
Provide,
Provider,
as_int,
as_float,
as_,
required,
invariant,
provided,
)
from .container import Container
from .service import Service
class TestClass:
@inject
def __init__(self, service: Service = Provide['service']):
self.service = service
@inject
def method(self, service: Service = Provide['service']):
return service
@classmethod
@inject
def class_method(cls, service: Service = Provide['service']):
return service
@staticmethod
@inject
def static_method(service: Service = Provide['service']):
return service
@inject
def test_function(service: Service = Provide['service']):
return service
@inject
def test_function_provider(service_provider: Callable[..., Service] = Provider['service']):
service = service_provider()
return service
@inject
def test_config_value(
value_int: int = Provide['config.a.b.c', as_int()],
value_float: float = Provide['config.a.b.c', as_float()],
value_str: str = Provide['config.a.b.c', as_(str)],
value_decimal: Decimal = Provide['config.a.b.c', as_(Decimal)],
value_required: str = Provide['config.a.b.c', required()],
value_required_int: int = Provide['config.a.b.c', required().as_int()],
value_required_float: float = Provide['config.a.b.c', required().as_float()],
value_required_str: str = Provide['config.a.b.c', required().as_(str)],
value_required_decimal: str = Provide['config.a.b.c', required().as_(Decimal)],
):
return (
value_int,
value_float,
value_str,
value_decimal,
value_required,
value_required_int,
value_required_float,
value_required_str,
value_required_decimal,
)
@inject
def test_config_value_required_undefined(
value_required: int = Provide['config.a.b.c', required()],
):
return value_required
@inject
def test_provide_provider(service_provider: Callable[..., Service] = Provide['service.provider']):
service = service_provider()
return service
@inject
def test_provided_instance(some_value: int = Provide['service', provided().foo['bar'].call()]):
return some_value
@inject
def test_subcontainer_provider(some_value: int = Provide['sub.int_object']):
return some_value
@inject
def test_config_invariant(some_value: int = Provide['config.option', invariant('config.switch')]):
return some_value
@inject
def test_provide_from_different_containers(
service: Service = Provide['service'],
some_value: int = Provide['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['service']):
return service
def test_container(container: Container = Provide['<container>']):
return container.service()