# HG changeset patch
# User Kim Alvefur <zash@zash.se>
# Date 1450856978 -3600
# Node ID 8ce592e376ff760a70d26e4b56486094e9ffa9d3
# Parent  c2ccbfe301130e823e305627ba781e4cb2559e5c
tests: Add small test for util.throttle

diff -r c2ccbfe30113 -r 8ce592e376ff tests/test.lua
--- a/tests/test.lua	Wed Dec 23 08:47:57 2015 +0100
+++ b/tests/test.lua	Wed Dec 23 08:49:38 2015 +0100
@@ -21,6 +21,7 @@
 	dotest "util.stanza"
 	dotest "util.sasl.scram"
 	dotest "util.cache"
+	dotest "util.throttle"
 
 	dosingletest("test_sasl.lua", "latin1toutf8");
 	dosingletest("test_utf8.lua", "valid");
diff -r c2ccbfe30113 -r 8ce592e376ff tests/test_util_throttle.lua
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/test_util_throttle.lua	Wed Dec 23 08:49:38 2015 +0100
@@ -0,0 +1,34 @@
+
+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
+