feat: auto committed

services/ai/tests/test_circuit_breaker.py

@@ -0,0 +1,111 @@
+"""熔断器测试."""
+
+from unittest.mock import patch
+
+from src.ai.providers.circuit_breaker import CircuitBreaker, CircuitState
+
+
+class TestCircuitBreaker:
+    """CircuitBreaker 状态机测试."""
+
+    def test_initial_state_closed(self) -> None:
+        """初始状态为 CLOSED."""
+        cb = CircuitBreaker()
+        assert cb.get_state("openai") == CircuitState.CLOSED
+
+    def test_record_success_resets_failures(self) -> None:
+        """成功重置失败计数."""
+        cb = CircuitBreaker()
+        cb.record_failure("openai")
+        cb.record_failure("openai")
+        cb.record_success("openai")
+        assert cb.get_state("openai") == CircuitState.CLOSED
+        status = cb.status()
+        assert status["openai"]["failures"] == 0
+
+    def test_threshold_opens_circuit(self) -> None:
+        """达阈值触发 OPEN."""
+        cb = CircuitBreaker(failure_threshold=3)
+        cb.record_failure("p1")
+        cb.record_failure("p1")
+        assert cb.get_state("p1") == CircuitState.CLOSED
+        cb.record_failure("p1")
+        assert cb.get_state("p1") == CircuitState.OPEN
+
+    def test_open_blocks_calls(self) -> None:
+        """OPEN 状态禁止调用."""
+        cb = CircuitBreaker(failure_threshold=1)
+        cb.record_failure("p1")
+        assert cb.is_closed("p1") is False
+
+    def test_closed_allows_calls(self) -> None:
+        """CLOSED 状态允许调用."""
+        cb = CircuitBreaker()
+        assert cb.is_closed("p1") is True
+
+    def test_half_open_after_cooldown(self) -> None:
+        """冷却后进入 HALF_OPEN."""
+        cb = CircuitBreaker(failure_threshold=1, cooldown_seconds=60.0)
+        with patch("src.ai.providers.circuit_breaker.time.monotonic", return_value=1000.0):
+            cb.record_failure("p1")
+            assert cb.get_state("p1") == CircuitState.OPEN
+        # 时间推进超过冷却期
+        with patch("src.ai.providers.circuit_breaker.time.monotonic", return_value=1061.0):
+            assert cb.get_state("p1") == CircuitState.HALF_OPEN
+
+    def test_half_open_allows_one_call(self) -> None:
+        """HALF_OPEN 允许 1 次试探."""
+        cb = CircuitBreaker(failure_threshold=1, cooldown_seconds=60.0, half_open_max_calls=1)
+        with patch("src.ai.providers.circuit_breaker.time.monotonic", return_value=1000.0):
+            cb.record_failure("p1")
+        # 推进时间触发 HALF_OPEN
+        with patch("src.ai.providers.circuit_breaker.time.monotonic", return_value=1061.0):
+            cb.get_state("p1")  # HALF_OPEN
+        assert cb.is_closed("p1") is True  # 第 1 次允许
+        assert cb.is_closed("p1") is False  # 第 2 次拒绝
+
+    def test_half_open_success_closes(self) -> None:
+        """HALF_OPEN 成功 → CLOSED."""
+        cb = CircuitBreaker(failure_threshold=1, cooldown_seconds=60.0)
+        with patch("src.ai.providers.circuit_breaker.time.monotonic", return_value=1000.0):
+            cb.record_failure("p1")
+        with patch("src.ai.providers.circuit_breaker.time.monotonic", return_value=1061.0):
+            cb.get_state("p1")  # HALF_OPEN
+        cb.record_success("p1")
+        assert cb.get_state("p1") == CircuitState.CLOSED
+
+    def test_half_open_failure_reopens(self) -> None:
+        """HALF_OPEN 失败 → 重新 OPEN."""
+        cb = CircuitBreaker(failure_threshold=1, cooldown_seconds=60.0)
+        with patch("src.ai.providers.circuit_breaker.time.monotonic", return_value=1000.0):
+            cb.record_failure("p1")
+        with patch("src.ai.providers.circuit_breaker.time.monotonic", return_value=1061.0):
+            cb.get_state("p1")  # HALF_OPEN
+        cb.record_failure("p1")
+        assert cb.get_state("p1") == CircuitState.OPEN
+
+    def test_reset_clears_state(self) -> None:
+        """reset 清除 Provider 状态."""
+        cb = CircuitBreaker(failure_threshold=1)
+        cb.record_failure("p1")
+        cb.reset("p1")
+        assert cb.get_state("p1") == CircuitState.CLOSED
+        assert "p1" not in cb.status()
+
+    def test_status_snapshot(self) -> None:
+        """status() 返回快照."""
+        cb = CircuitBreaker(failure_threshold=3)
+        cb.record_failure("p1")
+        cb.record_failure("p1")
+        cb.record_failure("p2")
+        status = cb.status()
+        assert "p1" in status
+        assert status["p1"]["failures"] == 2
+        assert "p2" in status
+
+    def test_different_providers_independent(self) -> None:
+        """不同 Provider 状态独立."""
+        cb = CircuitBreaker(failure_threshold=1)
+        cb.record_failure("p1")
+        assert cb.get_state("p1") == CircuitState.OPEN
+        assert cb.get_state("p2") == CircuitState.CLOSED