Coverage for src/app/controllers/mark_done_support_message_contoller.py: 58%

1from dto.support_message_dto import SupportMessageDTO 

2from models.support_message_model import SupportMessageModel 

3from usecases.mark_done_support_messages_useCase import MarkDoneSupportMessagesUseCase 

4 

5 

6class MarkDoneSupportMessagesController: 

7 """ 

8 Controller for retrieving support messages from the database. 

9 """ 

10 

11 def __init__(self, mark_done_support_messages_useCase: MarkDoneSupportMessagesUseCase): 

12 self.mark_done_support_messages_useCase = mark_done_support_messages_useCase 

13 

14 def mark_done_support_messages(self, support_message_dto: SupportMessageDTO)-> int: 

15 try: 

16 

17 support_message_model = SupportMessageModel( 

18 id = support_message_dto.get_id(), 

19 status= support_message_dto.get_status() 

20 ) 

21 

22 return self.mark_done_support_messages_useCase.mark_done_support_messages(support_message_model) 

23 

24 except Exception as e: 

25 raise e