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

1"""Module for testing the use case 'Delete team from training'.""" 

2 

3from kwai_bc_training.delete_team_from_training import ( 

4 DeleteTeamFromTraining, 

5 DeleteTeamFromTrainingCommand, 

6) 

7from kwai_bc_training.teams.team_db_repository import TeamDbRepository 

8from kwai_bc_training.trainings.training_db_repository import TrainingDbRepository 

9from kwai_core.domain.presenter import EntityPresenter 

10 

11 

12async def test_delete_team_from_training( 

13 database, make_training_in_db, make_team_in_db 

14): 

15 """Test the use case 'Delete team from training'.""" 

16 training = await make_training_in_db() 

17 team = await make_team_in_db() 

18 

19 presenter = EntityPresenter() 

20 command = DeleteTeamFromTrainingCommand( 

21 training_id=training.id.value, team_id=team.id.value 

22 ) 

23 await DeleteTeamFromTraining( 

24 TrainingDbRepository(database), 

25 TeamDbRepository(database), 

26 presenter, 

27 ).execute(command) 

28 

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

30 assert len(presenter.entity.teams) == 0, "There should be 0 teams."