Coverage for bc/kwai-bc-identity/src/kwai_bc_identity/tokens/refresh_token_query.py: 100%

1"""Module that defines an interface for querying a refresh token.""" 

2 

3from abc import abstractmethod 

4from typing import Self 

5 

6from kwai_core.domain.repository.query import Query 

7 

8from kwai_bc_identity.tokens.refresh_token import RefreshTokenIdentifier 

9from kwai_bc_identity.tokens.token_identifier import TokenIdentifier 

10from kwai_bc_identity.users.user_account import UserAccountEntity 

11 

12 

13class RefreshTokenQuery(Query): 

14 """An interface for querying a refresh token.""" 

15 

16 @abstractmethod 

17 def filter_by_id(self, id_: RefreshTokenIdentifier) -> "RefreshTokenQuery": 

18 """Filter for the given id.""" 

19 raise NotImplementedError 

20 

21 @abstractmethod 

22 def filter_by_token_identifier( 

23 self, identifier: TokenIdentifier 

24 ) -> "RefreshTokenQuery": 

25 """Filter for the given token identifier.""" 

26 raise NotImplementedError 

27 

28 @abstractmethod 

29 def filter_by_user_account(self, user_account: UserAccountEntity) -> Self: 

30 """Filter for the given user account.""" 

31 raise NotImplementedError