Skip to content

Commit f76ce16

Browse files
committed
add assertions
1 parent ba1b067 commit f76ce16

File tree

2 files changed

+16
-3
lines changed

2 files changed

+16
-3
lines changed

packages/ethereum-contracts/test/foundry/FoundrySuperfluidTester.t.sol

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1278,6 +1278,21 @@ contract FoundrySuperfluidTester is Test {
12781278
// _assertRealTimeBalances(ISuperToken(address(poolSuperToken)));
12791279
}
12801280

1281+
function _helperClaimAll(ISuperfluidPool pool_, address caller_, address member_) internal {
1282+
(int256 claimableBefore,) = pool_.getClaimableNow(member_);
1283+
(int256 balanceBefore,,,) = pool_.superToken().realtimeBalanceOfNow(member_);
1284+
1285+
vm.startPrank(caller_);
1286+
pool_.claimAll(member_);
1287+
vm.stopPrank();
1288+
1289+
(int256 claimableAfter,) = pool_.getClaimableNow(member_);
1290+
(int256 balanceAfter,,,) = pool_.superToken().realtimeBalanceOfNow(member_);
1291+
1292+
assertEq(claimableAfter, 0, "GDAv1.t: Member claimable amount should be 0");
1293+
assertEq(balanceAfter, balanceBefore + claimableBefore, "GDAv1.t: Member balance should increase by claimable amount");
1294+
}
1295+
12811296
function _helperConnectPool(address caller_, ISuperToken superToken_, ISuperfluidPool pool_, bool useForwarder_)
12821297
internal
12831298
{

packages/ethereum-contracts/test/foundry/agreements/gdav1/GeneralDistributionAgreement.t.sol

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -995,9 +995,7 @@ contract GeneralDistributionAgreementV1IntegrationTest is FoundrySuperfluidTeste
995995
address u4 = TEST_ACCOUNTS[1 + (s.v % N_MEMBERS)];
996996
emit log_named_string("action", "claimAll");
997997
emit log_named_address("claim for", u4);
998-
vm.startPrank(user);
999-
assert(freePool.claimAll(u4));
1000-
vm.stopPrank();
998+
_helperClaimAll(freePool, user, u4);
1001999
} else if (action == 3) {
10021000
bool doConnect = s.v % 2 == 0 ? false : true;
10031001
emit log_named_string("action", "doConnectPool");

0 commit comments

Comments
 (0)