ets-labs/python-dependency-injector
1
1
Fork 0
mirror of https://github.com/ets-labs/python-dependency-injector.git synced 2025-01-31 03:36:41 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add test container with element after resource

Browse Source
This commit is contained in:
jazzthief 2023-05-23 18:24:51 +02:00
parent 027ccfa161
commit 2dabfaaad3
No known key found for this signature in database
GPG Key ID: 650DC0A0E6B6947C
1 changed files with 34 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
tests/unit/samples/wiringstringids/resourceclosing.py
View File

@ -2,9 +2,14 @@ from dependency_injector import containers, providers
from dependency_injector.wiring import inject, Provide, Closing from dependency_injector.wiring import inject, Provide, Closing
class Singleton:
pass
class Service: class Service:
init_counter: int = 0 init_counter: int = 0
shutdown_counter: int = 0 shutdown_counter: int = 0
dependency: Singleton = None
@classmethod @classmethod
def reset_counter(cls): def reset_counter(cls):
@ -12,7 +17,9 @@ class Service:
cls.shutdown_counter = 0 cls.shutdown_counter = 0
@classmethod @classmethod
def init(cls): def init(cls, dependency: Singleton = None):
if dependency:
cls.dependency = dependency
cls.init_counter += 1 cls.init_counter += 1
@classmethod @classmethod
@ -37,11 +44,36 @@ def init_service():
service.shutdown() service.shutdown()
def init_service_with_singleton(singleton: Singleton):
service = Service()
service.init(singleton)
yield service
service.shutdown()
class Container(containers.DeclarativeContainer): class Container(containers.DeclarativeContainer):
service = providers.Resource(init_service) service = providers.Resource(init_service)
factory_service = providers.Factory(FactoryService, 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) nested_service = providers.Factory(NestedService, factory_service)