feat: auto committed

services/ai/tests/test_rate_limiter.py

@@ -0,0 +1,59 @@
+"""限流器测试."""
+
+
+from src.ai.rate_limiter import RateLimiter, RateLimitResult
+
+
+class TestRateLimiterDegraded:
+    """Redis 不可用时的降级行为（全并行模式）."""
+
+    async def test_no_redis_allows_all(self) -> None:
+        """Redis=None 时降级放行."""
+        limiter = RateLimiter(redis=None)
+        results = await limiter.check(user_id="u-1", ip="1.2.3.4", school_id="s-1")
+        assert len(results) == 3
+        assert all(r.allowed for r in results)
+
+    async def test_no_redis_no_identifiers(self) -> None:
+        """无任何标识符时返回空列表."""
+        limiter = RateLimiter(redis=None)
+        results = await limiter.check()
+        assert results == []
+
+    async def test_no_redis_only_user(self) -> None:
+        """仅 user_id."""
+        limiter = RateLimiter(redis=None)
+        results = await limiter.check(user_id="u-1")
+        assert len(results) == 1
+        assert results[0].dimension == "user"
+        assert results[0].allowed is True
+        assert results[0].remaining == 10  # default user_limit
+
+    async def test_no_redis_only_ip(self) -> None:
+        """仅 ip."""
+        limiter = RateLimiter(redis=None)
+        results = await limiter.check(ip="1.2.3.4")
+        assert len(results) == 1
+        assert results[0].dimension == "ip"
+        assert results[0].limit == 30
+
+    async def test_no_redis_only_school(self) -> None:
+        """仅 school_id."""
+        limiter = RateLimiter(redis=None)
+        results = await limiter.check(school_id="s-1")
+        assert len(results) == 1
+        assert results[0].dimension == "school"
+        assert results[0].limit == 100
+
+
+class TestRateLimitResult:
+    """RateLimitResult dataclass 测试."""
+
+    def test_default_values(self) -> None:
+        result = RateLimitResult(allowed=True, dimension="user", limit=10, remaining=5)
+        assert result.allowed is True
+        assert result.remaining == 5
+
+    def test_denied(self) -> None:
+        result = RateLimitResult(allowed=False, dimension="ip", limit=30, remaining=0)
+        assert result.allowed is False