@@ -520,10 +520,10 @@ contract BasedAppManagerTest is Test, OwnableUpgradeable {
520520 vm.assume (amount > 0 && amount < INITIAL_USER1_BALANCE_ERC20);
521521 testStrategyOptInToService ();
522522 vm.startPrank (USER1);
523- uint256 strategyTokenBalance = proxiedManager.strategyTokenBalances (1 , USER1, address (erc20mock ));
523+ uint256 strategyTokenBalance = proxiedManager.strategyTokenBalances (1 , USER1, address (erc20mock2 ));
524524 assertEq (strategyTokenBalance, 0 , "User strategy balance should be 0 " );
525- proxiedManager.depositERC20 (1 , erc20mock , amount);
526- strategyTokenBalance = proxiedManager.strategyTokenBalances (1 , USER1, address (erc20mock ));
525+ proxiedManager.depositERC20 (1 , erc20mock2 , amount);
526+ strategyTokenBalance = proxiedManager.strategyTokenBalances (1 , USER1, address (erc20mock2 ));
527527 assertEq (strategyTokenBalance, amount, "User strategy balance not matching " );
528528 vm.stopPrank ();
529529 }
@@ -592,8 +592,26 @@ contract BasedAppManagerTest is Test, OwnableUpgradeable {
592592 vm.stopPrank ();
593593 }
594594
595- function testCreateObligationToNotOwnedStrategyRevert () public {}
596- function testWithdrawErc20FromStrategy () public {}
595+ function testCreateObligationToNotOwnedStrategyRevert () public {
596+ vm.startPrank (ATTACKER);
597+ vm.expectRevert ("Not the strategy owner " );
598+ proxiedManager.createObligation (1 , SERVICE1, address (erc20mock), 100 );
599+ uint256 strategyTokenBalance = proxiedManager.strategyTokenBalances (1 , ATTACKER, address (erc20mock));
600+ assertEq (strategyTokenBalance, 0 , "User strategy balance should be 0 " );
601+ vm.stopPrank ();
602+ }
603+
604+ function testFastWithdrawErc20FromStrategy () public {
605+ testStrategyOwnerDepositERC20WithNoObligation (200 );
606+ vm.startPrank (USER1);
607+ uint256 strategyTokenBalance = proxiedManager.strategyTokenBalances (1 , USER1, address (erc20mock2));
608+ assertEq (strategyTokenBalance, 200 , "User strategy balance should be 200 " );
609+ proxiedManager.fastWithdrawERC20 (1 , erc20mock2, 50 );
610+ strategyTokenBalance = proxiedManager.strategyTokenBalances (1 , USER1, address (erc20mock2));
611+ assertEq (strategyTokenBalance, 150 , "User strategy balance should be 150 " );
612+ vm.stopPrank ();
613+ }
614+
597615 function testWithdrawETHFromStrategy () public {}
598616 function testUpdateStrategy () public {}
599617 function testRevertObligationWithNonMatchingToken () public {}
0 commit comments