aboutsummaryrefslogtreecommitdiffstats
path: root/tests/test_util_throttle.lua
diff options
context:
space:
mode:
authorKim Alvefur <zash@zash.se>2015-12-27 12:29:28 +0100
committerKim Alvefur <zash@zash.se>2015-12-27 12:29:28 +0100
commite1b247f298550899a261e9188784154dec7f4c15 (patch)
tree50c569b3a43e2bf21301dba4606a02940a6ed127 /tests/test_util_throttle.lua
parent9cf26570d1ed58175107e606a044506e5210f501 (diff)
parent6e6e46038c6d65b1f00a4c532cb4b90da007807f (diff)
downloadprosody-e1b247f298550899a261e9188784154dec7f4c15.tar.gz
prosody-e1b247f298550899a261e9188784154dec7f4c15.zip
Merge 0.10->trunk
Diffstat (limited to 'tests/test_util_throttle.lua')
-rw-r--r--tests/test_util_throttle.lua34
1 files changed, 34 insertions, 0 deletions
diff --git a/tests/test_util_throttle.lua b/tests/test_util_throttle.lua
new file mode 100644
index 00000000..582f499d
--- /dev/null
+++ b/tests/test_util_throttle.lua
@@ -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
+