openzeppelin-contracts/test/payment/PullPayment.test.js

var PullPaymentMock = artifacts.require('PullPaymentMock');

contract('PullPayment', function (accounts) {
  let ppce;
  let amount = 17 * 1e18;

  beforeEach(async function () {
    ppce = await PullPaymentMock.new({ value: amount });
  });

  it('can\'t call asyncSend externally', async function () {
    assert.isUndefined(ppce.asyncSend);
  });

  it('can record an async payment correctly', async function () {
    let AMOUNT = 100;
    await ppce.callSend(accounts[0], AMOUNT);
    let paymentsToAccount0 = await ppce.payments(accounts[0]);
    let totalPayments = await ppce.totalPayments();

    assert.equal(totalPayments, AMOUNT);
    assert.equal(paymentsToAccount0, AMOUNT);
  });

  it('can add multiple balances on one account', async function () {
    await ppce.callSend(accounts[0], 200);
    await ppce.callSend(accounts[0], 300);
    let paymentsToAccount0 = await ppce.payments(accounts[0]);
    let totalPayments = await ppce.totalPayments();

    assert.equal(totalPayments, 500);
    assert.equal(paymentsToAccount0, 500);
  });

  it('can add balances on multiple accounts', async function () {
    await ppce.callSend(accounts[0], 200);
    await ppce.callSend(accounts[1], 300);

    let paymentsToAccount0 = await ppce.payments(accounts[0]);
    assert.equal(paymentsToAccount0, 200);

    let paymentsToAccount1 = await ppce.payments(accounts[1]);
    assert.equal(paymentsToAccount1, 300);

    let totalPayments = await ppce.totalPayments();
    assert.equal(totalPayments, 500);
  });

  it('can withdraw payment', async function () {
    let payee = accounts[1];
    let initialBalance = web3.eth.getBalance(payee);

    await ppce.callSend(payee, amount);

    let payment1 = await ppce.payments(payee);
    assert.equal(payment1, amount);

    let totalPayments = await ppce.totalPayments();
    assert.equal(totalPayments, amount);

    await ppce.withdrawPayments({ from: payee });
    let payment2 = await ppce.payments(payee);
    assert.equal(payment2, 0);

    totalPayments = await ppce.totalPayments();
    assert.equal(totalPayments, 0);

    let balance = web3.eth.getBalance(payee);
    assert(Math.abs(balance - initialBalance - amount) < 1e16);
  });
});