Skip to content

Commit 67a0146

Browse files
author
Joey Santoro
committed
update tests
1 parent a767c3e commit 67a0146

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

src/test/xTRIBE.t.sol

+11-3
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,12 @@ contract xTRIBETest is DSTestPlus {
7373
uint128 transferAmount
7474
) public {
7575
// setup
76-
hevm.assume(mintAmount != 0 && transferAmount <= mintAmount);
76+
hevm.assume(
77+
mintAmount != 0 &&
78+
transferAmount <= mintAmount &&
79+
user != address(0) &&
80+
delegate != address(0)
81+
);
7782
rewardToken.mint(user, mintAmount);
7883
xTribe.setMaxDelegates(1);
7984

@@ -87,12 +92,12 @@ contract xTRIBETest is DSTestPlus {
8792
// expect revert and early return if user tries to delegate more than they have
8893
if (delegationAmount > mintAmount) {
8994
hevm.expectRevert(abi.encodeWithSignature("DelegationError()"));
90-
xTribe.delegate(delegate, delegationAmount);
95+
xTribe.incrementDelegation(delegate, delegationAmount);
9196
return;
9297
}
9398

9499
// user can successfully delegate
95-
xTribe.delegate(delegate, delegationAmount);
100+
xTribe.incrementDelegation(delegate, delegationAmount);
96101
require(xTribe.userDelegatedVotes(user) == delegationAmount);
97102
require(xTribe.numCheckpoints(delegate) == 1);
98103
require(xTribe.checkpoints(delegate, 0).votes == delegationAmount);
@@ -218,6 +223,9 @@ contract xTRIBETest is DSTestPlus {
218223
) public {
219224
xTribe.setMaxGauges(20);
220225
for (uint256 i = 0; i < 20; i++) {
226+
hevm.assume(
227+
!(xTribe.isGauge(gauges[i]) || gauges[i] == address(0))
228+
);
221229
uint32 warp = warps[i] % xTribe.gaugeCycleLength();
222230
address user = users[i];
223231
address gauge = gauges[i];

0 commit comments

Comments
 (0)