openzeppelin-contracts/test/RefundVault.js

const BigNumber = web3.BigNumber

require('chai')
  .use(require('chai-as-promised'))
  .use(require('chai-bignumber')(BigNumber))
  .should()

import ether from './helpers/ether'
import EVMThrow from './helpers/EVMThrow'

const RefundVault = artifacts.require('RefundVault')

contract('RefundVault', function ([_, owner, wallet, investor]) {

  const value = ether(42)

  beforeEach(async function () {
    this.vault = await RefundVault.new(wallet, {from: owner})
  })

  it('should accept contributions', async function () {
    await this.vault.deposit(investor, {value, from: owner}).should.be.fulfilled
  })

  it('should not refund contribution during active state', async function () {
    await this.vault.deposit(investor, {value, from: owner})
    await this.vault.refund(investor).should.be.rejectedWith(EVMThrow)
  })

  it('only owner can enter refund mode', async function () {
    await this.vault.enableRefunds({from: _}).should.be.rejectedWith(EVMThrow)
    await this.vault.enableRefunds({from: owner}).should.be.fulfilled
  })

  it('should refund contribution after entering refund mode', async function () {
    await this.vault.deposit(investor, {value, from: owner})
    await this.vault.enableRefunds({from: owner})

    const pre = web3.eth.getBalance(investor)
    await this.vault.refund(investor)
    const post = web3.eth.getBalance(investor)

    post.minus(pre).should.be.bignumber.equal(value)
  })

  it('only owner can close', async function () {
    await this.vault.close({from: _}).should.be.rejectedWith(EVMThrow)
    await this.vault.close({from: owner}).should.be.fulfilled
  })

  it('should forward funds to wallet after closing', async function () {
    await this.vault.deposit(investor, {value, from: owner})

    const pre = web3.eth.getBalance(wallet)
    await this.vault.close({from: owner})
    const post = web3.eth.getBalance(wallet)

    post.minus(pre).should.be.bignumber.equal(value)
  })

})