1c220e175d
* Add withdrawWithGas
* Improve docs
* Add changelog entry
* Update contracts/payment/PullPayment.sol
Co-Authored-By: Francisco Giordano <frangio.1@gmail.com>
* Remove repeated comment
* Update changelog entry
* Fix inline docs
* Fix changelog formatting
(cherry picked from commit d6e10ab786)
58 lines
2.1 KiB
JavaScript
58 lines
2.1 KiB
JavaScript
const { balance, ether } = require('openzeppelin-test-helpers');
|
|
|
|
const { expect } = require('chai');
|
|
|
|
const PullPaymentMock = artifacts.require('PullPaymentMock');
|
|
|
|
contract('PullPayment', function ([_, payer, payee1, payee2]) {
|
|
const amount = ether('17');
|
|
|
|
beforeEach(async function () {
|
|
this.contract = await PullPaymentMock.new({ value: amount });
|
|
});
|
|
|
|
it('can record an async payment correctly', async function () {
|
|
await this.contract.callTransfer(payee1, 100, { from: payer });
|
|
expect(await this.contract.payments(payee1)).to.be.bignumber.equal('100');
|
|
});
|
|
|
|
it('can add multiple balances on one account', async function () {
|
|
await this.contract.callTransfer(payee1, 200, { from: payer });
|
|
await this.contract.callTransfer(payee1, 300, { from: payer });
|
|
expect(await this.contract.payments(payee1)).to.be.bignumber.equal('500');
|
|
});
|
|
|
|
it('can add balances on multiple accounts', async function () {
|
|
await this.contract.callTransfer(payee1, 200, { from: payer });
|
|
await this.contract.callTransfer(payee2, 300, { from: payer });
|
|
|
|
expect(await this.contract.payments(payee1)).to.be.bignumber.equal('200');
|
|
|
|
expect(await this.contract.payments(payee2)).to.be.bignumber.equal('300');
|
|
});
|
|
|
|
it('can withdraw payment', async function () {
|
|
const balanceTracker = await balance.tracker(payee1);
|
|
|
|
await this.contract.callTransfer(payee1, amount, { from: payer });
|
|
(await this.contract.payments(payee1)).should.be.bignumber.equal(amount);
|
|
|
|
await this.contract.withdrawPayments(payee1);
|
|
|
|
(await balanceTracker.delta()).should.be.bignumber.equal(amount);
|
|
(await this.contract.payments(payee1)).should.be.bignumber.equal('0');
|
|
});
|
|
|
|
it('can withdraw payment forwarding all gas', async function () {
|
|
const balanceTracker = await balance.tracker(payee1);
|
|
|
|
await this.contract.callTransfer(payee1, amount, { from: payer });
|
|
(await this.contract.payments(payee1)).should.be.bignumber.equal(amount);
|
|
|
|
await this.contract.withdrawPaymentsWithGas(payee1);
|
|
|
|
(await balanceTracker.delta()).should.be.bignumber.equal(amount);
|
|
(await this.contract.payments(payee1)).should.be.bignumber.equal('0');
|
|
});
|
|
});
|