Merge pull request #26 from 0xProject/erc20Wrapper

Implement TokenWrapper functions & tests

6 files changed, 206 insertions, 2 deletions

diff --git a/src/0x.js.ts b/src/0x.js.ts
index 69c0cc567..d231c579e 100644
--- a/src/0x.js.ts
+++ b/src/0x.js.ts
@@ -13,6 +13,7 @@ import compareVersions = require('compare-versions');
 import {ExchangeWrapper} from './contract_wrappers/exchange_wrapper';
 import {TokenRegistryWrapper} from './contract_wrappers/token_registry_wrapper';
 import {ecSignatureSchema} from './schemas/ec_signature_schema';
+import {TokenWrapper} from './contract_wrappers/token_wrapper';
 import {SolidityTypes, ECSignature, ZeroExError} from './types';
 
 const MAX_DIGITS_IN_UNSIGNED_256_INT = 78;
@@ -20,6 +21,7 @@ const MAX_DIGITS_IN_UNSIGNED_256_INT = 78;
 export class ZeroEx {
     public exchange: ExchangeWrapper;
     public tokenRegistry: TokenRegistryWrapper;
+    public token: TokenWrapper;
     private web3Wrapper: Web3Wrapper;
     /**
      * Computes the orderHash given the order parameters and returns it as a hex encoded string.
@@ -135,6 +137,7 @@ export class ZeroEx {
         this.web3Wrapper = new Web3Wrapper(web3);
         this.exchange = new ExchangeWrapper(this.web3Wrapper);
         this.tokenRegistry = new TokenRegistryWrapper(this.web3Wrapper);
+        this.token = new TokenWrapper(this.web3Wrapper);
     }
     /**
      * Sets a new provider for the web3 instance used by 0x.js
@@ -143,6 +146,7 @@ export class ZeroEx {
         this.web3Wrapper.setProvider(provider);
         this.exchange.invalidateContractInstance();
         this.tokenRegistry.invalidateContractInstance();
+        this.token.invalidateContractInstances();
     }
     /**
      * Signs an orderHash and returns it's elliptic curve signature
diff --git a/src/contract_wrappers/token_wrapper.ts b/src/contract_wrappers/token_wrapper.ts
new file mode 100644
index 000000000..35e008905
--- /dev/null
+++ b/src/contract_wrappers/token_wrapper.ts
@@ -0,0 +1,93 @@
+import * as _ from 'lodash';
+import * as BigNumber from 'bignumber.js';
+import {Web3Wrapper} from '../web3_wrapper';
+import {assert} from '../utils/assert';
+import {constants} from '../utils/constants';
+import {ContractWrapper} from './contract_wrapper';
+import * as TokenArtifacts from '../artifacts/Token.json';
+import * as ProxyArtifacts from '../artifacts/Proxy.json';
+import {TokenContract, ZeroExError} from '../types';
+
+const ALLOWANCE_TO_ZERO_GAS_AMOUNT = 45730;
+
+export class TokenWrapper extends ContractWrapper {
+    private tokenContractsByAddress: {[address: string]: TokenContract};
+    constructor(web3Wrapper: Web3Wrapper) {
+        super(web3Wrapper);
+        this.tokenContractsByAddress = {};
+    }
+    public invalidateContractInstances() {
+        this.tokenContractsByAddress = {};
+    }
+    /**
+     * Returns an owner's ERC20 token balance.
+     */
+    public async getBalanceAsync(tokenAddress: string, ownerAddress: string): Promise<BigNumber.BigNumber> {
+        assert.isETHAddressHex('ownerAddress', ownerAddress);
+        assert.isETHAddressHex('tokenAddress', tokenAddress);
+
+        const tokenContract = await this.getTokenContractAsync(tokenAddress);
+        let balance = await tokenContract.balanceOf.call(ownerAddress);
+        // The BigNumber instance returned by Web3 is of a much older version then our own, we therefore
+        // should always re-instantiate the returned BigNumber after retrieval.
+        balance = new BigNumber(balance);
+        return balance;
+    }
+    /**
+     * Retrieves the allowance in baseUnits of the ERC20 token set to the 0x proxy contract
+     * by an owner address.
+     */
+    public async getProxyAllowanceAsync(tokenAddress: string, ownerAddress: string) {
+        assert.isETHAddressHex('ownerAddress', ownerAddress);
+        assert.isETHAddressHex('tokenAddress', tokenAddress);
+
+        const tokenContract = await this.getTokenContractAsync(tokenAddress);
+        const proxyAddress = await this.getProxyAddressAsync();
+        let allowanceInBaseUnits = await tokenContract.allowance.call(ownerAddress, proxyAddress);
+        allowanceInBaseUnits = new BigNumber(allowanceInBaseUnits);
+        return allowanceInBaseUnits;
+    }
+    /**
+     * Sets the 0x proxy contract's allowance to a specified number of a tokens' baseUnits on behalf
+     * of an owner address.
+     */
+    public async setProxyAllowanceAsync(tokenAddress: string, ownerAddress: string,
+                                        amountInBaseUnits: BigNumber.BigNumber) {
+        assert.isETHAddressHex('ownerAddress', ownerAddress);
+        assert.isETHAddressHex('tokenAddress', tokenAddress);
+        assert.isBigNumber('amountInBaseUnits', amountInBaseUnits);
+
+        const tokenContract = await this.getTokenContractAsync(tokenAddress);
+        const proxyAddress = await this.getProxyAddressAsync();
+        // Hack: for some reason default estimated gas amount causes `base fee exceeds gas limit` exception
+        // on testrpc. Probably related to https://github.com/ethereumjs/testrpc/issues/294
+        // TODO: Debug issue in testrpc and submit a PR, then remove this hack
+        const networkIdIfExists = await this.web3Wrapper.getNetworkIdIfExistsAsync();
+        const gas = networkIdIfExists === constants.TESTRPC_NETWORK_ID ? ALLOWANCE_TO_ZERO_GAS_AMOUNT : undefined;
+        await tokenContract.approve(proxyAddress, amountInBaseUnits, {
+            from: ownerAddress,
+            gas,
+        });
+    }
+    private async getTokenContractAsync(tokenAddress: string): Promise<TokenContract> {
+        let tokenContract = this.tokenContractsByAddress[tokenAddress];
+        if (!_.isUndefined(tokenContract)) {
+            return tokenContract;
+        }
+        const contractInstance = await this.instantiateContractIfExistsAsync((TokenArtifacts as any), tokenAddress);
+        tokenContract = contractInstance as TokenContract;
+        this.tokenContractsByAddress[tokenAddress] = tokenContract;
+        return tokenContract;
+    }
+    private async getProxyAddressAsync() {
+        const networkIdIfExists = await this.web3Wrapper.getNetworkIdIfExistsAsync();
+        const proxyNetworkConfigsIfExists = _.isUndefined(networkIdIfExists) ?
+                                       undefined :
+                                       (ProxyArtifacts as any).networks[networkIdIfExists];
+        if (_.isUndefined(proxyNetworkConfigsIfExists)) {
+            throw new Error(ZeroExError.CONTRACT_NOT_DEPLOYED_ON_NETWORK);
+        }
+        const proxyAddress = proxyNetworkConfigsIfExists.address;
+        return proxyAddress;
+    }
+}
diff --git a/src/types.ts b/src/types.ts
index 6fce95706..3619c800f 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -14,6 +14,7 @@ export const ZeroExError = strEnum([
   'UNHANDLED_ERROR',
   'USER_HAS_NO_ASSOCIATED_ADDRESSES',
   'INVALID_SIGNATURE',
+  'CONTRACT_NOT_DEPLOYED_ON_NETWORK',
 ]);
 export type ZeroExError = keyof typeof ZeroExError;
 
@@ -30,6 +31,16 @@ export interface ExchangeContract {
     isValidSignature: any;
 }
 
+export interface TokenContract {
+    balanceOf: {
+        call: (address: string) => Promise<BigNumber.BigNumber>;
+    };
+    allowance: {
+        call: (ownerAddress: string, allowedAddress: string) => Promise<BigNumber.BigNumber>;
+    };
+    approve: (proxyAddress: string, amountInBaseUnits: BigNumber.BigNumber, txOpts: TxOpts) => void;
+}
+
 export interface TokenRegistryContract {
     getTokenMetaData: {
         call: (address: string) => Promise<TokenMetadata>;
@@ -55,3 +66,8 @@ export interface Token {
     decimals: number;
     url: string;
 };
+
+export interface TxOpts {
+    from: string;
+    gas?: number;
+};
diff --git a/src/utils/constants.ts b/src/utils/constants.ts
index ec2fe744a..5a5ba0e0a 100644
--- a/src/utils/constants.ts
+++ b/src/utils/constants.ts
@@ -1,3 +1,4 @@
 export const constants = {
     NULL_ADDRESS: '0x0000000000000000000000000000000000000000',
+    TESTRPC_NETWORK_ID: 50,
 };
diff --git a/test/exchange_wrapper.ts b/test/exchange_wrapper.ts
index 55b84ce36..e42454089 100644
--- a/test/exchange_wrapper.ts
+++ b/test/exchange_wrapper.ts
@@ -1,12 +1,10 @@
 import 'mocha';
 import * as chai from 'chai';
-import chaiAsPromised = require('chai-as-promised');
 import {web3Factory} from './utils/web3_factory';
 import {ZeroEx} from '../src/0x.js';
 import {BlockchainLifecycle} from './utils/blockchain_lifecycle';
 
 const expect = chai.expect;
-chai.use(chaiAsPromised);
 const blockchainLifecycle = new BlockchainLifecycle();
 
 describe('ExchangeWrapper', () => {
diff --git a/test/token_wrapper_test.ts b/test/token_wrapper_test.ts
new file mode 100644
index 000000000..5ce3efb14
--- /dev/null
+++ b/test/token_wrapper_test.ts
@@ -0,0 +1,92 @@
+import 'mocha';
+import * as chai from 'chai';
+import * as Web3 from 'web3';
+import * as BigNumber from 'bignumber.js';
+import promisify = require('es6-promisify');
+import {web3Factory} from './utils/web3_factory';
+import {ZeroEx} from '../src/0x.js';
+import {ZeroExError, Token} from '../src/types';
+import {BlockchainLifecycle} from './utils/blockchain_lifecycle';
+
+const expect = chai.expect;
+const blockchainLifecycle = new BlockchainLifecycle();
+
+describe('TokenWrapper', () => {
+    let web3: Web3;
+    let zeroEx: ZeroEx;
+    let userAddresses: string[];
+    let tokens: Token[];
+    before(async () => {
+        web3 = web3Factory.create();
+        zeroEx = new ZeroEx(web3);
+        userAddresses = await promisify(web3.eth.getAccounts)();
+        tokens = await zeroEx.tokenRegistry.getTokensAsync();
+    });
+    beforeEach(async () => {
+        await blockchainLifecycle.startAsync();
+    });
+    afterEach(async () => {
+        await blockchainLifecycle.revertAsync();
+    });
+    describe('#getBalanceAsync', () => {
+        it('should return the balance for an existing ERC20 token', async () => {
+            const aToken = tokens[0];
+            const aOwnerAddress = userAddresses[0];
+            const balance = await zeroEx.token.getBalanceAsync(aToken.address, aOwnerAddress);
+            const expectedBalance = new BigNumber('100000000000000000000000000');
+            expect(balance).to.be.bignumber.equal(expectedBalance);
+        });
+        it('should throw a CONTRACT_DOES_NOT_EXIST error for a non-existent token contract', async () => {
+            const nonExistentTokenAddress = '0x9dd402f14d67e001d8efbe6583e51bf9706aa065';
+            const aOwnerAddress = userAddresses[0];
+            expect(zeroEx.token.getBalanceAsync(nonExistentTokenAddress, aOwnerAddress))
+                .to.be.rejectedWith(ZeroExError.CONTRACT_DOES_NOT_EXIST);
+        });
+        it('should return a balance of 0 for a non-existent owner address', async () => {
+            const aToken = tokens[0];
+            const aNonExistentOwner = '0x198C6Ad858F213Fb31b6FE809E25040E6B964593';
+            const balance = await zeroEx.token.getBalanceAsync(aToken.address, aNonExistentOwner);
+            const expectedBalance = new BigNumber('0');
+            expect(balance).to.be.bignumber.equal(expectedBalance);
+        });
+    });
+    describe('#getProxyAllowanceAsync', () => {
+        it('should get the proxy allowance', async () => {
+            const aToken = tokens[0];
+            const aOwnerAddress = userAddresses[0];
+
+            const amountInUnits = new BigNumber('50');
+            const amountInBaseUnits = ZeroEx.toBaseUnitAmount(amountInUnits, aToken.decimals);
+            await zeroEx.token.setProxyAllowanceAsync(aToken.address, aOwnerAddress, amountInBaseUnits);
+
+            const allowance = await zeroEx.token.getProxyAllowanceAsync(aToken.address, aOwnerAddress);
+            const expectedAllowance = amountInBaseUnits;
+            expect(allowance).to.be.bignumber.equal(expectedAllowance);
+        });
+        it('should return 0 if no allowance set yet', async () => {
+            const aToken = tokens[0];
+            const aOwnerAddress = userAddresses[0];
+            const allowance = await zeroEx.token.getProxyAllowanceAsync(aToken.address, aOwnerAddress);
+            const expectedAllowance = new BigNumber('0');
+            expect(allowance).to.be.bignumber.equal(expectedAllowance);
+        });
+    });
+    describe('#setProxyAllowanceAsync', () => {
+        it('should set the proxy allowance', async () => {
+            const aToken = tokens[0];
+            const aOwnerAddress = userAddresses[0];
+
+            const allowanceBeforeSet = await zeroEx.token.getProxyAllowanceAsync(aToken.address, aOwnerAddress);
+            const expectedAllowanceBeforeAllowanceSet = new BigNumber('0');
+            expect(allowanceBeforeSet).to.be.bignumber.equal(expectedAllowanceBeforeAllowanceSet);
+
+            const amountInUnits = new BigNumber('50');
+            const amountInBaseUnits = ZeroEx.toBaseUnitAmount(amountInUnits, aToken.decimals);
+            await zeroEx.token.setProxyAllowanceAsync(aToken.address, aOwnerAddress, amountInBaseUnits);
+
+            const allowanceAfterSet = await zeroEx.token.getProxyAllowanceAsync(aToken.address, aOwnerAddress);
+            const expectedAllowanceAfterAllowanceSet = amountInBaseUnits;
+            expect(allowanceAfterSet).to.be.bignumber.equal(expectedAllowanceAfterAllowanceSet);
+        });
+    });
+});