fix: use monkeypatch

lidofinance · Feb 12, 2025 · 1b3caa4 · 1b3caa4
1 parent c5bc4a9
commit 1b3caa4
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/modules/accounting/bunker/test_bunker_abnormal_cl_rebase.py b/tests/modules/accounting/bunker/test_bunker_abnormal_cl_rebase.py
@@ -347,6 +347,7 @@ def test_get_nearest_and_distant_blockstamps(
     ],
 )
 def test_calculate_cl_rebase_between_blocks(
+    monkeypatch,
     web3,
     prev_lido_validators,
     curr_lido_validators,
@@ -363,7 +364,12 @@ def test_calculate_cl_rebase_between_blocks(
     abnormal_case.w3.lido_contracts = Mock()
     abnormal_case.w3.cc.get_validators_no_cache = Mock()
 
-    LidoValidatorsProvider.merge_validators_with_keys = Mock(return_value=prev_lido_validators)
+    with monkeypatch.context():
+        monkeypatch.setattr(
+            LidoValidatorsProvider,
+            "merge_validators_with_keys",
+            Mock(return_value=prev_lido_validators),
+        )
     abnormal_case.lido_validators = curr_lido_validators
     abnormal_case.w3.lido_contracts.get_withdrawal_balance_no_cache = Mock(
         side_effect=[curr_withdrawals_vault_balance, prev_withdrawals_vault_balance]