Initial public release

tests/test_graceful_shutdown.py

@@ -0,0 +1,19 @@
+import asyncio
+import pytest
+from unittest.mock import AsyncMock, MagicMock, patch
+from hydra.resilience.graceful import GracefulManager
+
+
+@pytest.mark.asyncio
+async def test_shutdown_saves_state():
+    order_queue = MagicMock()
+    order_queue.block_new_orders = MagicMock()
+    position_tracker = AsyncMock()
+    position_tracker.snapshot.return_value = {"positions": []}
+    redis_client = MagicMock()
+
+    manager = GracefulManager(order_queue, position_tracker, redis_client)
+    await manager.shutdown("SIGTERM")
+
+    order_queue.block_new_orders.assert_called_once()
+    position_tracker.snapshot.assert_called_once()