Initial public release

tests/test_circuit_breaker.py

@@ -0,0 +1,26 @@
+import pytest
+from pybreaker import CircuitBreakerError
+from hydra.resilience.circuit_breaker import create_breaker
+
+
+def failing_fn():
+    raise ConnectionError("exchange down")
+
+
+def test_breaker_open_after_failures():
+    breaker = create_breaker("test", fail_max=3, reset_timeout=60)
+    for _ in range(2):
+        with pytest.raises(ConnectionError):
+            breaker.call(failing_fn)
+    # 3번째 실패에서 circuit이 open된다
+    with pytest.raises(CircuitBreakerError):
+        breaker.call(failing_fn)
+    # circuit이 open되었으므로 이후 호출도 CircuitBreakerError
+    with pytest.raises(CircuitBreakerError):
+        breaker.call(failing_fn)
+
+
+def test_breaker_passes_success():
+    breaker = create_breaker("test2", fail_max=5, reset_timeout=60)
+    result = breaker.call(lambda: "ok")
+    assert result == "ok"