mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2025-02-07 15:10:47 +03:00
feed916f46
* Add support of async injections into wiring * Add support of async functions and async generators for resources * Update resource provider typing stub for stutdown * Add resource base class for async resources * Fix tests * Add tests for async injections in wiring @inject * Refactor provider tests * Add tests for async resources * Rework async resources callbacks to .add_done_callback() style (fixes pypy3 issue) * Add awaits into async resource class test * Refactor FastAPI tests * Implement async resources initialization in container * Move container async resource tests to a separate module for Python 3.6+ * Fix init async resources in container on Python 2 * Add first dirty async injections implementation * Fix isawaitable error * Turm asyncio import to conditional for safer Py2 usage * Refactor kwargs injections * Implement positional injections, add tests and make refactoring * Implement attribute injections and add tests * Add singleton implementation + tests for all singleton types * Implement injections in thread-local and thread-safe singleton providers * Update .provided + fix resource concurent initialization issue * Implement async mode for Dependency provider * Add async mode for the provider * Add overload for Factory typing * Add typing stubs for async resource * Refactor abstract* providers __call__() * Add async mode API + tests * Add typing stubs & tests for async mode API * Add tests for async mode auto configuration * Refactor Provider.__call__() to use async mode api * Refactor Dependency provider to use async mode api * Add tests for Dependency provider async mode * Add support of async mode for FactoryAggregate provider + tests * Refactor Singleton provider to use async mode api * Refactor ThreadSafeSingleton provider to use async mode api * Refactor ThreadLocalSingleton provider to use async mode api * Finish Singleton refactoring to use async mode api * Refactor Resource provider to use async mode api * Add Provider.async_() method + tests * Add typing stubs for async_() method + tests * Refactor Singleton typing stubs to return singleton from argument methods * Refactor provider typing stubs * Improve resource typing stub * Add tests for async context kwargs injections * Fix typo in resource provider tests * Cover shutdown of not initialized resource * Add test to cover resource initialization with an error * Fix Singleton and ThreadLocalSingleton to handle initialization errors * Add FastAPI + Redis example * Make cosmetic fixes to FastAPI + Redis example * Add missing development requirements * Update module docblock in fastapi + redis example * Add FastAPI + Redis example docs * Add references to FastAPI + Redis example * Refactor resource docs * Add asynchronous resources docs * Refactor wiring docs * Add async injections docs for wiring * Add async injections page and update docs index, readme, and key features pages * Add providers async injections example * Add docs on provider async mode enabling * Reword async provider docs * Add provider async mode docs * Add cross links to async docs * Mute flake8 errors in async provider examples * Update changelog * Make cosmetic fix to containers.pyx
332 lines
11 KiB
Python
332 lines
11 KiB
Python
from decimal import Decimal
|
|
import unittest
|
|
|
|
from dependency_injector.wiring import wire, Provide, Closing
|
|
|
|
# Runtime import to avoid syntax errors in samples on Python < 3.5
|
|
import os
|
|
_TOP_DIR = os.path.abspath(
|
|
os.path.sep.join((
|
|
os.path.dirname(__file__),
|
|
'../',
|
|
)),
|
|
)
|
|
_SAMPLES_DIR = os.path.abspath(
|
|
os.path.sep.join((
|
|
os.path.dirname(__file__),
|
|
'../samples/',
|
|
)),
|
|
)
|
|
import sys
|
|
sys.path.append(_TOP_DIR)
|
|
sys.path.append(_SAMPLES_DIR)
|
|
|
|
from asyncutils import AsyncTestCase
|
|
|
|
from wiringsamples import module, package
|
|
from wiringsamples.service import Service
|
|
from wiringsamples.container import Container, SubContainer
|
|
|
|
|
|
class WiringTest(unittest.TestCase):
|
|
|
|
container: Container
|
|
|
|
def setUp(self) -> None:
|
|
self.container = Container(config={'a': {'b': {'c': 10}}})
|
|
self.container.wire(
|
|
modules=[module],
|
|
packages=[package],
|
|
)
|
|
self.addCleanup(self.container.unwire)
|
|
|
|
def test_package_lookup(self):
|
|
from wiringsamples.package import test_package_function
|
|
service = test_package_function()
|
|
self.assertIsInstance(service, Service)
|
|
|
|
def test_package_subpackage_lookup(self):
|
|
from wiringsamples.package.subpackage import test_package_function
|
|
service = test_package_function()
|
|
self.assertIsInstance(service, Service)
|
|
|
|
def test_package_submodule_lookup(self):
|
|
from wiringsamples.package.subpackage.submodule import test_function
|
|
service = test_function()
|
|
self.assertIsInstance(service, Service)
|
|
|
|
def test_class_wiring(self):
|
|
test_class_object = module.TestClass()
|
|
self.assertIsInstance(test_class_object.service, Service)
|
|
|
|
def test_class_wiring_context_arg(self):
|
|
test_service = self.container.service()
|
|
|
|
test_class_object = module.TestClass(service=test_service)
|
|
self.assertIs(test_class_object.service, test_service)
|
|
|
|
def test_class_method_wiring(self):
|
|
test_class_object = module.TestClass()
|
|
service = test_class_object.method()
|
|
self.assertIsInstance(service, Service)
|
|
|
|
def test_class_classmethod_wiring(self):
|
|
service = module.TestClass.class_method()
|
|
self.assertIsInstance(service, Service)
|
|
|
|
def test_instance_classmethod_wiring(self):
|
|
instance = module.TestClass()
|
|
service = instance.class_method()
|
|
self.assertIsInstance(service, Service)
|
|
|
|
def test_class_staticmethod_wiring(self):
|
|
service = module.TestClass.static_method()
|
|
self.assertIsInstance(service, Service)
|
|
|
|
def test_instance_staticmethod_wiring(self):
|
|
instance = module.TestClass()
|
|
service = instance.static_method()
|
|
self.assertIsInstance(service, Service)
|
|
|
|
def test_function_wiring(self):
|
|
service = module.test_function()
|
|
self.assertIsInstance(service, Service)
|
|
|
|
def test_function_wiring_context_arg(self):
|
|
test_service = self.container.service()
|
|
|
|
service = module.test_function(service=test_service)
|
|
self.assertIs(service, test_service)
|
|
|
|
def test_function_wiring_provider(self):
|
|
service = module.test_function_provider()
|
|
self.assertIsInstance(service, Service)
|
|
|
|
def test_function_wiring_provider_context_arg(self):
|
|
test_service = self.container.service()
|
|
|
|
service = module.test_function_provider(service_provider=lambda: test_service)
|
|
self.assertIs(service, test_service)
|
|
|
|
def test_configuration_option(self):
|
|
int_value, str_value, decimal_value = module.test_config_value()
|
|
self.assertEqual(int_value, 10)
|
|
self.assertEqual(str_value, '10')
|
|
self.assertEqual(decimal_value, Decimal(10))
|
|
|
|
def test_provide_provider(self):
|
|
service = module.test_provide_provider()
|
|
self.assertIsInstance(service, Service)
|
|
|
|
def test_provided_instance(self):
|
|
class TestService:
|
|
foo = {
|
|
'bar': lambda: 10,
|
|
}
|
|
|
|
with self.container.service.override(TestService()):
|
|
some_value = module.test_provided_instance()
|
|
self.assertEqual(some_value, 10)
|
|
|
|
def test_subcontainer(self):
|
|
some_value = module.test_subcontainer_provider()
|
|
self.assertEqual(some_value, 1)
|
|
|
|
def test_config_invariant(self):
|
|
config = {
|
|
'option': {
|
|
'a': 1,
|
|
'b': 2,
|
|
},
|
|
'switch': 'a',
|
|
}
|
|
self.container.config.from_dict(config)
|
|
|
|
with self.container.config.switch.override('a'):
|
|
value_a = module.test_config_invariant()
|
|
self.assertEqual(value_a, 1)
|
|
|
|
with self.container.config.switch.override('b'):
|
|
value_b = module.test_config_invariant()
|
|
self.assertEqual(value_b, 2)
|
|
|
|
def test_wire_with_class_error(self):
|
|
with self.assertRaises(Exception):
|
|
wire(
|
|
container=Container,
|
|
modules=[module],
|
|
)
|
|
|
|
def test_unwire_function(self):
|
|
self.container.unwire()
|
|
self.assertIsInstance(module.test_function(), Provide)
|
|
|
|
def test_unwire_class(self):
|
|
self.container.unwire()
|
|
test_class_object = module.TestClass()
|
|
self.assertIsInstance(test_class_object.service, Provide)
|
|
|
|
def test_unwire_class_method(self):
|
|
self.container.unwire()
|
|
test_class_object = module.TestClass()
|
|
self.assertIsInstance(test_class_object.method(), Provide)
|
|
|
|
def test_unwire_package_function(self):
|
|
self.container.unwire()
|
|
from wiringsamples.package.subpackage.submodule import test_function
|
|
self.assertIsInstance(test_function(), Provide)
|
|
|
|
def test_unwire_package_function_by_reference(self):
|
|
from wiringsamples.package.subpackage import submodule
|
|
self.container.unwire()
|
|
self.assertIsInstance(submodule.test_function(), Provide)
|
|
|
|
def test_wire_multiple_containers(self):
|
|
sub_container = SubContainer()
|
|
sub_container.wire(
|
|
modules=[module],
|
|
packages=[package],
|
|
)
|
|
self.addCleanup(sub_container.unwire)
|
|
|
|
service, some_value = module.test_provide_from_different_containers()
|
|
|
|
self.assertIsInstance(service, Service)
|
|
self.assertEqual(some_value, 1)
|
|
|
|
def test_closing_resource(self):
|
|
from wiringsamples import resourceclosing
|
|
|
|
resourceclosing.Service.reset_counter()
|
|
|
|
container = resourceclosing.Container()
|
|
container.wire(modules=[resourceclosing])
|
|
self.addCleanup(container.unwire)
|
|
|
|
result_1 = resourceclosing.test_function()
|
|
self.assertIsInstance(result_1, resourceclosing.Service)
|
|
self.assertEqual(result_1.init_counter, 1)
|
|
self.assertEqual(result_1.shutdown_counter, 1)
|
|
|
|
result_2 = resourceclosing.test_function()
|
|
self.assertIsInstance(result_2, resourceclosing.Service)
|
|
self.assertEqual(result_2.init_counter, 2)
|
|
self.assertEqual(result_2.shutdown_counter, 2)
|
|
|
|
self.assertIsNot(result_1, result_2)
|
|
|
|
def test_closing_resource_context(self):
|
|
from wiringsamples import resourceclosing
|
|
|
|
resourceclosing.Service.reset_counter()
|
|
service = resourceclosing.Service()
|
|
|
|
container = resourceclosing.Container()
|
|
container.wire(modules=[resourceclosing])
|
|
self.addCleanup(container.unwire)
|
|
|
|
result_1 = resourceclosing.test_function(service=service)
|
|
self.assertIs(result_1, service)
|
|
self.assertEqual(result_1.init_counter, 0)
|
|
self.assertEqual(result_1.shutdown_counter, 0)
|
|
|
|
result_2 = resourceclosing.test_function(service=service)
|
|
self.assertIs(result_2, service)
|
|
self.assertEqual(result_2.init_counter, 0)
|
|
self.assertEqual(result_2.shutdown_counter, 0)
|
|
|
|
def test_class_decorator(self):
|
|
service = module.test_class_decorator()
|
|
self.assertIsInstance(service, Service)
|
|
|
|
|
|
class WiringAndFastAPITest(unittest.TestCase):
|
|
|
|
container: Container
|
|
|
|
def test_bypass_marker_injection(self):
|
|
container = Container()
|
|
container.wire(modules=[module])
|
|
self.addCleanup(container.unwire)
|
|
|
|
service = module.test_function(service=Provide[Container.service])
|
|
self.assertIsInstance(service, Service)
|
|
|
|
def test_closing_resource_bypass_marker_injection(self):
|
|
from wiringsamples import resourceclosing
|
|
|
|
resourceclosing.Service.reset_counter()
|
|
|
|
container = resourceclosing.Container()
|
|
container.wire(modules=[resourceclosing])
|
|
self.addCleanup(container.unwire)
|
|
|
|
result_1 = resourceclosing.test_function(
|
|
service=Closing[Provide[resourceclosing.Container.service]],
|
|
)
|
|
self.assertIsInstance(result_1, resourceclosing.Service)
|
|
self.assertEqual(result_1.init_counter, 1)
|
|
self.assertEqual(result_1.shutdown_counter, 1)
|
|
|
|
result_2 = resourceclosing.test_function(
|
|
service=Closing[Provide[resourceclosing.Container.service]],
|
|
)
|
|
self.assertIsInstance(result_2, resourceclosing.Service)
|
|
self.assertEqual(result_2.init_counter, 2)
|
|
self.assertEqual(result_2.shutdown_counter, 2)
|
|
|
|
self.assertIsNot(result_1, result_2)
|
|
|
|
|
|
class WiringAsyncInjectionsTest(AsyncTestCase):
|
|
|
|
def test_async_injections(self):
|
|
from wiringsamples import asyncinjections
|
|
|
|
container = asyncinjections.Container()
|
|
container.wire(modules=[asyncinjections])
|
|
self.addCleanup(container.unwire)
|
|
|
|
asyncinjections.resource1.reset_counters()
|
|
asyncinjections.resource2.reset_counters()
|
|
|
|
resource1, resource2 = self._run(asyncinjections.async_injection())
|
|
|
|
self.assertIs(resource1, asyncinjections.resource1)
|
|
self.assertEqual(asyncinjections.resource1.init_counter, 1)
|
|
self.assertEqual(asyncinjections.resource1.shutdown_counter, 0)
|
|
|
|
self.assertIs(resource2, asyncinjections.resource2)
|
|
self.assertEqual(asyncinjections.resource2.init_counter, 1)
|
|
self.assertEqual(asyncinjections.resource2.shutdown_counter, 0)
|
|
|
|
def test_async_injections_with_closing(self):
|
|
from wiringsamples import asyncinjections
|
|
|
|
container = asyncinjections.Container()
|
|
container.wire(modules=[asyncinjections])
|
|
self.addCleanup(container.unwire)
|
|
|
|
asyncinjections.resource1.reset_counters()
|
|
asyncinjections.resource2.reset_counters()
|
|
|
|
resource1, resource2 = self._run(asyncinjections.async_injection_with_closing())
|
|
|
|
self.assertIs(resource1, asyncinjections.resource1)
|
|
self.assertEqual(asyncinjections.resource1.init_counter, 1)
|
|
self.assertEqual(asyncinjections.resource1.shutdown_counter, 1)
|
|
|
|
self.assertIs(resource2, asyncinjections.resource2)
|
|
self.assertEqual(asyncinjections.resource2.init_counter, 1)
|
|
self.assertEqual(asyncinjections.resource2.shutdown_counter, 1)
|
|
|
|
resource1, resource2 = self._run(asyncinjections.async_injection_with_closing())
|
|
|
|
self.assertIs(resource1, asyncinjections.resource1)
|
|
self.assertEqual(asyncinjections.resource1.init_counter, 2)
|
|
self.assertEqual(asyncinjections.resource1.shutdown_counter, 2)
|
|
|
|
self.assertIs(resource2, asyncinjections.resource2)
|
|
self.assertEqual(asyncinjections.resource2.init_counter, 2)
|
|
self.assertEqual(asyncinjections.resource2.shutdown_counter, 2)
|