Coverage for src/tests/modules/training/trainings/test_training_schedule_db_query.py: 84%

1"""Module for testing TrainingSchemaDbQuery.""" 

2 

3import pytest 

4 

5from kwai_bc_training.trainings.training_schedule import TrainingScheduleIdentifier 

6from kwai_bc_training.trainings.training_schedule_db_query import ( 

7 TrainingScheduleDbQuery, 

8) 

9from kwai_core.db.database import Database 

10 

11 

12pytestmark = pytest.mark.db 

13 

14 

15async def test_filter_by_id(database: Database): 

16 """Test filtering on id.""" 

17 query = TrainingScheduleDbQuery(database) 

18 query.filter_by_id(TrainingScheduleIdentifier(1)) 

19 

20 result = query.fetch() 

21 try: 

22 await anext(result) 

23 except StopAsyncIteration: 

24 ok = True 

25 else: 

26 ok = True 

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

28 

29 

30async def test_filter_by_ids(database: Database): 

31 """Test filtering on ids.""" 

32 query = TrainingScheduleDbQuery(database) 

33 query.filter_by_ids(TrainingScheduleIdentifier(1), TrainingScheduleIdentifier(2)) 

34 

35 result = query.fetch() 

36 try: 

37 await anext(result) 

38 except StopAsyncIteration: 

39 ok = True 

40 else: 

41 ok = True 

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