python-dependency-injector/tests/unit/samples/wiringsamples/module.py
2020-10-28 13:44:11 -04:00

68 lines
1.9 KiB
Python

"""Test module for wiring."""
from decimal import Decimal
from typing import Callable
from dependency_injector.wiring import Provide, Provider
from .container import Container, SubContainer
from .service import Service
class TestClass:
def __init__(self, service: Service = Provide[Container.service]):
self.service = service
def method(self, service: Service = Provide[Container.service]):
return service
@classmethod
def class_method(cls, service: Service = Provide[Container.service]):
return service
@staticmethod
def static_method(service: Service = Provide[Container.service]):
return service
def test_function(service: Service = Provide[Container.service]):
return service
def test_function_provider(service_provider: Callable[..., Service] = Provider[Container.service]):
service = service_provider()
return service
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
def test_provide_provider(service_provider: Callable[..., Service] = Provider[Container.service.provider]):
service = service_provider()
return service
def test_provided_instance(some_value: int = Provide[Container.service.provided.foo['bar'].call()]):
return some_value
def test_subcontainer_provider(some_value: int = Provide[Container.sub.int_object]):
return some_value
def test_config_invariant(some_value: int = Provide[Container.config.option[Container.config.switch]]):
return some_value
def test_provide_from_different_containers(
service: Service = Provide[Container.service],
some_value: int = Provide[SubContainer.int_object],
):
return service, some_value