From 684745a6b47fa485effba809a0db7096cca807df Mon Sep 17 00:00:00 2001 From: Roman Mogylatov Date: Fri, 14 Aug 2020 15:31:33 -0400 Subject: [PATCH] Add providers overriding to demo 2 --- examples/di_demo2/demo.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/examples/di_demo2/demo.py b/examples/di_demo2/demo.py index ccfccd97..886b0de1 100644 --- a/examples/di_demo2/demo.py +++ b/examples/di_demo2/demo.py @@ -1,4 +1,5 @@ from dependency_injector import containers, providers +from unittest import mock class ApiClient: @@ -35,5 +36,9 @@ if __name__ == '__main__': 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) +