Coverage for src/tests/modules/portal/repositories/test_author_db_query.py: 82%

1"""Module for testing AuthorDbQuery.""" 

2 

3import pytest 

4 

5from kwai_bc_portal.domain.author import AuthorIdentifier 

6from kwai_bc_portal.repositories.author_db_query import AuthorDbQuery 

7from kwai_bc_portal.repositories.author_query import AuthorQuery 

8from kwai_core.db.database import Database 

9from kwai_core.domain.value_objects.unique_id import UniqueId 

10 

11 

12pytestmark = [pytest.mark.db] 

13 

14 

15@pytest.fixture(scope="module") 

16def query(database: Database) -> AuthorQuery: 

17 """Fixture for creating the author query.""" 

18 return AuthorDbQuery(database) 

19 

20 

21async def test_filter_by_id(query: AuthorQuery): 

22 """Test filter by id.""" 

23 query.filter_by_id(AuthorIdentifier(1)) 

24 try: 

25 await query.fetch_one() 

26 except Exception as exc: 

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

28 

29 

30async def test_filter_by_uuid(query: AuthorQuery): 

31 """Test filter by uuid.""" 

32 query.filter_by_uuid(UniqueId.generate()) 

33 try: 

34 await query.fetch_one() 

35 except Exception as exc: 

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