Coverage for src/tests/modules/identity/tokens/conftest.py: 100%

1"""Module that defines fixtures for testing identity tokens.""" 

2 

3import pytest 

4 

5from kwai_bc_identity.tokens.access_token_db_repository import ( 

6 AccessTokenDbRepository, 

7) 

8from kwai_bc_identity.tokens.access_token_repository import AccessTokenRepository 

9from kwai_bc_identity.tokens.refresh_token_db_repository import ( 

10 RefreshTokenDbRepository, 

11) 

12from kwai_bc_identity.tokens.refresh_token_repository import RefreshTokenRepository 

13from kwai_core.db.database import Database 

14 

15 

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

17def access_token_repo(database: Database) -> AccessTokenRepository: 

18 """Fixture for creating an access token repository.""" 

19 return AccessTokenDbRepository(database) 

20 

21 

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

23def refresh_token_repo(database: Database) -> RefreshTokenRepository: 

24 """Fixture for creating the refresh token repository.""" 

25 return RefreshTokenDbRepository(database)