diff options
author | Kim Alvefur <zash@zash.se> | 2015-12-27 12:29:28 +0100 |
---|---|---|
committer | Kim Alvefur <zash@zash.se> | 2015-12-27 12:29:28 +0100 |
commit | 0664d7ebf65a3ca356ec80ad4f706919ac745300 (patch) | |
tree | 50c569b3a43e2bf21301dba4606a02940a6ed127 /tests/test_util_throttle.lua | |
parent | b3d8c070f940572d53e8bea734a7abef38411c56 (diff) | |
parent | b50935a87c03246fac1541b275da4464a5b63b3b (diff) | |
download | prosody-0664d7ebf65a3ca356ec80ad4f706919ac745300.tar.gz prosody-0664d7ebf65a3ca356ec80ad4f706919ac745300.zip |
Merge 0.10->trunk
Diffstat (limited to 'tests/test_util_throttle.lua')
-rw-r--r-- | tests/test_util_throttle.lua | 34 |
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 + |