const assert = require('assert') const PendingBalanceCalculator = require('../../app/scripts/lib/pending-balance-calculator') const MockTxGen = require('../lib/mock-tx-gen') const BN = require('ethereumjs-util').BN let providerResultStub = {} const zeroBn = new BN(0) const etherBn = new BN(String(1e18)) const ether = '0x' + etherBn.toString(16) describe('PendingBalanceCalculator', function () { let balanceCalculator describe('#valueFor(tx)', function () { it('returns a BN for a given tx value', function () { const txGen = new MockTxGen() pendingTxs = txGen.generate({ status: 'submitted', txParams: { value: ether, gasPrice: '0x0', gas: '0x0', } }, { count: 1 }) const balanceCalculator = generateBalanceCalcWith([], zeroBn) const result = balanceCalculator.valueFor(pendingTxs[0]) assert.equal(result.toString(), etherBn.toString(), 'computes one ether') }) }) describe('if you have no pending txs and one ether', function () { beforeEach(function () { balanceCalculator = generateBalanceCalcWith([], zeroBn) }) it('returns the network balance', async function () { const result = await balanceCalculator.getBalance() assert.equal(result, ether, `gave ${result} needed ${ether}`) }) }) describe('if you have a one ether pending tx and one ether', function () { beforeEach(function () { const txGen = new MockTxGen() pendingTxs = txGen.generate({ status: 'submitted', txParams: { value: ether, gasPrice: '0x0', gas: '0x0', } }, { count: 1 }) balanceCalculator = generateBalanceCalcWith(pendingTxs, etherBn) }) it('returns the network balance', async function () { console.log('one') console.dir(balanceCalculator) console.dir(balanceCalculator.getBalance.toString()) const result = await balanceCalculator.getBalance() console.log('two') console.dir(result) assert.equal(result, '0x0', `gave ${result} needed '0x0'`) return true }) }) }) function generateBalanceCalcWith (transactions, providerStub = zeroBn) { const getPendingTransactions = () => Promise.resolve(transactions) const getBalance = () => Promise.resolve(providerStub) providerResultStub.result = providerStub const provider = { sendAsync: (_, cb) => { cb(undefined, providerResultStub) }, _blockTracker: { getCurrentBlock: () => '0x11b568', }, } return new PendingBalanceCalculator({ getBalance, getPendingTransactions, }) }