diff --git a/tests/bots/test_depositor.py b/tests/bots/test_depositor.py index 61b52d1..30cf700 100644 --- 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)