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

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

2 

3from kwai_bc_training.add_team_to_training import ( 

4 AddTeamToTraining, 

5 AddTeamToTrainingCommand, 

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_add_team_to_training(database, make_training_in_db, make_team_in_db): 

13 """Test the use case 'Add team to training'.""" 

14 training = await make_training_in_db() 

15 team = await make_team_in_db() 

16 

17 presenter = EntityPresenter() 

18 command = AddTeamToTrainingCommand( 

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

20 ) 

21 await AddTeamToTraining( 

22 TrainingDbRepository(database), 

23 TeamDbRepository(database), 

24 presenter, 

25 ).execute(command) 

26 

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

28 assert len(presenter.entity.teams) == 1, "There should be one team."