78 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			TypeScript
		
	
	
	
	
	
			
		
		
	
	
			78 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			TypeScript
		
	
	
	
	
	
import { ERC20ProxyContract, ERC20Wrapper } from '@0x/contracts-asset-proxy';
 | 
						|
import { DummyERC20TokenContract } from '@0x/contracts-erc20';
 | 
						|
import { chaiSetup, expectTransactionFailedAsync, provider, web3Wrapper } from '@0x/contracts-test-utils';
 | 
						|
import { BlockchainLifecycle } from '@0x/dev-utils';
 | 
						|
import { RevertReason } from '@0x/types';
 | 
						|
import { BigNumber } from '@0x/utils';
 | 
						|
import * as _ from 'lodash';
 | 
						|
 | 
						|
import { StakingWrapper } from './utils/staking_wrapper';
 | 
						|
 | 
						|
chaiSetup.configure();
 | 
						|
const blockchainLifecycle = new BlockchainLifecycle(web3Wrapper);
 | 
						|
// tslint:disable:no-unnecessary-type-assertion
 | 
						|
describe('Staking Vaults', () => {
 | 
						|
    // constants
 | 
						|
    const ZRX_TOKEN_DECIMALS = new BigNumber(18);
 | 
						|
    // tokens & addresses
 | 
						|
    let accounts: string[];
 | 
						|
    let owner: string;
 | 
						|
    let users: string[];
 | 
						|
    let zrxTokenContract: DummyERC20TokenContract;
 | 
						|
    let erc20ProxyContract: ERC20ProxyContract;
 | 
						|
    // wrappers
 | 
						|
    let stakingWrapper: StakingWrapper;
 | 
						|
    let erc20Wrapper: ERC20Wrapper;
 | 
						|
    // tests
 | 
						|
    before(async () => {
 | 
						|
        await blockchainLifecycle.startAsync();
 | 
						|
    });
 | 
						|
    after(async () => {
 | 
						|
        await blockchainLifecycle.revertAsync();
 | 
						|
    });
 | 
						|
    before(async () => {
 | 
						|
        // create accounts
 | 
						|
        accounts = await web3Wrapper.getAvailableAddressesAsync();
 | 
						|
        owner = accounts[0];
 | 
						|
        users = accounts.slice(1);
 | 
						|
        // deploy erc20 proxy
 | 
						|
        erc20Wrapper = new ERC20Wrapper(provider, accounts, owner);
 | 
						|
        erc20ProxyContract = await erc20Wrapper.deployProxyAsync();
 | 
						|
        // deploy zrx token
 | 
						|
        [zrxTokenContract] = await erc20Wrapper.deployDummyTokensAsync(1, ZRX_TOKEN_DECIMALS);
 | 
						|
        await erc20Wrapper.setBalancesAndAllowancesAsync();
 | 
						|
        // deploy staking contracts
 | 
						|
        stakingWrapper = new StakingWrapper(provider, owner, erc20ProxyContract, zrxTokenContract, accounts);
 | 
						|
        await stakingWrapper.deployAndConfigureContractsAsync();
 | 
						|
    });
 | 
						|
    beforeEach(async () => {
 | 
						|
        await blockchainLifecycle.startAsync();
 | 
						|
    });
 | 
						|
    afterEach(async () => {
 | 
						|
        await blockchainLifecycle.revertAsync();
 | 
						|
    });
 | 
						|
    describe('Reward Vault', () => {
 | 
						|
        it.skip('basic management', async () => {
 | 
						|
            // 1 setup test parameters
 | 
						|
            const poolOperator = users[0];
 | 
						|
            const operatorShare = 39;
 | 
						|
            const poolId = await stakingWrapper.createStakingPoolAsync(poolOperator, operatorShare);
 | 
						|
            const stakingContractAddress = stakingWrapper.getStakingContract().address;
 | 
						|
            const notStakingContractAddress = poolOperator;
 | 
						|
            // create pool in vault
 | 
						|
            await stakingWrapper.rewardVaultRegisterPoolAsync(poolId, operatorShare, stakingContractAddress);
 | 
						|
            // should fail to create pool if it already exists
 | 
						|
            await expectTransactionFailedAsync(
 | 
						|
                stakingWrapper.rewardVaultRegisterPoolAsync(poolId, operatorShare, stakingContractAddress),
 | 
						|
                RevertReason.PoolAlreadyExists,
 | 
						|
            );
 | 
						|
            // should fail to create a pool from an address other than the staking contract
 | 
						|
            await expectTransactionFailedAsync(
 | 
						|
                stakingWrapper.rewardVaultRegisterPoolAsync(poolId, operatorShare, notStakingContractAddress),
 | 
						|
                RevertReason.OnlyCallableByStakingContract,
 | 
						|
            );
 | 
						|
        });
 | 
						|
    });
 | 
						|
});
 | 
						|
// tslint:enable:no-unnecessary-type-assertion
 |