Fix test

tradingstrategy-ai · Dec 31, 2024 · 7b0dac8 · 7b0dac8
1 parent b5e1370
commit 7b0dac8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/tests/safe-integration/test_guard_safe_e2e.py b/tests/safe-integration/test_guard_safe_e2e.py
@@ -378,4 +378,3 @@ def test_swap_through_module_unauthorised(
     with pytest.raises(ValueError) as e:
         ts_module.functions.performCall(target, call_data).transact({"from": attacker_account})
     assert "validateCall: Sender not allowed" in str(e)
-
diff --git a/tests/safe-integration/test_guard_safe_uniswap_v2.py b/tests/safe-integration/test_guard_safe_uniswap_v2.py
@@ -219,8 +219,7 @@ def test_safe_module_initialised(
     """Vault and guard are initialised for the owner."""
     assert guard.functions.owner().call() == owner
     assert guard.functions.isAllowedSender(asset_manager).call() is True
-    assert guard.functions.isAllowedWithdrawDestination(owner).call() is True
-    assert guard.functions.isAllowedWithdrawDestination(asset_manager).call() is False
+    assert guard.functions.isAllowedSender(safe.address).call() is False
 
     # We have accessed needed for a swap
     assert guard.functions.callSiteCount().call() == 5
Original file line number	Diff line number	Diff line change
Expand Up		@@ -378,4 +378,3 @@ def test_swap_through_module_unauthorised(
		with pytest.raises(ValueError) as e:
		ts_module.functions.performCall(target, call_data).transact({"from": attacker_account})
		assert "validateCall: Sender not allowed" in str(e)