mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2024-11-26 19:43:58 +03:00
45 lines
964 B
Python
45 lines
964 B
Python
from dependency_injector import containers, providers
|
|
from unittest import mock
|
|
|
|
|
|
class ApiClient:
|
|
|
|
def __init__(self, api_key: str, timeout: int):
|
|
self.api_key = api_key
|
|
self.timeout = timeout
|
|
|
|
|
|
class Service:
|
|
|
|
def __init__(self, api_client: ApiClient):
|
|
self.api_client = api_client
|
|
|
|
|
|
class Container(containers.DeclarativeContainer):
|
|
|
|
config = providers.Configuration()
|
|
|
|
api_client = providers.Singleton(
|
|
ApiClient,
|
|
api_key=config.api_key,
|
|
timeout=config.timeout,
|
|
)
|
|
|
|
service = providers.Factory(
|
|
Service,
|
|
api_client=api_client,
|
|
)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
container = Container()
|
|
container.config.from_yaml('config.yml')
|
|
|
|
service = container.service()
|
|
assert isinstance(service.api_client, ApiClient)
|
|
|
|
with container.api_client.override(mock.Mock()):
|
|
service = container.service()
|
|
assert isinstance(service.api_client, mock.Mock)
|
|
|