Coverage for src/tests/modules/training/teams/test_team_db_repository.py: 76%

1"""Module for testing the team database repository.""" 

2 

3import pytest 

4 

5from kwai_bc_training.teams.team import TeamIdentifier 

6from kwai_bc_training.teams.team_db_repository import TeamDbRepository 

7from kwai_core.db.database import Database 

8from kwai_core.db.exceptions import QueryException 

9 

10 

11async def test_get_by_ids(database: Database): 

12 """Test get_by_ids method.""" 

13 repo = TeamDbRepository(database) 

14 

15 try: 

16 { 

17 team.id: team 

18 async for team in repo.get_by_ids(TeamIdentifier(1), TeamIdentifier(2)) 

19 } 

20 except QueryException as qe: 

21 pytest.fail(str(qe)) 

22 

23 

24async def test_get_all(database: Database): 

25 """Test get_all.""" 

26 repo = TeamDbRepository(database) 

27 

28 try: 

29 repo.get_all() 

30 except QueryException as qe: 

31 pytest.fail(str(qe))