From 2dabfaaad3469296cf44da3ca56782ce3577a493 Mon Sep 17 00:00:00 2001
From: jazzthief <mynameisyegor@gmail.com>
Date: Tue, 23 May 2023 18:24:51 +0200
Subject: [PATCH] Add test container with element after resource

---
 .../wiringstringids/resourceclosing.py        | 36 +++++++++++++++++--
 1 file changed, 34 insertions(+), 2 deletions(-)

diff --git a/tests/unit/samples/wiringstringids/resourceclosing.py b/tests/unit/samples/wiringstringids/resourceclosing.py
index a5df21a8..e2817cf3 100644
--- a/tests/unit/samples/wiringstringids/resourceclosing.py
+++ b/tests/unit/samples/wiringstringids/resourceclosing.py
@@ -2,9 +2,14 @@ from dependency_injector import containers, providers
 from dependency_injector.wiring import inject, Provide, Closing
 
 
+class Singleton:
+    pass
+
+
 class Service:
     init_counter: int = 0
     shutdown_counter: int = 0
+    dependency: Singleton = None
 
     @classmethod
     def reset_counter(cls):
@@ -12,7 +17,9 @@ class Service:
         cls.shutdown_counter = 0
 
     @classmethod
-    def init(cls):
+    def init(cls, dependency: Singleton = None):
+        if dependency:
+            cls.dependency = dependency
         cls.init_counter += 1
 
     @classmethod
@@ -37,11 +44,36 @@ def init_service():
     service.shutdown()
 
 
+def init_service_with_singleton(singleton: Singleton):
+    service = Service()
+    service.init(singleton)
+    yield service
+    service.shutdown()
+
+
 class Container(containers.DeclarativeContainer):
 
     service = providers.Resource(init_service)
     factory_service = providers.Factory(FactoryService, service)
-    factory_service_kwargs = providers.Factory(FactoryService, service=service)
+    factory_service_kwargs = providers.Factory(
+        FactoryService,
+        service=service
+    )
+    nested_service = providers.Factory(NestedService, factory_service)
+
+
+class ContainerSingleton(containers.DeclarativeContainer):
+
+    singleton = providers.Resource(Singleton)
+    service = providers.Resource(
+        init_service_with_singleton,
+        singleton
+    )
+    factory_service = providers.Factory(FactoryService, service)
+    factory_service_kwargs = providers.Factory(
+        FactoryService,
+        service=service
+    )
     nested_service = providers.Factory(NestedService, factory_service)