add more limit tests to LimitBalance

This commit is contained in:
Manuel Araoz
2016-11-18 12:33:52 -03:00
parent 44138018dc
commit 4cd8aa7900
2 changed files with 27 additions and 1 deletions

View File

@ -8,7 +8,7 @@ contract LimitBalance {
}
modifier limitedPayable() {
if (this.balance + msg.value > limit) {
if (this.balance > limit) {
throw;
}
_;

View File

@ -35,4 +35,30 @@ contract('LimitBalance', function(accounts) {
.then(done)
});
it("should allow multiple sends below limit", function(done) {
var amount = 500;
return lb.limitedDeposit({value: amount})
.then(function() {
assert.equal(web3.eth.getBalance(lb.address), amount);
return lb.limitedDeposit({value: amount})
})
.then(function() {
assert.equal(web3.eth.getBalance(lb.address), amount*2);
})
.then(done)
});
it("shouldnt allow multiple sends above limit", function(done) {
var amount = 500;
return lb.limitedDeposit({value: amount})
.then(function() {
assert.equal(web3.eth.getBalance(lb.address), amount);
return lb.limitedDeposit({value: amount+1})
})
.catch(function(error) {
if (error.message.search('invalid JUMP') == -1) throw error;
})
.then(done)
});
});