"""登出黑名单单元测试。 守的是:黑名单的 **TTL 必须等于令牌剩余有效期**。 写死固定值会出现「登出 N 分钟后令牌又活了」; 完全不设过期则 Redis 会随登出次数无限增长。两者都是真实事故。 """ from app.core.constants import REDIS_KEY_TOKEN_BLACKLIST from app.services import token_blacklist def _ttl(jti: str) -> int: from app.core.redis_client import get_redis return get_redis().ttl(REDIS_KEY_TOKEN_BLACKLIST.format(jti=jti)) class TestRevoke: """拉黑。""" def test_拉黑后可检出(self, fake_redis) -> None: token_blacklist.revoke("jti-1", 600) assert token_blacklist.is_revoked("jti-1") is True def test_TTL等于传入的剩余秒数(self, fake_redis) -> None: """容忍 1 秒偏差:Redis 的 TTL 以毫秒精度存绝对过期时刻, 命令返回时向下取整,刚写入就查询必然是 N 或 N-1。 断言死等会变成偶发失败的"闪烁测试"。""" token_blacklist.revoke("jti-1", 1234) assert _ttl("jti-1") in (1234, 1233) def test_键带过期时间而非永久存在(self, fake_redis) -> None: """永久存储会让 Redis 随登出次数无限增长。""" token_blacklist.revoke("jti-1", 600) assert _ttl("jti-1") > 0 def test_已过期令牌不写入(self, fake_redis) -> None: """剩余有效期为 0 或负数时无凭证可撤销,写入只会浪费内存。""" token_blacklist.revoke("jti-expired", 0) token_blacklist.revoke("jti-expired2", -100) assert token_blacklist.is_revoked("jti-expired") is False assert token_blacklist.is_revoked("jti-expired2") is False def test_重复拉黑同一jti不报错且TTL更新(self, fake_redis) -> None: token_blacklist.revoke("jti-1", 600) token_blacklist.revoke("jti-1", 300) assert _ttl("jti-1") in (300, 299) class TestIsRevoked: """检测。""" def test_未拉黑的令牌返回False(self, fake_redis) -> None: assert token_blacklist.is_revoked("never-revoked") is False def test_不同jti之间互不影响(self, fake_redis) -> None: """jti 是唯一键,串号会导致误踢其他用户的登录态。""" token_blacklist.revoke("jti-1", 600) assert token_blacklist.is_revoked("jti-1") is True assert token_blacklist.is_revoked("jti-2") is False