NoBey/openzeppelin-contracts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor time manipulation in TokenTimelock tests

Browse Source
This commit is contained in:
Jakub Wojciechowski
2017-08-13 12:48:31 +02:00
parent 6e66ba321e
commit 99e0f5b5cb
2 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
test/TokenTimelock.js
View File

@ -5,10 +5,9 @@ require('chai')
.use(require('chai-bignumber')(BigNumber)) .use(require('chai-bignumber')(BigNumber))
.should() .should()
import moment from 'moment'
import latestTime from './helpers/latestTime' import latestTime from './helpers/latestTime'
import increaseTime from './helpers/increaseTime' import {increaseTimeTo, duration} from './helpers/increaseTime'
const MintableToken = artifacts.require('MintableToken') const MintableToken = artifacts.require('MintableToken')
const TokenTimelock = artifacts.require('TokenTimelock') const TokenTimelock = artifacts.require('TokenTimelock')
@ -19,7 +18,7 @@ contract('TokenTimelock', function ([_, owner, beneficiary]) {
beforeEach(async function () { beforeEach(async function () {
this.token = await MintableToken.new({from: owner}) this.token = await MintableToken.new({from: owner})
this.releaseTime = latestTime().add(1, 'year').unix() this.releaseTime = latestTime().unix() + duration.years(1)
this.timelock = await TokenTimelock.new(this.token.address, beneficiary, this.releaseTime) this.timelock = await TokenTimelock.new(this.token.address, beneficiary, this.releaseTime)
await this.token.mint(this.timelock.address, amount, {from: owner}) await this.token.mint(this.timelock.address, amount, {from: owner})
}) })
@ -29,26 +28,26 @@ contract('TokenTimelock', function ([_, owner, beneficiary]) {
}) })
it('cannot be released just before time limit', async function () { it('cannot be released just before time limit', async function () {
await increaseTime(moment.duration(0.99, 'year').asSeconds()) await increaseTimeTo(this.releaseTime - duration.seconds(3))
await this.timelock.release().should.be.rejected await this.timelock.release().should.be.rejected
}) })
it('can be released just after limit', async function () { it('can be released just after limit', async function () {
await increaseTime(moment.duration(1.01, 'year').asSeconds()) await increaseTimeTo(this.releaseTime + duration.seconds(1))
await this.timelock.release().should.be.fulfilled await this.timelock.release().should.be.fulfilled
const balance = await this.token.balanceOf(beneficiary) const balance = await this.token.balanceOf(beneficiary)
balance.should.be.bignumber.equal(amount) balance.should.be.bignumber.equal(amount)
}) })
it('can be released after time limit', async function () { it('can be released after time limit', async function () {
await increaseTime(moment.duration(2, 'year').asSeconds()) await increaseTimeTo(this.releaseTime + duration.years(1))
await this.timelock.release().should.be.fulfilled await this.timelock.release().should.be.fulfilled
const balance = await this.token.balanceOf(beneficiary) const balance = await this.token.balanceOf(beneficiary)
balance.should.be.bignumber.equal(amount) balance.should.be.bignumber.equal(amount)
}) })
it('cannot be released twice', async function () { it('cannot be released twice', async function () {
await increaseTime(moment.duration(2, 'year').asSeconds()) await increaseTimeTo(this.releaseTime + duration.years(1))
await this.timelock.release().should.be.fulfilled await this.timelock.release().should.be.fulfilled
await this.timelock.release().should.be.rejected await this.timelock.release().should.be.rejected
const balance = await this.token.balanceOf(beneficiary) const balance = await this.token.balanceOf(beneficiary)

4
test/helpers/increaseTime.js
View File

@ -43,5 +43,7 @@ export const duration = {
minutes: function(val) { return val * this.seconds(60) }, minutes: function(val) { return val * this.seconds(60) },
hours: function(val) { return val * this.minutes(60) }, hours: function(val) { return val * this.minutes(60) },
days: function(val) { return val * this.hours(24) }, days: function(val) { return val * this.hours(24) },
weeks: function(val) { return val * this.days(7) } weeks: function(val) { return val * this.days(7) },
months: function(val) { return val * this.days(30)},
years: function(val) { return val * this.days(365)}
}; };