From c20c57ae7cb3d2061d3ea8be6b9f36936e094786 Mon Sep 17 00:00:00 2001
From: Roman Mogylatov <rmogilatov@gmail.com>
Date: Sat, 26 Sep 2020 00:23:11 -0400
Subject: [PATCH] Update demo

---
 examples/demo/demo.py |  4 ++++
 examples/demo/test.py | 11 -----------
 2 files changed, 4 insertions(+), 11 deletions(-)
 delete mode 100644 examples/demo/test.py

diff --git a/examples/demo/demo.py b/examples/demo/demo.py
index 1edb469c..12eccb97 100644
--- a/examples/demo/demo.py
+++ b/examples/demo/demo.py
@@ -1,4 +1,5 @@
 import sys
+from unittest import mock
 
 from dependency_injector import containers, providers
 from dependency_injector.wiring import Provide
@@ -35,3 +36,6 @@ if __name__ == '__main__':
     container.wire(modules=[sys.modules[__name__]])
 
     main()
+
+    with container.api_client.override(mock.Mock()):
+        main()
diff --git a/examples/demo/test.py b/examples/demo/test.py
deleted file mode 100644
index ce5e0b6c..00000000
--- a/examples/demo/test.py
+++ /dev/null
@@ -1,11 +0,0 @@
-from unittest import mock
-
-import demo
-
-
-if __name__ == '__main__':
-    container = demo.Container()
-    container.wire(modules=[demo])
-
-    with container.api_client.override(mock.Mock()):
-        demo.main()