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

wayfinder_paths/strategies/moonwell_wsteth_loop_strategy/test_strategy.py

@@ -56,7 +56,6 @@ def strategy():
 
 @pytest.fixture
 def mock_adapter_responses(strategy):
-    # Mock balance adapter
     strategy.balance_adapter.get_balance = AsyncMock(return_value=(True, 1000000))
     strategy.balance_adapter.move_from_main_wallet_to_strategy_wallet = AsyncMock(
         return_value=(True, "success")
@@ -65,7 +64,6 @@ def mock_adapter_responses(strategy):
         return_value=(True, "success")
     )
 
-    # Mock token adapter
     strategy.token_adapter.get_token = AsyncMock(
         return_value=(True, {"decimals": 18, "symbol": "TEST"})
     )
@@ -73,7 +71,6 @@ def mock_adapter_responses(strategy):
         return_value=(True, {"current_price": 1.0})
     )
 
-    # Mock moonwell adapter
     strategy.moonwell_adapter.get_pos = AsyncMock(
         return_value=(
             True,
@@ -102,7 +99,6 @@ def mock_adapter_responses(strategy):
     strategy.moonwell_adapter.set_collateral = AsyncMock(return_value=(True, "success"))
     strategy.moonwell_adapter.claim_rewards = AsyncMock(return_value={})
 
-    # Mock brap adapter
     strategy.brap_adapter.swap_from_token_ids = AsyncMock(
         return_value=(True, {"to_amount": 1000000000000000000})
     )
@@ -116,8 +112,8 @@ async def test_smoke(strategy, mock_adapter_responses):
     examples = load_strategy_examples(Path(__file__))
     smoke_data = examples["smoke"]
 
-    # Mock quote to return positive APY
-    with patch.object(strategy, "quote", new_callable=AsyncMock) as mock_quote:
+    # Mock _quote to return positive APY
+    with patch.object(strategy, "_quote", new_callable=AsyncMock) as mock_quote:
         mock_quote.return_value = {"apy": 0.1, "data": {}}
 
         # Status test
@@ -166,7 +162,7 @@ async def test_canonical_usage(strategy, mock_adapter_responses):
 
     for example_name, example_data in canonical.items():
         # Mock methods for canonical usage tests
-        with patch.object(strategy, "quote", new_callable=AsyncMock) as mock_quote:
+        with patch.object(strategy, "_quote", new_callable=AsyncMock) as mock_quote:
             mock_quote.return_value = {"apy": 0.1, "data": {}}
 
             if "deposit" in example_data:
@@ -226,7 +222,7 @@ async def test_status_returns_status_dict(strategy, mock_adapter_responses):
             ) as mock_peg:
                 mock_peg.return_value = 0.001
                 with patch.object(
-                    strategy, "quote", new_callable=AsyncMock
+                    strategy, "_quote", new_callable=AsyncMock
                 ) as mock_quote:
                     mock_quote.return_value = {"apy": 0.1, "data": {}}
 
@@ -288,7 +284,7 @@ async def test_quote_returns_apy_info(strategy, mock_adapter_responses):
             return_value=mock_response
         )
 
-        quote = await strategy.quote()
+        quote = await strategy._quote()
 
         assert "apy" in quote
         assert "information" in quote or "data" in quote
@@ -624,7 +620,6 @@ async def test_atomic_deposit_iteration_swaps_from_eth_when_borrow_surfaces_as_e
     strategy.moonwell_adapter.borrow = AsyncMock(side_effect=borrow_side_effect)
 
     async def wrap_eth_side_effect(*, amount: int):
-        # Wrap ETH to WETH
         balances[ETH_TOKEN_ID] -= int(amount)
         balances[WETH_TOKEN_ID] += int(amount)
         return (True, {"block_number": 12346})
@@ -634,10 +629,8 @@ async def test_atomic_deposit_iteration_swaps_from_eth_when_borrow_surfaces_as_e
     async def swap_side_effect(
         *, from_token_id: str, to_token_id: str, amount: int, **_
     ):
-        # After wrapping, the swap should be WETH→wstETH
         assert from_token_id == WETH_TOKEN_ID
         assert to_token_id == WSTETH_TOKEN_ID
-        # Simulate receiving wstETH
         balances[WSTETH_TOKEN_ID] += 123
         return {"to_amount": 123}
 
@@ -897,7 +890,6 @@ async def test_partial_liquidate_prefers_wsteth_when_excess(strategy):
     strategy.token_adapter.get_token = AsyncMock(side_effect=mock_get_token)
     strategy.token_adapter.get_token_price = AsyncMock(side_effect=mock_get_price)
 
-    # Wallet balances (raw)
     balances: dict[str, int] = {USDC_TOKEN_ID: 0, WSTETH_TOKEN_ID: 0}
 
     async def mock_get_balance_raw(*, token_id: str, wallet_address: str, **_):
@@ -915,7 +907,6 @@ async def test_partial_liquidate_prefers_wsteth_when_excess(strategy):
     snap.wsteth_price = 2000.0
     snap.wsteth_dec = 18
 
-    # Collateral factors
     strategy.moonwell_adapter.get_collateral_factor = AsyncMock(
         return_value=(True, 0.8)
     )
@@ -987,7 +978,6 @@ async def test_partial_liquidate_prefers_wsteth_when_excess(strategy):
 
 @pytest.mark.asyncio
 async def test_partial_liquidate_uses_usdc_collateral_when_no_wsteth_excess(strategy):
-    # Token metadata
     strategy.token_adapter.get_token = AsyncMock(
         side_effect=lambda token_id: (
             True,