From 21c6b777530b4a027aec9c12bf63092e5a7c006d Mon Sep 17 00:00:00 2001
From: TheSiahxyz <164138827+TheSiahxyz@users.noreply.github.com>
Date: Wed, 1 Apr 2026 17:24:57 +0900
Subject: feat: add multi-exchange support via ccxt factory

---
 shared/tests/test_exchange.py | 55 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 shared/tests/test_exchange.py

(limited to 'shared/tests/test_exchange.py')

diff --git a/shared/tests/test_exchange.py b/shared/tests/test_exchange.py
new file mode 100644
index 0000000..95dc7d7
--- /dev/null
+++ b/shared/tests/test_exchange.py
@@ -0,0 +1,55 @@
+"""Tests for the exchange factory."""
+
+from unittest.mock import patch
+
+import ccxt.async_support as ccxt
+import pytest
+
+from shared.exchange import create_exchange
+
+
+def test_create_exchange_binance():
+    """Verify create_exchange returns a ccxt.binance instance."""
+    exchange = create_exchange(
+        exchange_id="binance",
+        api_key="test-key",
+        api_secret="test-secret",
+    )
+    assert isinstance(exchange, ccxt.binance)
+    assert exchange.apiKey == "test-key"
+    assert exchange.secret == "test-secret"
+    assert exchange.enableRateLimit is True
+
+
+def test_create_exchange_unknown():
+    """Verify create_exchange raises ValueError for unknown exchange."""
+    with pytest.raises(ValueError, match="Unknown exchange 'not_a_real_exchange'"):
+        create_exchange(
+            exchange_id="not_a_real_exchange",
+            api_key="key",
+            api_secret="secret",
+        )
+
+
+def test_create_exchange_with_sandbox():
+    """Verify sandbox mode is activated when sandbox=True."""
+    with patch.object(ccxt.binance, "set_sandbox_mode") as mock_sandbox:
+        exchange = create_exchange(
+            exchange_id="binance",
+            api_key="key",
+            api_secret="secret",
+            sandbox=True,
+        )
+        mock_sandbox.assert_called_once_with(True)
+    assert isinstance(exchange, ccxt.binance)
+
+
+def test_create_exchange_no_sandbox_by_default():
+    """Verify sandbox mode is not set when sandbox=False (default)."""
+    with patch.object(ccxt.binance, "set_sandbox_mode") as mock_sandbox:
+        create_exchange(
+            exchange_id="binance",
+            api_key="key",
+            api_secret="secret",
+        )
+        mock_sandbox.assert_not_called()
-- 
cgit v1.2.3