"""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 service: Service = Provide["service"] service_provider: Callable[..., Service] = Provider["service"] undefined: Callable = Provide["undefined"] class TestClass: service: Service = Provide["service"] service_provider: Callable[..., Service] = Provider["service"] undefined: Callable = Provide["undefined"] @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_provider_provider(service_provider: Callable[..., Service] = Provider["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[""]): return container.service()