commit
e033d802ff
@ -0,0 +1,11 @@ |
|||||||
|
pragma solidity ^0.4.0; |
||||||
|
import '../StandardToken.sol'; |
||||||
|
|
||||||
|
// mock class using StandardToken |
||||||
|
contract StandardTokenMock is StandardToken { |
||||||
|
|
||||||
|
function StandardTokenMock(address initialAccount, uint initialBalance) { |
||||||
|
balances[initialAccount] = initialBalance; |
||||||
|
} |
||||||
|
|
||||||
|
} |
@ -0,0 +1,22 @@ |
|||||||
|
pragma solidity ^0.4.0; |
||||||
|
import '../Stoppable.sol'; |
||||||
|
|
||||||
|
// mock class using Stoppable |
||||||
|
contract StoppableMock is Stoppable(msg.sender) { |
||||||
|
bool public drasticMeasureTaken; |
||||||
|
uint public count; |
||||||
|
|
||||||
|
function StoppableMock() Stoppable(msg.sender){ |
||||||
|
drasticMeasureTaken = false; |
||||||
|
count = 0; |
||||||
|
} |
||||||
|
|
||||||
|
function normalProcess() external stopInEmergency { |
||||||
|
count++; |
||||||
|
} |
||||||
|
|
||||||
|
function drasticMeasure() external onlyInEmergency { |
||||||
|
drasticMeasureTaken = true; |
||||||
|
} |
||||||
|
|
||||||
|
} |
@ -0,0 +1,81 @@ |
|||||||
|
contract('StandardToken', function(accounts) { |
||||||
|
|
||||||
|
it("should return the correct allowance amount after approval", function(done) { |
||||||
|
var token; |
||||||
|
return StandardTokenMock.new() |
||||||
|
.then(function(_token) { |
||||||
|
token = _token; |
||||||
|
return token.approve(accounts[1], 100); |
||||||
|
}) |
||||||
|
.then(function() { |
||||||
|
return token.allowance(accounts[0], accounts[1]); |
||||||
|
}) |
||||||
|
.then(function(allowance) { |
||||||
|
assert.equal(allowance, 100); |
||||||
|
}) |
||||||
|
.then(done); |
||||||
|
}); |
||||||
|
|
||||||
|
it("should return correct balances after transfer", function(done) { |
||||||
|
var token; |
||||||
|
return StandardTokenMock.new(accounts[0], 100) |
||||||
|
.then(function(_token) { |
||||||
|
token = _token; |
||||||
|
return token.transfer(accounts[1], 100); |
||||||
|
}) |
||||||
|
.then(function() { |
||||||
|
return token.balanceOf(accounts[0]); |
||||||
|
}) |
||||||
|
.then(function(balance) { |
||||||
|
assert.equal(balance, 0); |
||||||
|
}) |
||||||
|
.then(function() { |
||||||
|
return token.balanceOf(accounts[1]); |
||||||
|
}) |
||||||
|
.then(function(balance) { |
||||||
|
assert.equal(balance, 100); |
||||||
|
}) |
||||||
|
.then(done); |
||||||
|
}); |
||||||
|
|
||||||
|
it("should throw an error when trying to transfer more than balance", function(done) { |
||||||
|
var token; |
||||||
|
return StandardTokenMock.new(accounts[0], 100) |
||||||
|
.then(function(_token) { |
||||||
|
token = _token; |
||||||
|
return token.transfer(accounts[1], 101); |
||||||
|
}) |
||||||
|
.catch(function(error) { |
||||||
|
if (error.message.search('invalid JUMP') == -1) throw error |
||||||
|
}) |
||||||
|
.then(done); |
||||||
|
}); |
||||||
|
|
||||||
|
it("should return correct balances after transfering from another account", function(done) { |
||||||
|
var token; |
||||||
|
return StandardTokenMock.new(accounts[0], 100) |
||||||
|
.then(function(_token) { |
||||||
|
token = _token; |
||||||
|
return token.approve(accounts[1], 100); |
||||||
|
}) |
||||||
|
.then(function() { |
||||||
|
return token.transferFrom(accounts[0], accounts[2], 100, {from: accounts[1]}); |
||||||
|
}) |
||||||
|
.then(function() { |
||||||
|
return token.balanceOf(accounts[0]); |
||||||
|
}) |
||||||
|
.then(function(balance) { |
||||||
|
assert.equal(balance, 0); |
||||||
|
return token.balanceOf(accounts[2]); |
||||||
|
}) |
||||||
|
.then(function(balance) { |
||||||
|
assert.equal(balance, 100) |
||||||
|
return token.balanceOf(accounts[1]); |
||||||
|
}) |
||||||
|
.then(function(balance) { |
||||||
|
assert.equal(balance, 0); |
||||||
|
}) |
||||||
|
.then(done); |
||||||
|
}); |
||||||
|
|
||||||
|
}); |
@ -0,0 +1,108 @@ |
|||||||
|
contract('Stoppable', function(accounts) { |
||||||
|
|
||||||
|
it("can perform normal process in non-emergency", function(done) { |
||||||
|
var stoppable; |
||||||
|
return StoppableMock.new(accounts[0]) |
||||||
|
.then(function(_stoppable) { |
||||||
|
stoppable = _stoppable; |
||||||
|
return stoppable.count(); |
||||||
|
}) |
||||||
|
.then(function(count) { |
||||||
|
assert.equal(count, 0); |
||||||
|
}) |
||||||
|
.then(function () { |
||||||
|
return stoppable.normalProcess(); |
||||||
|
}) |
||||||
|
.then(function() { |
||||||
|
return stoppable.count(); |
||||||
|
}) |
||||||
|
.then(function(count) { |
||||||
|
assert.equal(count, 1); |
||||||
|
}) |
||||||
|
.then(done); |
||||||
|
}); |
||||||
|
|
||||||
|
it("can not perform normal process in emergency", function(done) { |
||||||
|
var stoppable; |
||||||
|
return StoppableMock.new(accounts[0]) |
||||||
|
.then(function(_stoppable) { |
||||||
|
stoppable = _stoppable; |
||||||
|
return stoppable.emergencyStop(); |
||||||
|
}) |
||||||
|
.then(function () { |
||||||
|
return stoppable.count(); |
||||||
|
}) |
||||||
|
.then(function(count) { |
||||||
|
assert.equal(count, 0); |
||||||
|
}) |
||||||
|
.then(function () { |
||||||
|
return stoppable.normalProcess(); |
||||||
|
}) |
||||||
|
.then(function() { |
||||||
|
return stoppable.count(); |
||||||
|
}) |
||||||
|
.then(function(count) { |
||||||
|
assert.equal(count, 0); |
||||||
|
}) |
||||||
|
.then(done); |
||||||
|
}); |
||||||
|
|
||||||
|
|
||||||
|
it("can not take drastic measure in non-emergency", function(done) { |
||||||
|
var stoppable; |
||||||
|
return StoppableMock.new(accounts[0]) |
||||||
|
.then(function(_stoppable) { |
||||||
|
stoppable = _stoppable; |
||||||
|
return stoppable.drasticMeasure(); |
||||||
|
}) |
||||||
|
.then(function() { |
||||||
|
return stoppable.drasticMeasureTaken(); |
||||||
|
}) |
||||||
|
.then(function(taken) { |
||||||
|
assert.isFalse(taken); |
||||||
|
}) |
||||||
|
.then(done); |
||||||
|
}); |
||||||
|
|
||||||
|
it("can take a drastic measure in an emergency", function(done) { |
||||||
|
var stoppable; |
||||||
|
return StoppableMock.new(accounts[0]) |
||||||
|
.then(function(_stoppable) { |
||||||
|
stoppable = _stoppable; |
||||||
|
return stoppable.emergencyStop(); |
||||||
|
}) |
||||||
|
.then(function() { |
||||||
|
return stoppable.drasticMeasure(); |
||||||
|
}) |
||||||
|
.then(function() { |
||||||
|
return stoppable.drasticMeasureTaken(); |
||||||
|
}) |
||||||
|
.then(function(taken) { |
||||||
|
assert.isTrue(taken); |
||||||
|
}) |
||||||
|
.then(done); |
||||||
|
}); |
||||||
|
|
||||||
|
it("should resume allowing normal process after emergency is over", function(done) { |
||||||
|
var stoppable; |
||||||
|
return StoppableMock.new(accounts[0]) |
||||||
|
.then(function(_stoppable) { |
||||||
|
stoppable = _stoppable; |
||||||
|
return stoppable.emergencyStop(); |
||||||
|
}) |
||||||
|
.then(function () { |
||||||
|
return stoppable.release(); |
||||||
|
}) |
||||||
|
.then(function() { |
||||||
|
return stoppable.normalProcess(); |
||||||
|
}) |
||||||
|
.then(function() { |
||||||
|
return stoppable.count(); |
||||||
|
}) |
||||||
|
.then(function(count) { |
||||||
|
assert.equal(count, 1); |
||||||
|
}) |
||||||
|
.then(done); |
||||||
|
}); |
||||||
|
|
||||||
|
}); |
Loading…
Reference in new issue