crowetic/protocol
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add test for withdrawing rewards before epoch is finalized

Browse Source
This commit is contained in:
Amir Bandeali 2019-10-19 20:32:39 -07:00
parent fd9084b345
commit 359b804001
1 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
contracts/staking/test/rewards_test.ts
View File

@ -1,5 +1,6 @@
import { ERC20Wrapper } from '@0x/contracts-asset-proxy'; import { ERC20Wrapper } from '@0x/contracts-asset-proxy';
import { blockchainTests, constants, describe, expect, shortZip } from '@0x/contracts-test-utils'; import { blockchainTests, constants, describe, expect, shortZip } from '@0x/contracts-test-utils';
import { StakingRevertErrors } from '@0x/order-utils';
import { BigNumber } from '@0x/utils'; import { BigNumber } from '@0x/utils';
import * as _ from 'lodash'; import * as _ from 'lodash';
@ -584,6 +585,31 @@ blockchainTests.resets('Testing Rewards', env => {
poolRewardBalance: constants.ZERO_AMOUNT, poolRewardBalance: constants.ZERO_AMOUNT,
}); });
}); });
it('should fail to withdraw delegator rewards if the pool has not been finalized for the previous epoch', async () => {
const rewardForDelegator = toBaseUnitAmount(10);
const stakeAmount = toBaseUnitAmount(4);
await stakers[0].stakeAsync(stakeAmount);
await stakers[0].moveStakeAsync(
new StakeInfo(StakeStatus.Undelegated),
new StakeInfo(StakeStatus.Delegated, poolId),
stakeAmount,
);
await stakingApiWrapper.stakingContract.payProtocolFee.awaitTransactionSuccessAsync(
poolOperator.getOwner(),
takerAddress,
rewardForDelegator,
{ from: exchangeAddress, value: rewardForDelegator },
);
const currentEpoch = await stakingApiWrapper.stakingContract.currentEpoch.callAsync();
await stakingApiWrapper.utils.fastForwardToNextEpochAsync();
await stakingApiWrapper.utils.endEpochAsync();
const expectedError = new StakingRevertErrors.PoolNotFinalizedError(poolId, currentEpoch);
expect(
stakingApiWrapper.stakingContract.withdrawDelegatorRewards.awaitTransactionSuccessAsync(poolId, {
from: stakers[0].getOwner(),
}),
).to.revertWith(expectedError);
});
it(`payout should be based on stake at the time of rewards`, async () => { it(`payout should be based on stake at the time of rewards`, async () => {
const staker = stakers[0]; const staker = stakers[0];
const stakeAmount = toBaseUnitAmount(5); const stakeAmount = toBaseUnitAmount(5);