mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2024-11-23 10:03:56 +03:00
131 lines
3.1 KiB
Python
131 lines
3.1 KiB
Python
"""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()
|