ets-labs/python-dependency-injector
1
1
Fork 0
mirror of https://github.com/ets-labs/python-dependency-injector.git synced 2024-11-29 04:53:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update provider overriding example to use container and fix bug

Browse Source
This commit is contained in:
Roman Mogylatov 2020-09-03 17:42:15 -04:00
parent d61281a0b9
commit 4f111aae9b
1 changed files with 21 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
examples/providers/overriding.py
View File

@ -1,8 +1,9 @@
"""Simple providers overriding example."""
import dataclasses
import unittest.mock
from dependency_injector import providers
from dependency_injector import containers, providers
class ApiClient:
@ -13,30 +14,35 @@ class ApiClientStub(ApiClient):
...
@dataclasses.dataclass
class Service:
def __init__(self, api_client: ApiClient):
self._api_client = api_client
api_client: ApiClient
api_client_factory = providers.Factory(ApiClient)
service_factory = providers.Factory(
Service,
api_client=api_client_factory,
)
class Container(containers.DeclarativeContainer):
api_client_factory = providers.Factory(ApiClient)
service_factory = providers.Factory(
Service,
api_client=api_client_factory,
)
if __name__ == '__main__':
container = Container()
# 1. Use .override() to replace the API client with stub
api_client_factory.override(providers.Factory(ApiClientStub))
service1 = service_factory()
container.api_client_factory.override(providers.Factory(ApiClientStub))
service1 = container.service_factory()
assert isinstance(service1.api_client, ApiClientStub)
# 2. Use .override() as a context manager to mock the API client in testing
with api_client_factory.override(unittest.mock.Mock(ApiClient)):
service3 = service_factory()
assert isinstance(service3.api_client, unittest.mock.Mock)
with container.api_client_factory.override(unittest.mock.Mock(ApiClient)):
service2 = container.service_factory()
assert isinstance(service2.api_client, unittest.mock.Mock)
# 3. Use .reset_override() to get back to normal
api_client_factory.reset_override()
service3 = service_factory()
container.api_client_factory.reset_override()
service3 = container.service_factory()
assert isinstance(service3.api_client, ApiClient)