staking end-to-end works

2019-05-22 14:17:23 -07:00
parent 13d2cca2bc
commit c4ca72cf22
3 changed files with 23 additions and 16 deletions
--- a/contracts/staking/test/utils/staking_wrapper.ts
+++ b/contracts/staking/test/utils/staking_wrapper.ts
@@ -77,6 +77,10 @@ export class StakingWrapper {
        const balance = await this._zrxTokenContract.balanceOf.callAsync(holder);
        return balance;
    }
+    public async getZrxTokenBalanceOfZrxVault(): Promise<BigNumber> {
+        const balance = await this._zrxTokenContract.balanceOf.callAsync(this.getZrxVaultContract().address);
+        return balance;
+    }
    public toBaseUnitAmount(amount: BigNumber | number): BigNumber {
        const decimals = 18;
        const amountAsBigNumber = typeof(amount)  === 'number' ? new BigNumber(amount) : amount;