wayfinder-paths 0.1.22__py3-none-any.whl → 0.1.24__py3-none-any.whl

wayfinder_paths/adapters/balance_adapter/test_adapter.py

@@ -6,157 +6,74 @@ from wayfinder_paths.adapters.balance_adapter.adapter import BalanceAdapter
 
 
 class TestBalanceAdapter:
-    @pytest.fixture
-    def mock_wallet_client(self):
-        mock_client = AsyncMock()
-        return mock_client
-
     @pytest.fixture
     def mock_token_client(self):
-        mock_client = AsyncMock()
-        return mock_client
+        return AsyncMock()
 
     @pytest.fixture
-    def adapter(self, mock_wallet_client, mock_token_client):
-        with (
-            patch(
-                "wayfinder_paths.adapters.balance_adapter.adapter.WalletClient",
-                return_value=mock_wallet_client,
-            ),
-            patch(
-                "wayfinder_paths.adapters.balance_adapter.adapter.TokenClient",
-                return_value=mock_token_client,
-            ),
+    def adapter(self, mock_token_client):
+        with patch(
+            "wayfinder_paths.adapters.balance_adapter.adapter.TokenClient",
+            return_value=mock_token_client,
         ):
             return BalanceAdapter(config={})
 
-    @pytest.mark.asyncio
-    async def test_health_check(self, adapter):
-        health = await adapter.health_check()
-        assert isinstance(health, dict)
-        assert health.get("status") in {"healthy", "unhealthy", "error"}
-
-    @pytest.mark.asyncio
-    async def test_connect(self, adapter):
-        ok = await adapter.connect()
-        assert isinstance(ok, bool)
-
     def test_adapter_type(self, adapter):
         assert adapter.adapter_type == "BALANCE"
 
     @pytest.mark.asyncio
-    async def test_get_balance_with_query_string(
-        self, adapter, mock_token_client, mock_wallet_client
-    ):
+    async def test_get_balance_with_token_id(self, adapter, mock_token_client):
         mock_token_client.get_token_details = AsyncMock(
             return_value={
                 "token_id": "usd-coin-base",
-                "address": "0x123",
+                "address": "0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913",
                 "chain": {"id": 8453, "code": "base"},
             }
         )
-        mock_wallet_client.get_token_balance_for_address = AsyncMock(
-            return_value={"balance": 1000000}
-        )
-
-        success, balance = await adapter.get_balance(
-            query="usd-coin-base",
-            wallet_address="0xWallet",
-        )
 
-        assert success
-        assert balance == 1000000
-        mock_token_client.get_token_details.assert_called_once_with("usd-coin-base")
-        mock_wallet_client.get_token_balance_for_address.assert_called_once_with(
-            wallet_address="0xWallet",
-            query="usd-coin-base",
-            chain_id=8453,
-        )
+        with patch(
+            "wayfinder_paths.adapters.balance_adapter.adapter.get_token_balance",
+            new_callable=AsyncMock,
+            return_value=1000000,
+        ) as mock_get_balance:
+            success, balance = await adapter.get_balance(
+                token_id="usd-coin-base",
+                wallet_address="0xWallet",
+            )
+
+            assert success
+            assert balance == 1000000
+            mock_token_client.get_token_details.assert_called_once_with("usd-coin-base")
+            mock_get_balance.assert_called_once_with(
+                "0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913", 8453, "0xWallet"
+            )
 
     @pytest.mark.asyncio
-    async def test_get_balance_with_query_dict(
-        self, adapter, mock_token_client, mock_wallet_client
-    ):
-        mock_token_client.get_token_details = AsyncMock(
-            return_value={
-                "token_id": "wsteth-base",
-                "address": "0x456",
-                "chain": {"id": 8453, "code": "base"},
-            }
-        )
-        mock_wallet_client.get_token_balance_for_address = AsyncMock(
-            return_value={"balance": 3000000}
-        )
-
-        success, balance = await adapter.get_balance(
-            query={"token_id": "wsteth-base"},
-            wallet_address="0x123",
-        )
-        assert success
-        assert balance == 3000000
-        mock_token_client.get_token_details.assert_called_once_with("wsteth-base")
-        mock_wallet_client.get_token_balance_for_address.assert_called_once_with(
-            wallet_address="0x123",
-            query="wsteth-base",
-            chain_id=8453,
-        )
-
-    @pytest.mark.asyncio
-    async def test_get_balance_missing_query(self, adapter):
-        success, result = await adapter.get_balance(query={}, wallet_address="0xabc")
-        assert success is False
-        assert "missing query" in str(result)
-
-    @pytest.mark.asyncio
-    async def test_get_balance_with_pool_address(
-        self, adapter, mock_token_client, mock_wallet_client
-    ):
-        mock_wallet_client.get_token_balance_for_address = AsyncMock(
-            return_value={"balance": 5000000}
-        )
-        mock_token_client.get_token_details = AsyncMock()
-
-        success, balance = await adapter.get_balance(
-            query="0xPoolAddress",
-            wallet_address="0xWallet",
-            chain_id=8453,
-        )
-
-        assert success
-        assert balance == 5000000
-        mock_wallet_client.get_token_balance_for_address.assert_called_once_with(
-            wallet_address="0xWallet",
-            query="0xPoolAddress",
-            chain_id=8453,
-        )
-        mock_token_client.get_token_details.assert_not_called()
+    async def test_get_balance_with_token_address(self, adapter, mock_token_client):
+        with patch(
+            "wayfinder_paths.adapters.balance_adapter.adapter.get_token_balance",
+            new_callable=AsyncMock,
+            return_value=5000000,
+        ) as mock_get_balance:
+            success, balance = await adapter.get_balance(
+                token_address="0xTokenAddress",
+                wallet_address="0xWallet",
+                chain_id=8453,
+            )
+
+            assert success
+            assert balance == 5000000
+            mock_get_balance.assert_called_once_with("0xTokenAddress", 8453, "0xWallet")
+            mock_token_client.get_token_details.assert_not_called()
 
     @pytest.mark.asyncio
     async def test_get_balance_token_not_found(self, adapter, mock_token_client):
         mock_token_client.get_token_details = AsyncMock(return_value=None)
 
         success, error = await adapter.get_balance(
-            query="invalid-token",
-            wallet_address="0xWallet",
-        )
-
-        assert success is False
-        assert "Token not found" in str(error)
-
-    @pytest.mark.asyncio
-    async def test_get_balance_missing_chain_id(self, adapter, mock_token_client):
-        mock_token_client.get_token_details = AsyncMock(
-            return_value={
-                "token_id": "token-without-chain",
-                "address": "0x123",
-                "chain": {},
-            }
-        )
-
-        success, error = await adapter.get_balance(
-            query="token-without-chain",
+            token_id="invalid-token",
             wallet_address="0xWallet",
         )
 
         assert success is False
-        assert "missing a chain id" in str(error)
+        assert "NoneType" in error or "subscriptable" in error

wayfinder_paths/adapters/boros_adapter/__init__.py

@@ -0,0 +1,17 @@
+"""Boros Adapter - wraps Boros API for fixed-rate market operations."""
+
+from .adapter import BorosAdapter
+from .types import (
+    BorosLimitOrder,
+    BorosMarketQuote,
+    BorosTenorQuote,
+    MarginHealth,
+)
+
+__all__ = [
+    "BorosAdapter",
+    "BorosMarketQuote",
+    "BorosTenorQuote",
+    "BorosLimitOrder",
+    "MarginHealth",
+]