You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.8 KiB
88 lines
2.8 KiB
|
|
import latestTime from './helpers/latestTime';
|
|
import { increaseTimeTo, duration } from './helpers/increaseTime';
|
|
|
|
import assertRevert from './helpers/assertRevert';
|
|
|
|
const DayLimitMock = artifacts.require('DayLimitMock');
|
|
|
|
contract('DayLimit', function (accounts) {
|
|
let dayLimit;
|
|
let initLimit = 10;
|
|
|
|
beforeEach(async function () {
|
|
this.startTime = latestTime();
|
|
dayLimit = await DayLimitMock.new(initLimit);
|
|
});
|
|
|
|
it('should construct with the passed daily limit', async function () {
|
|
let dailyLimit = await dayLimit.dailyLimit();
|
|
assert.equal(initLimit, dailyLimit);
|
|
});
|
|
|
|
it('should be able to spend if daily limit is not reached', async function () {
|
|
await dayLimit.attemptSpend(8);
|
|
let spentToday = await dayLimit.spentToday();
|
|
assert.equal(spentToday, 8);
|
|
|
|
await dayLimit.attemptSpend(2);
|
|
spentToday = await dayLimit.spentToday();
|
|
assert.equal(spentToday, 10);
|
|
});
|
|
|
|
it('should prevent spending if daily limit is reached', async function () {
|
|
await dayLimit.attemptSpend(8);
|
|
let spentToday = await dayLimit.spentToday();
|
|
assert.equal(spentToday, 8);
|
|
await assertRevert(dayLimit.attemptSpend(3));
|
|
});
|
|
|
|
it('should allow spending if daily limit is reached and then set higher', async function () {
|
|
await dayLimit.attemptSpend(8);
|
|
let spentToday = await dayLimit.spentToday();
|
|
assert.equal(spentToday, 8);
|
|
|
|
await assertRevert(dayLimit.attemptSpend(3));
|
|
spentToday = await dayLimit.spentToday();
|
|
assert.equal(spentToday, 8);
|
|
|
|
await dayLimit.setDailyLimit(15);
|
|
await dayLimit.attemptSpend(3);
|
|
spentToday = await dayLimit.spentToday();
|
|
assert.equal(spentToday, 11);
|
|
});
|
|
|
|
it('should allow spending if daily limit is reached and then amount spent is reset', async function () {
|
|
await dayLimit.attemptSpend(8);
|
|
let spentToday = await dayLimit.spentToday();
|
|
assert.equal(spentToday, 8);
|
|
|
|
await assertRevert(dayLimit.attemptSpend(3));
|
|
spentToday = await dayLimit.spentToday();
|
|
assert.equal(spentToday, 8);
|
|
|
|
await dayLimit.resetSpentToday();
|
|
await dayLimit.attemptSpend(3);
|
|
spentToday = await dayLimit.spentToday();
|
|
assert.equal(spentToday, 3);
|
|
});
|
|
|
|
it('should allow spending if daily limit is reached and then the next has come', async function () {
|
|
let limit = 10;
|
|
let dayLimit = await DayLimitMock.new(limit);
|
|
|
|
await dayLimit.attemptSpend(8);
|
|
let spentToday = await dayLimit.spentToday();
|
|
assert.equal(spentToday, 8);
|
|
|
|
await assertRevert(dayLimit.attemptSpend(3));
|
|
spentToday = await dayLimit.spentToday();
|
|
assert.equal(spentToday, 8);
|
|
|
|
await increaseTimeTo(this.startTime + duration.days(1));
|
|
|
|
await dayLimit.attemptSpend(3);
|
|
spentToday = await dayLimit.spentToday();
|
|
assert.equal(spentToday, 3);
|
|
});
|
|
});
|
|
|