openzeppelin-contracts/test/PullPayment.js

var PullPaymentMock = artifacts.require("./helpers/PullPaymentMock.sol");

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;
    let callSend = 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() {
    let call1 = await ppce.callSend(accounts[0], 200);
    let call2 = 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() {
    let call1 = await ppce.callSend(accounts[0], 200);
    let call2 = 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);

    let call1 = await ppce.callSend(payee, amount);

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

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

    let withdraw = 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);
  });

});