"""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