import * as chai from 'chai'; import { Artifacts } from '../../../util/artifacts'; import { Balances } from '../../../util/balances'; import { constants } from '../../../util/constants'; import { ContractInstance } from '../../../util/types'; import { chaiSetup } from '../utils/chai_setup'; chaiSetup.configure(); const expect = chai.expect; const { TokenTransferProxy, DummyToken, TokenRegistry } = new Artifacts(artifacts); contract('TokenTransferProxy', (accounts: string[]) => { const INIT_BAL = 100000000; const INIT_ALLOW = 100000000; const owner = accounts[0]; const notAuthorized = owner; let tokenTransferProxy: ContractInstance; let tokenRegistry: ContractInstance; let rep: ContractInstance; let dmyBalances: Balances; before(async () => { [tokenTransferProxy, tokenRegistry] = await Promise.all([ TokenTransferProxy.deployed(), TokenRegistry.deployed(), ]); const repAddress = await tokenRegistry.getTokenAddressBySymbol('REP'); rep = DummyToken.at(repAddress); dmyBalances = new Balances([rep], [accounts[0], accounts[1]]); await Promise.all([ rep.approve(TokenTransferProxy.address, INIT_ALLOW, { from: accounts[0], }), rep.setBalance(accounts[0], INIT_BAL, { from: owner }), rep.approve(TokenTransferProxy.address, INIT_ALLOW, { from: accounts[1], }), rep.setBalance(accounts[1], INIT_BAL, { from: owner }), ]); }); describe('transferFrom', () => { it('should throw when called by an unauthorized address', async () => { expect( tokenTransferProxy.transferFrom(rep.address, accounts[0], accounts[1], 1000, { from: notAuthorized }), ).to.be.rejectedWith(constants.REVERT); }); it('should allow an authorized address to transfer', async () => { const balances = await dmyBalances.getAsync(); await tokenTransferProxy.addAuthorizedAddress(notAuthorized, { from: owner, }); const transferAmt = 10000; await tokenTransferProxy.transferFrom(rep.address, accounts[0], accounts[1], transferAmt, { from: notAuthorized, }); const newBalances = await dmyBalances.getAsync(); expect(newBalances[accounts[0]][rep.address]).to.be.bignumber.equal( balances[accounts[0]][rep.address].minus(transferAmt), ); expect(newBalances[accounts[1]][rep.address]).to.be.bignumber.equal( balances[accounts[1]][rep.address].add(transferAmt), ); }); }); });