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

1"""Module for testing the access token entity.""" 

2 

3from kwai_bc_identity.tokens.access_token import AccessTokenEntity 

4from kwai_bc_identity.tokens.token_identifier import TokenIdentifier 

5from kwai_core.domain.value_objects.timestamp import Timestamp 

6 

7 

8def test_create(make_user_account) -> None: 

9 """Test creation of an access token.""" 

10 token = AccessTokenEntity( 

11 identifier=TokenIdentifier.generate(), 

12 expiration=Timestamp.create_now(), 

13 user_account=make_user_account(), 

14 ) 

15 

16 assert token.revoked is False, "A new token should not be revoked." 

17 

18 

19def test_expired(make_user_account) -> None: 

20 """Test if an access token is expired.""" 

21 token = AccessTokenEntity( 

22 identifier=TokenIdentifier.generate(), 

23 expiration=Timestamp.create_now(), 

24 user_account=make_user_account(), 

25 ) 

26 

27 assert token.expired is True, "The token should be expired." 

28 

29 

30def test_not_expired(make_user_account) -> None: 

31 """Test if an access token is not expired.""" 

32 token = AccessTokenEntity( 

33 identifier=TokenIdentifier.generate(), 

34 expiration=Timestamp.create_now().add_delta(days=1), 

35 user_account=make_user_account(), 

36 ) 

37 

38 assert token.expired is False, "The token should not be expired."