Fix random 1delta error

tradingstrategy-ai · Jul 23, 2024 · 423d93b · 423d93b
1 parent 05f0a40
commit 423d93b
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/eth_defi/enzyme/generic_adapter_vault.py b/eth_defi/enzyme/generic_adapter_vault.py
@@ -48,6 +48,7 @@
 
 def _get_chain_slug(web3: Web3) -> str:
     return {
+        31337: "anvil",  # only for testing
         1: "ethereum",
         137: "polygon",
         42161: "arbitrum",

diff --git a/tests/guard/test_guard_simple_vault_aave_v3.py b/tests/guard/test_guard_simple_vault_aave_v3.py
@@ -64,7 +64,7 @@ def anvil_polygon_chain_fork(request, large_usdc_holder) -> str:
     launch = fork_network_anvil(
         mainnet_rpc,
         unlocked_addresses=[large_usdc_holder],
-        fork_block_number=51_000_000,
+        fork_block_number=58_000_000,
     )
     try:
         yield launch.json_rpc_url

diff --git a/tests/guard/test_guard_simple_vault_one_delta.py b/tests/guard/test_guard_simple_vault_one_delta.py
@@ -325,6 +325,8 @@ def test_guard_can_short(
     tx_hash = vault.functions.performCall(target, call_data).transact({"from": asset_manager})
     assert_transaction_success_with_explanation(web3, tx_hash, tracing=True)
 
+    mine(web3)
+
     # we should have some vWETH in the vault
     assert vweth.functions.balanceOf(vault.address).call() == pytest.approx(weth_amount_to_short)
     assert ausdc.functions.balanceOf(vault.address).call() == pytest.approx(12233287803)
@@ -346,6 +348,8 @@ def test_guard_can_short(
     tx_hash = vault.functions.performCall(target, call_data).transact({"from": asset_manager})
     assert_transaction_success_with_explanation(web3, tx_hash, tracing=True)
 
+    mine(web3)
+
     # check new position size
     assert vweth.functions.balanceOf(vault.address).call() == pytest.approx(weth_amount_to_short * 2)
     assert ausdc.functions.balanceOf(vault.address).call() == pytest.approx(24462991693)
@@ -369,6 +373,8 @@ def test_guard_can_short(
     tx_hash = vault.functions.performCall(target, call_data).transact({"from": asset_manager})
     assert_transaction_success_with_explanation(web3, tx_hash, tracing=True)
 
+    mine(web3)
+
     # check new position size
     assert vweth.functions.balanceOf(vault.address).call() == pytest.approx(weth_amount_to_short)
     assert ausdc.functions.balanceOf(vault.address).call() == pytest.approx(22219843923)