aboutsummaryrefslogtreecommitdiffstats
path: root/spec/util_throttle_spec.lua
blob: a588e262a664aad85a113e53937eed6d0369d187 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30


-- Mock util.time
local now = 0; -- wibbly-wobbly... timey-wimey... stuff
local function later(n)
	now = now + n; -- time passes at a different rate
end
package.loaded["util.time"] = {
	now = function() return now; end
}


local throttle = require "util.throttle";

describe("util.sasl.scram", function()
	describe("#Hi()", function()
		it("should work", function()
			local a = throttle.create(3, 10);

			assert.are.equal(a:poll(1), true);  -- 3 -> 2
			assert.are.equal(a:poll(1), true);  -- 2 -> 1
			assert.are.equal(a:poll(1), true);  -- 1 -> 0
			assert.are.equal(a:poll(1), false); -- MEEP, out of credits!
			later(1);                       -- ... what about
			assert.are.equal(a:poll(1), false); -- now? - Still no!
			later(9);                       -- Later that day
			assert.are.equal(a:poll(1), true);  -- Should be back at 3 credits ... 2
		end);
	end);
end);