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

Using modifier to check block on delayedClaimable, tests fixed

Browse Source
This commit is contained in:
AugustoL
2016-11-18 20:32:22 -03:00
parent 0aa4d02044
commit 85a4013f49
2 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
contracts/DelayedClaimable.sol
View File

@ -10,13 +10,16 @@ import './Claimable.sol';
contract DelayedClaimable is Ownable, Claimable {
uint public claimBefore;
modifier onTime() {
if (block.number < claimBefore)
_;
}
function setDelay(uint _claimBefore) onlyOwner {
claimBefore = _claimBefore;
}
function claimOwnership() onlyPendingOwner {
if (block.number > claimBefore)
throw;
function claimOwnership() onlyPendingOwner onTime {
owner = pendingOwner;
pendingOwner = 0x0;
claimBefore = 0;

8
test/DelayedClaimble.js
View File

@ -22,7 +22,9 @@ contract('DelayedClaimable', function(accounts) {
})
.then(function(pendingOwner) {
assert.isTrue(pendingOwner === newOwner);
delayedClaimable.claimOwnership({from: newOwner});
return delayedClaimable.claimOwnership({from: newOwner});
})
.then(function() {
return delayedClaimable.owner();
})
.then(function(owner) {
@ -46,7 +48,9 @@ contract('DelayedClaimable', function(accounts) {
})
.then(function(pendingOwner) {
assert.isTrue(pendingOwner === newOwner);
// delayedClaimable.claimOwnership({from: newOwner}); Uncomment to break the test.
return delayedClaimable.claimOwnership({from: newOwner});
})
.then(function() {
return delayedClaimable.owner();
})
.then(function(owner) {