File

tests/test_util_throttle.lua @ 7567:495de404a8ae

ejabberdsql2prosody: rename variable 'host' to prevent shadowing upvalue [luacheck] Functions roster(), roster_pending(), roster_group(), private_storage() and offline_msg() have argument named "host", which used to shadow upvalue of this variable before this change. Instead of renaming this argument, let's rename the variable to match what the script says in usage: Usage: ejabberdsql2prosody.lua filename.txt hostname
author Anton Shestakov <av6@dwimlabs.net>
date Fri, 12 Aug 2016 13:44:47 +0800
parent 7024:8ce592e376ff
child 8262:e6f3e440c843
line wrap: on
line source


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