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

1"""Module for tests of the get teams use case.""" 

2 

3from kwai_bc_training.get_teams import GetTeams 

4from kwai_bc_training.teams.team_db_repository import TeamDbRepository 

5from kwai_core.db.database import Database 

6 

7 

8async def test_get_teams(database: Database, make_team_in_db): 

9 """Test use case get teams.""" 

10 team = await make_team_in_db() 

11 count, iterator = await GetTeams(TeamDbRepository(database)).execute() 

12 assert count > 0, "There should be a team" 

13 teams = {team.id: team async for team in iterator} 

14 assert team.id in teams, "The team should be in the result"