mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2024-11-24 18:43:58 +03:00
94aca21fb8
* Add pytest and pytest-asyncio to the requirements * Update aiohttp ext test * Update setup.cfg * Update tox.ini * Add pytest to the tox requirements * Update tox.ini * Move configuration to tox.ini * Add pytest configs * Rename pytest-py34-py35.ini -> pytest-py35.ini * Update config file paths * Update makefile * Migrate common tests to pytest * Migrate FastAPI and Flask wiring tests * Rename flask and fastapi wiring test files * Move wiring autoloader tests * Add pytest-asyncio to the tox.ini * Migrate wiring async injection tests * Migrate main wiring tests * Migrate wiring string module and package names tests * Migrate wiring config tests * Migrate misc wiring tests * Update tests structure * Migrate misc wiring tests * Refactor container.from_schema() API tests * Migrate container.from_schema() integration tests * Rename schema samples * Update sample imports * Migrate container tests * Refactor container tests * Migrate container self tests * Migrate container instance tests * Migrate container custom string attribute name tests * Migrate container async resource tests * Fix py2 container tests * Migrate container cls tests * Migrate container class custom string cls as atrribute name tests * Migrate ext.aiohttp tests * Migrate ext.flasks tests * Update ext package tests doc block * Migrate provider utils tests * Migrate Factory async mode tests * Migrate async tests * Rename common test module * Refactor asserts in provider tests * Migrate factory tests * Migrate selector provider tests * Migrate object provider tests * Migrate self provider tests * Migrate delegate provider tests * Migrate provider tests * Migrate dependency provider tests * Migrate dependencies container provider tests * Fix warnings * Migrate list provider tests * Migrate dict provider tests * Migrate callable tests * Migrate injection tests * Migrate container provider tests * Migrate coroutine providers * Migrate traversal tests * Migrate resource tests * Migrate configuration tests * Migrate provided instance provider tests * Update doc blocks and imports * Migrate singleton tests * Update changelog and cosmetic fixes
148 lines
4.3 KiB
Python
148 lines
4.3 KiB
Python
"""Tests for container async resources."""
|
|
|
|
import asyncio
|
|
|
|
from dependency_injector import containers, providers
|
|
from pytest import mark, raises
|
|
|
|
|
|
@mark.asyncio
|
|
async def test_init_and_shutdown_ordering():
|
|
"""Test init and shutdown resources.
|
|
|
|
Methods .init_resources() and .shutdown_resources() should respect resources dependencies.
|
|
Initialization should first initialize resources without dependencies and then provide
|
|
these resources to other resources. Resources shutdown should follow the same rule: first
|
|
shutdown resources without initialized dependencies and then continue correspondingly
|
|
until all resources are shutdown.
|
|
"""
|
|
initialized_resources = []
|
|
shutdown_resources = []
|
|
|
|
async def _resource(name, delay, **_):
|
|
await asyncio.sleep(delay)
|
|
initialized_resources.append(name)
|
|
|
|
yield name
|
|
|
|
await asyncio.sleep(delay)
|
|
shutdown_resources.append(name)
|
|
|
|
class Container(containers.DeclarativeContainer):
|
|
resource1 = providers.Resource(
|
|
_resource,
|
|
name="r1",
|
|
delay=0.03,
|
|
)
|
|
resource2 = providers.Resource(
|
|
_resource,
|
|
name="r2",
|
|
delay=0.02,
|
|
r1=resource1,
|
|
)
|
|
resource3 = providers.Resource(
|
|
_resource,
|
|
name="r3",
|
|
delay=0.01,
|
|
r2=resource2,
|
|
)
|
|
|
|
container = Container()
|
|
|
|
await container.init_resources()
|
|
assert initialized_resources == ["r1", "r2", "r3"]
|
|
assert shutdown_resources == []
|
|
|
|
await container.shutdown_resources()
|
|
assert initialized_resources == ["r1", "r2", "r3"]
|
|
assert shutdown_resources == ["r3", "r2", "r1"]
|
|
|
|
await container.init_resources()
|
|
assert initialized_resources == ["r1", "r2", "r3", "r1", "r2", "r3"]
|
|
assert shutdown_resources == ["r3", "r2", "r1"]
|
|
|
|
await container.shutdown_resources()
|
|
assert initialized_resources == ["r1", "r2", "r3", "r1", "r2", "r3"]
|
|
assert shutdown_resources == ["r3", "r2", "r1", "r3", "r2", "r1"]
|
|
|
|
|
|
@mark.asyncio
|
|
async def test_shutdown_circular_dependencies_breaker():
|
|
async def _resource(name, **_):
|
|
yield name
|
|
|
|
class Container(containers.DeclarativeContainer):
|
|
resource1 = providers.Resource(
|
|
_resource,
|
|
name="r1",
|
|
)
|
|
resource2 = providers.Resource(
|
|
_resource,
|
|
name="r2",
|
|
r1=resource1,
|
|
)
|
|
resource3 = providers.Resource(
|
|
_resource,
|
|
name="r3",
|
|
r2=resource2,
|
|
)
|
|
|
|
container = Container()
|
|
await container.init_resources()
|
|
|
|
# Create circular dependency after initialization (r3 -> r2 -> r1 -> r3 -> ...)
|
|
container.resource1.add_kwargs(r3=container.resource3)
|
|
|
|
with raises(RuntimeError, match="Unable to resolve resources shutdown order"):
|
|
await container.shutdown_resources()
|
|
|
|
|
|
@mark.asyncio
|
|
async def test_shutdown_sync_and_async_ordering():
|
|
initialized_resources = []
|
|
shutdown_resources = []
|
|
|
|
def _sync_resource(name, **_):
|
|
initialized_resources.append(name)
|
|
yield name
|
|
shutdown_resources.append(name)
|
|
|
|
async def _async_resource(name, **_):
|
|
initialized_resources.append(name)
|
|
yield name
|
|
shutdown_resources.append(name)
|
|
|
|
class Container(containers.DeclarativeContainer):
|
|
resource1 = providers.Resource(
|
|
_sync_resource,
|
|
name="r1",
|
|
)
|
|
resource2 = providers.Resource(
|
|
_sync_resource,
|
|
name="r2",
|
|
r1=resource1,
|
|
)
|
|
resource3 = providers.Resource(
|
|
_async_resource,
|
|
name="r3",
|
|
r2=resource2,
|
|
)
|
|
|
|
container = Container()
|
|
|
|
await container.init_resources()
|
|
assert initialized_resources == ["r1", "r2", "r3"]
|
|
assert shutdown_resources == []
|
|
|
|
await container.shutdown_resources()
|
|
assert initialized_resources == ["r1", "r2", "r3"]
|
|
assert shutdown_resources == ["r3", "r2", "r1"]
|
|
|
|
await container.init_resources()
|
|
assert initialized_resources == ["r1", "r2", "r3", "r1", "r2", "r3"]
|
|
assert shutdown_resources == ["r3", "r2", "r1"]
|
|
|
|
await container.shutdown_resources()
|
|
assert initialized_resources == ["r1", "r2", "r3", "r1", "r2", "r3"]
|
|
assert shutdown_resources == ["r3", "r2", "r1", "r3", "r2", "r1"]
|