Coverage for apps/outers/containers/use_cases/document_processor_container.py: 100%

1from dependency_injector import providers 

2from dependency_injector.containers import DeclarativeContainer 

3 

4from apps.inners.use_cases.document_processor.category_document_processor import CategoryDocumentProcessor 

5from apps.inners.use_cases.document_processor.partition_document_processor import PartitionDocumentProcessor 

6from apps.inners.use_cases.document_processor.summary_document_processor import SummaryDocumentProcessor 

7 

8 

9class DocumentProcessorContainer(DeclarativeContainer): 

10 managements = providers.DependenciesContainer() 

11 

12 partition = providers.Singleton( 

13 PartitionDocumentProcessor, 

14 document_management=managements.document, 

15 file_document_management=managements.file_document, 

16 text_document_management=managements.text_document, 

17 web_document_management=managements.web_document 

18 ) 

19 summary = providers.Singleton( 

20 SummaryDocumentProcessor, 

21 ) 

22 category = providers.Singleton( 

23 CategoryDocumentProcessor, 

24 summary_document_processor=summary 

25 )