Coverage for src/tests/modules/identity/user_recoveries/test_user_recovery.py: 100%

1"""Module for testing the user recovery entity.""" 

2 

3from kwai_bc_identity.user_recoveries.user_recovery import UserRecoveryEntity 

4from kwai_bc_identity.users.user import UserEntity 

5from kwai_core.domain.value_objects.email_address import EmailAddress 

6from kwai_core.domain.value_objects.name import Name 

7from kwai_core.domain.value_objects.timestamp import Timestamp 

8 

9 

10def test_expired_user_recovery(): 

11 """Test expire of a user recovery.""" 

12 user_recovery = UserRecoveryEntity( 

13 user=UserEntity( 

14 email=EmailAddress("jigoro.kano@kwai.com"), 

15 name=Name(first_name="Jigoro", last_name="Kano"), 

16 ), 

17 expiration=Timestamp.create_with_delta(hours=-2), 

18 ) 

19 

20 assert user_recovery.is_expired 

21 

22 

23def test_confirm_user_recovery(): 

24 """Test confirming a user recovery.""" 

25 user_recovery = UserRecoveryEntity( 

26 user=UserEntity( 

27 email=EmailAddress("jigoro.kano@kwai.com"), 

28 name=Name(first_name="Jigoro", last_name="Kano"), 

29 ) 

30 ) 

31 user_recovery = user_recovery.confirm() 

32 

33 assert user_recovery.confirmed is True, "The user recovery should be confirmed."