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

1"""Module for defining tests for the use case 'Get Teams'.""" 

2 

3import pytest 

4 

5from kwai_bc_teams.domain.team import TeamEntity 

6from kwai_bc_teams.get_teams import GetTeams, GetTeamsCommand 

7from kwai_bc_teams.repositories.team_db_repository import TeamDbRepository 

8from kwai_core.db.database import Database 

9from kwai_core.domain.presenter import CountIterableAsyncPresenter 

10 

11 

12pytestmark = pytest.mark.db 

13 

14 

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

16 """Test get teams.""" 

17 await make_team_in_db() 

18 command = GetTeamsCommand() 

19 presenter = CountIterableAsyncPresenter[TeamEntity]() 

20 await GetTeams(TeamDbRepository(database), presenter).execute(command) 

21 assert presenter.count > 0, "There should be at least one team."