mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2024-11-27 12:04:00 +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
118 lines
2.9 KiB
Python
118 lines
2.9 KiB
Python
"""Singleton provider async mode tests."""
|
|
|
|
import asyncio
|
|
import random
|
|
|
|
from dependency_injector import providers
|
|
from pytest import mark, raises
|
|
|
|
from .common import RESOURCE1, RESOURCE2, BaseContainer, Client, Service
|
|
|
|
|
|
@mark.asyncio
|
|
async def test_injections():
|
|
class ContainerWithSingletons(BaseContainer):
|
|
client = providers.Singleton(
|
|
Client,
|
|
resource1=BaseContainer.resource1,
|
|
resource2=BaseContainer.resource2,
|
|
)
|
|
|
|
service = providers.Singleton(
|
|
Service,
|
|
client=client,
|
|
)
|
|
|
|
container = ContainerWithSingletons()
|
|
|
|
client1 = await container.client()
|
|
client2 = await container.client()
|
|
|
|
assert isinstance(client1, Client)
|
|
assert client1.resource1 is RESOURCE1
|
|
assert client1.resource2 is RESOURCE2
|
|
|
|
assert isinstance(client2, Client)
|
|
assert client2.resource1 is RESOURCE1
|
|
assert client2.resource2 is RESOURCE2
|
|
|
|
service1 = await container.service()
|
|
service2 = await container.service()
|
|
|
|
assert isinstance(service1, Service)
|
|
assert isinstance(service1.client, Client)
|
|
assert service1.client.resource1 is RESOURCE1
|
|
assert service1.client.resource2 is RESOURCE2
|
|
|
|
assert isinstance(service2, Service)
|
|
assert isinstance(service2.client, Client)
|
|
assert service2.client.resource1 is RESOURCE1
|
|
assert service2.client.resource2 is RESOURCE2
|
|
|
|
assert service1 is service2
|
|
assert service1.client is service2.client
|
|
assert service1.client is client1
|
|
|
|
assert service2.client is client2
|
|
assert client1 is client2
|
|
|
|
|
|
@mark.asyncio
|
|
async def test_async_mode():
|
|
instance = object()
|
|
|
|
async def create_instance():
|
|
return instance
|
|
|
|
provider = providers.Singleton(create_instance)
|
|
|
|
instance1 = await provider()
|
|
instance2 = await provider()
|
|
|
|
assert instance1 is instance2
|
|
assert instance1 is instance
|
|
assert instance2 is instance
|
|
|
|
|
|
@mark.asyncio
|
|
async def test_concurrent_init():
|
|
async def create_instance():
|
|
await asyncio.sleep(random.randint(1, 10) / 1000)
|
|
return object()
|
|
|
|
provider = providers.Singleton(create_instance)
|
|
|
|
future_instance1 = provider()
|
|
future_instance2 = provider()
|
|
|
|
instance1, instance2 = await asyncio.gather(future_instance1, future_instance2)
|
|
instance3 = await provider()
|
|
|
|
assert instance1 is instance2 is instance3
|
|
|
|
|
|
@mark.asyncio
|
|
async def test_async_init_with_error():
|
|
async def create_instance():
|
|
create_instance.counter += 1
|
|
raise RuntimeError()
|
|
|
|
create_instance.counter = 0
|
|
|
|
provider = providers.Singleton(create_instance)
|
|
|
|
future = provider()
|
|
assert provider.is_async_mode_enabled() is True
|
|
|
|
with raises(RuntimeError):
|
|
await future
|
|
|
|
assert create_instance.counter == 1
|
|
assert provider.is_async_mode_enabled() is True
|
|
|
|
with raises(RuntimeError):
|
|
await provider()
|
|
|
|
assert create_instance.counter == 2
|
|
assert provider.is_async_mode_enabled() is True
|