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

Improve VestedToken tests

Browse Source
This commit is contained in:
Jorge Izquierdo
2017-04-05 13:20:18 +02:00
parent 41d2fde952
commit 7a2fda9076
1 changed files with 21 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
test/VestedToken.js
View File

@ -40,7 +40,7 @@ contract('VestedToken', function(accounts) {
}) })
it('all tokens are transferable after vesting', async () => { it('all tokens are transferable after vesting', async () => {
assert.equal(await token.transferableTokens(receiver, now + vesting + 1), tokenAmount); assert.equal(await token.transferableTokens(receiver, now + vesting), tokenAmount);
}) })
it('throws when trying to transfer non vested tokens', async () => { it('throws when trying to transfer non vested tokens', async () => {
@ -84,16 +84,34 @@ contract('VestedToken', function(accounts) {
}) })
it('can transfer all tokens after vesting ends', async () => { it('can transfer all tokens after vesting ends', async () => {
await timer(vesting + 1); await timer(vesting);
await token.transfer(accounts[7], tokenAmount, { from: receiver }) await token.transfer(accounts[7], tokenAmount, { from: receiver })
assert.equal(await token.balanceOf(accounts[7]), tokenAmount); assert.equal(await token.balanceOf(accounts[7]), tokenAmount);
}) })
it('can approve and transferFrom all tokens after vesting ends', async () => { it('can approve and transferFrom all tokens after vesting ends', async () => {
await timer(vesting + 1); await timer(vesting);
await token.approve(accounts[7], tokenAmount, { from: receiver }) await token.approve(accounts[7], tokenAmount, { from: receiver })
await token.transferFrom(receiver, accounts[7], tokenAmount, { from: accounts[7] }) await token.transferFrom(receiver, accounts[7], tokenAmount, { from: accounts[7] })
assert.equal(await token.balanceOf(accounts[7]), tokenAmount); assert.equal(await token.balanceOf(accounts[7]), tokenAmount);
}) })
it('can handle composed vesting schedules', async () => {
await timer(cliff);
await token.transfer(accounts[7], 12, { from: receiver })
assert.equal(await token.balanceOf(accounts[7]), 12);
let newNow = web3.eth.getBlock(web3.eth.blockNumber).timestamp
await token.grantVestedTokens(receiver, tokenAmount, newNow, newNow + cliff, newNow + vesting, { from: granter })
await token.transfer(accounts[7], 13, { from: receiver })
assert.equal(await token.balanceOf(accounts[7]), tokenAmount / 2);
assert.equal(await token.balanceOf(receiver), 3 * tokenAmount / 2)
assert.equal(await token.transferableTokens(receiver, newNow), 0)
await timer(vesting);
await token.transfer(accounts[7], 3 * tokenAmount / 2, { from: receiver })
assert.equal(await token.balanceOf(accounts[7]), tokenAmount * 2)
})
}) })
}); });