Coverage for src/tests/modules/teams/repositories/test_team_member_db_query.py: 85%

1"""Module for testing the TeamMemberDbQuery class.""" 

2 

3import pytest 

4 

5from kwai_bc_teams.domain.team import TeamIdentifier 

6from kwai_bc_teams.repositories.team_member_db_query import TeamMemberDbQuery 

7from kwai_core.db.database import Database 

8 

9 

10@pytest.fixture 

11def query(database: Database) -> TeamMemberDbQuery: 

12 """A fixture for a team member database query.""" 

13 return TeamMemberDbQuery(database) 

14 

15 

16async def test_filter_by_teams(query: TeamMemberDbQuery): 

17 """Test filter by teams.""" 

18 query.filter_by_teams(TeamIdentifier(1)) 

19 try: 

20 await query.fetch_team_members() 

21 except Exception as exc: 

22 pytest.fail(f"An exception occurred: f{exc}")