Coverage for src/tests/modules/portal/pages/test_page_db_query.py: 100%

1"""Module for testing page database query.""" 

2 

3import pytest 

4 

5from kwai_bc_portal.pages.page_db_query import PageDbQuery 

6from kwai_bc_portal.pages.page_query import PageQuery 

7from kwai_core.db.database import Database 

8 

9 

10@pytest.fixture 

11def page_query(database: Database) -> PageQuery: 

12 """Fixture for a page query.""" 

13 return PageDbQuery(database) 

14 

15 

16async def test_page_db_query(page_query: PageQuery): 

17 """Test the query.""" 

18 count = await page_query.count() 

19 assert count >= 0, "There should be 0 or more pages." 

20 

21 

22async def test_filter_by_active(page_query: PageQuery): 

23 """Test the query with filtering on active pages.""" 

24 query = page_query.filter_by_active() 

25 count = await query.count() 

26 assert count >= 0, "There should be 0 or more pages." 

27 

28 

29async def test_filter_by_user(page_query: PageQuery): 

30 """Test the query with filtering on author.""" 

31 page_query.filter_by_user(1) 

32 count = await page_query.count() 

33 assert count >= 0, "There should be 0 or more stories."