IMPORTANT: due to a drive failure, as of 13-Mar-2021, the Mercurial repository had to be re-mirrored, which changed every commit SHA. The old SHAs and trees are backed up in the vault branches. Please migrate to the new branches as soon as you can.
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.
 
 
 
 
prosody/tests/test_util_throttle.lua

34 lines
934 B

local now = 0; -- wibbly-wobbly... timey-wimey... stuff
local function predictable_gettime()
return now;
end
local function later(n)
now = now + n; -- time passes at a different rate
end
local function override_gettime(throttle)
local i = 0;
repeat
i = i + 1;
local name = debug.getupvalue(throttle.update, i);
if name then
debug.setupvalue(throttle.update, i, predictable_gettime);
return throttle;
end
until not name;
end
function create(create)
local a = override_gettime( create(3, 10) );
assert_equal(a:poll(1), true); -- 3 -> 2
assert_equal(a:poll(1), true); -- 2 -> 1
assert_equal(a:poll(1), true); -- 1 -> 0
assert_equal(a:poll(1), false); -- MEEP, out of credits!
later(1); -- ... what about
assert_equal(a:poll(1), false); -- now? - Still no!
later(9); -- Later that day
assert_equal(a:poll(1), true); -- Should be back at 3 credits ... 2
end