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

"""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_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()