Add csm module in tests

lidofinance · Nov 11, 2024 · 4d73699 · 4d73699
1 parent 436eb3d
commit 4d73699
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/bots/test_depositor.py b/tests/bots/test_depositor.py
@@ -18,8 +18,8 @@ def depositor_bot(
     module_recommender,
 ):
     variables.MESSAGE_TRANSPORTS = ''
-    variables.DEPOSIT_MODULES_WHITELIST = [1, 2]
-    web3_lido_unit.lido.staking_router.get_staking_module_ids = Mock(return_value=[1, 2])
+    variables.DEPOSIT_MODULES_WHITELIST = [1, 2, 3]
+    web3_lido_unit.lido.staking_router.get_staking_module_ids = Mock(return_value=[1, 2, 3])
     web3_lido_unit.eth.get_block = Mock(return_value=block_data)
     yield DepositorBot(web3_lido_unit, deposit_transaction_sender, base_deposit_strategy, csm_strategy, module_recommender)