Initial public release

tests/test_kill_switch.py

@@ -0,0 +1,61 @@
+import pytest
+from unittest.mock import AsyncMock, MagicMock, patch, call
+from hydra.core.kill_switch import KillSwitch, KillSwitchResult
+
+KILL_BLOCKED_KEY = "hydra:kill_switch_active"
+
+
+@pytest.fixture
+def ks(mock_redis, mock_exchange):
+    telegram = AsyncMock()
+    telegram.send_message = AsyncMock()
+    positions = MagicMock()
+    positions.get_all.return_value = [
+        {"market": "kr", "symbol": "005930", "qty": 10, "side": "buy"}
+    ]
+    return KillSwitch(
+        exchanges={"kr": mock_exchange},
+        position_tracker=positions,
+        telegram=telegram,
+        redis_client=mock_redis,
+    )
+
+
+@pytest.mark.asyncio
+async def test_kill_via_cli(ks, mock_exchange):
+    result = await ks.execute(reason="test", source="cli")
+    assert result.success
+    mock_exchange.cancel_all.assert_called_once()
+
+
+@pytest.mark.asyncio
+async def test_kill_via_api(ks, mock_exchange):
+    result = await ks.execute(reason="manual", source="api")
+    assert result.success
+
+
+@pytest.mark.asyncio
+async def test_kill_blocks_new_orders(ks, mock_redis):
+    await ks.execute(reason="test", source="cli")
+    mock_redis.set.assert_any_call(KILL_BLOCKED_KEY, "1")
+
+
+@pytest.mark.asyncio
+async def test_kill_sends_telegram(ks):
+    await ks.execute(reason="test", source="cli")
+    ks._telegram.send_message.assert_called_once()
+
+
+@pytest.mark.asyncio
+async def test_auto_trigger_daily_loss(ks, mock_redis):
+    mock_redis.get.return_value = "-0.06"
+    triggered, reason = await ks.check_auto_triggers()
+    assert triggered
+    assert "daily_loss" in reason
+
+
+@pytest.mark.asyncio
+async def test_auto_trigger_no_trigger(ks, mock_redis):
+    mock_redis.get.return_value = "-0.01"
+    triggered, _ = await ks.check_auto_triggers()
+    assert not triggered