Software / code / prosody
Changeset
7024:8ce592e376ff
tests: Add small test for util.throttle
| author | Kim Alvefur <zash@zash.se> |
|---|---|
| date | Wed, 23 Dec 2015 08:49:38 +0100 |
| parents | 7023:c2ccbfe30113 |
| children | 7025:236e8d1ee96c |
| files | tests/test.lua tests/test_util_throttle.lua |
| diffstat | 2 files changed, 35 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- 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");
--- /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 +
