Coverage for src/tests/modules/identity/tokens/test_user_token_db_repository.py: 82%

1"""Module for testing the user token repository for a database.""" 

2 

3import pytest 

4 

5from kwai_bc_identity.tokens.user_token_db_repository import UserTokenDbRepository 

6from kwai_core.db.database import Database 

7 

8 

9pytestmark = pytest.mark.db 

10 

11 

12async def test_revoke(database: Database, make_user_account_in_db): 

13 """Test revoking access and refresh tokens of a user account.""" 

14 user_account = await make_user_account_in_db() 

15 

16 repo = UserTokenDbRepository(database) 

17 try: 

18 await repo.revoke(user_account) 

19 except Exception as exc: 

20 pytest.fail(f"Revoke failed with exception: {exc}")