@@ -9,6 +9,7 @@ import "src/test/mocks/ERC20Mock.sol";
99import "src/test/mocks/ERC20_SetTransferReverting_Mock.sol " ;
1010import "src/test/mocks/Reverter.sol " ;
1111import "src/test/mocks/Reenterer.sol " ;
12+ import "src/test/mocks/MockDecimals.sol " ;
1213import "src/test/events/IStrategyManagerEvents.sol " ;
1314import "src/test/utils/EigenLayerUnitTestSetup.sol " ;
1415
@@ -367,7 +368,7 @@ contract StrategyManagerUnitTests_depositIntoStrategy is StrategyManagerUnitTest
367368
368369 function test_Revert_WhenTokenSafeTransferFromReverts () external {
369370 // replace 'dummyStrat' with one that uses a reverting token
370- dummyToken = IERC20 (address (new Reverter ()));
371+ dummyToken = IERC20 (address (new ReverterWithDecimals ()));
371372 dummyStrat = _deployNewStrategy (dummyToken, strategyManager, pauserRegistry, dummyAdmin);
372373
373374 // whitelist the strategy for deposit
@@ -390,8 +391,8 @@ contract StrategyManagerUnitTests_depositIntoStrategy is StrategyManagerUnitTest
390391 }
391392
392393 function test_Revert_WhenTokenDoesNotExist () external {
393- // replace 'dummyStrat' with one that uses a non-existent token
394- dummyToken = IERC20 (address (5678 ));
394+ // replace 'dummyStrat' with one that uses a non-existent token, but will pass the initializer decimals check
395+ dummyToken = IERC20 (address (new MockDecimals () ));
395396 dummyStrat = _deployNewStrategy (dummyToken, strategyManager, pauserRegistry, dummyAdmin);
396397
397398 // whitelist the strategy for deposit
@@ -408,7 +409,7 @@ contract StrategyManagerUnitTests_depositIntoStrategy is StrategyManagerUnitTest
408409 IStrategy strategy = dummyStrat;
409410
410411 cheats.prank (staker);
411- cheats.expectRevert ("Address: call to non-contract " );
412+ cheats.expectRevert ("SafeERC20: low-level call failed " );
412413 strategyManager.depositIntoStrategy (strategy, token, amount);
413414 }
414415
0 commit comments