Coverage for src/kwai/modules/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 

7from kwai.modules.identity.tokens.refresh_token import RefreshTokenIdentifier 

8from kwai.modules.identity.tokens.token_identifier import TokenIdentifier 

9from kwai.modules.identity.users.user_account import UserAccountEntity 

10 

11 

12class RefreshTokenQuery(Query): 

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

14 

15 @abstractmethod 

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

17 """Filter for the given id.""" 

18 raise NotImplementedError 

19 

20 @abstractmethod 

21 def filter_by_token_identifier( 

22 self, identifier: TokenIdentifier 

23 ) -> "RefreshTokenQuery": 

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

25 raise NotImplementedError 

26 

27 @abstractmethod 

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

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

30 raise NotImplementedError