Coverage for src/tests/modules/training/trainings/test_training_team_db_query.py: 95%

1"""Module for testing TrainingTeamDbQuery.""" 

2 

3import pytest 

4 

5from kwai_bc_training.trainings.training import TrainingIdentifier 

6from kwai_bc_training.trainings.training_team_db_query import TrainingTeamDbQuery 

7from kwai_core.db.database import Database 

8 

9 

10pytestmark = pytest.mark.db 

11 

12 

13async def test_filter_by_training(database: Database): 

14 """Test filtering on training(s).""" 

15 query = TrainingTeamDbQuery(database) 

16 query.filter_by_trainings(TrainingIdentifier(10000)) 

17 

18 result = query.fetch() 

19 try: 

20 await anext(result) 

21 except StopAsyncIteration: 

22 ok = True 

23 else: 

24 ok = True 

25 assert ok is True, "Query is not executed" 

26 

27 

28async def test_fetch_teams(database: Database): 

29 """Test filtering on training(s).""" 

30 query = TrainingTeamDbQuery(database) 

31 query.filter_by_trainings(TrainingIdentifier(1), TrainingIdentifier(2)) 

32 

33 result = await query.fetch_teams() 

34 assert result is not None, "There should be a result"