Coverage for src/tests/modules/training/test_delete_coach_to_training.py: 100%

1"""Module for testing the use case 'Add team to training'.""" 

2 

3from kwai_bc_training.delete_coach_from_training import ( 

4 DeleteCoachFromTraining, 

5 DeleteCoachFromTrainingCommand, 

6) 

7from kwai_bc_training.trainings.training_db_repository import TrainingDbRepository 

8from kwai_core.domain.presenter import EntityPresenter 

9 

10 

11async def test_delete_coach_from_training(database, make_training_in_db): 

12 """Test the use case 'Delete coach from training'.""" 

13 training = await make_training_in_db() 

14 

15 presenter = EntityPresenter() 

16 command = DeleteCoachFromTrainingCommand( 

17 training_id=training.id.value, 

18 coach_id=list(training.coaches)[0].coach.id.value, 

19 ) 

20 await DeleteCoachFromTraining( 

21 TrainingDbRepository(database), 

22 presenter, 

23 ).execute(command) 

24 

25 assert presenter.entity is not None, "There should be an entity present." 

26 assert len(presenter.entity.coaches) == 0, "There should be no coaches."