Coverage for src/tests/modules/identity/tokens/test_access_token_db_query.py: 78%

1"""Module for testing the access token database query.""" 

2 

3import pytest 

4 

5from kwai_bc_identity.tokens.access_token import AccessTokenIdentifier 

6from kwai_bc_identity.tokens.access_token_db_query import AccessTokenDbQuery 

7from kwai_bc_identity.tokens.access_token_query import AccessTokenQuery 

8from kwai_bc_identity.tokens.token_identifier import TokenIdentifier 

9from kwai_core.db.database import Database 

10 

11 

12@pytest.fixture 

13def query(database: Database) -> AccessTokenQuery: 

14 """A fixture for a user account query.""" 

15 return AccessTokenDbQuery(database) 

16 

17 

18async def test_filter_by_id(query: AccessTokenQuery): 

19 """Test filtering by id.""" 

20 query.filter_by_id(AccessTokenIdentifier(1)) 

21 

22 try: 

23 await query.fetch_one() 

24 except Exception as exc: 

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

26 

27 

28async def test_filter_by_token_identifier(query: AccessTokenQuery): 

29 """Test filtering by token identifier.""" 

30 query.filter_by_token_identifier(TokenIdentifier.generate()) 

31 

32 try: 

33 await query.fetch_one() 

34 except Exception as exc: 

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

36 

37 

38async def test_filter_by_user_account(query: AccessTokenQuery, make_user_account): 

39 """Test filtering by user account.""" 

40 query.filter_by_user_account(make_user_account()) 

41 

42 try: 

43 await query.fetch_one() 

44 except Exception as exc: 

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