selfprivacy-rest-api/tests/test_models.py

import pytest
from datetime import datetime, timedelta

from selfprivacy_api.models.tokens.recovery_key import RecoveryKey
from selfprivacy_api.models.tokens.new_device_key import NewDeviceKey


def test_recovery_key_expired():
    expiration = datetime.now() - timedelta(minutes=5)
    key = RecoveryKey.generate(expiration=expiration, uses_left=2)
    assert not key.is_valid()


def test_new_device_key_expired():
    expiration = datetime.now() - timedelta(minutes=5)
    key = NewDeviceKey.generate()
    key.expires_at = expiration
    assert not key.is_valid()