Skip to content

Commit dd5c1fd

Browse files
test(service): <- more fast withdraws
1 parent 4d801ca commit dd5c1fd

File tree

1 file changed

+23
-5
lines changed

1 file changed

+23
-5
lines changed

test/BasedAppManager.t.sol

Lines changed: 23 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)