3 files changed, 303 insertions, 0 deletions
diff --git a/tests/test.lua b/tests/test.lua
index bc33bb76..ab029cc4 100644
--- a/tests/test.lua
+++ b/tests/test.lua
@@ -21,6 +21,7 @@ function run_all_tests()
 	dotest "util.json"
 	dotest "util.stanza"
 	dotest "util.sasl.scram"
+	dotest "util.async"
 	dotest "util.cache"
 	dotest "util.throttle"
 	dotest "util.uuid"
diff --git a/tests/test_util_async.lua b/tests/test_util_async.lua
new file mode 100644
index 00000000..1a74e79c
--- /dev/null
+++ b/tests/test_util_async.lua
@@ -0,0 +1,301 @@
+
+-- Test passing nil to runner
+-- Test runners work correctly after errors (coroutine gets recreated)
+-- What happens if an error is thrown, but more items are in the queue? (I think runner might stall)
+-- Test errors thrown halfway through a queue
+-- Multiple runners
+
+function runner(new_runner, async)
+	local function new(func, name)
+		local log = {};
+		return new_runner(func, setmetatable({}, {
+			__index = function (_, event)
+				return function (runner, err)
+					print(name, "event", event, err)
+					table.insert(log, { event = event, err = err });
+				end;
+			end;
+		})), log;
+	end
+	
+	--------------------
+	local r, l = new(function (item) assert(type(item) == "number") end);
+	r:run(1);
+	r:run(2);
+	--for k, v in ipairs(l) do print(k,v) end
+
+	--------------------
+	local wait, done;
+
+	local r, l = new(function (item)
+		assert(type(item) == "number")
+		if item == 3 then
+			wait, done = async.waiter();
+			wait();
+		end
+	end);
+	
+	r:run(1);
+	assert(r.state == "ready");
+	r:run(2);
+	assert(r.state == "ready");
+	r:run(3);
+	assert(r.state == "waiting");
+	done();
+	assert(r.state == "ready");
+	--for k, v in ipairs(l) do print(k,v) end
+
+	--------------------
+	local wait, done;
+	local last_item = 0;
+	local r, l = new(function (item)
+		assert(type(item) == "number")
+		assert(item == last_item + 1);
+		last_item = item;
+		if item == 3 then
+			wait, done = async.waiter();
+			wait();
+		end
+	end);
+	
+	r:run(1);
+	assert(r.state == "ready");
+	r:run(2);
+	assert(r.state == "ready");
+	r:run(3);
+	assert(r.state == "waiting");
+	r:run(4);
+	assert(r.state == "waiting");
+	done();
+	assert(r.state == "ready");
+	--for k, v in ipairs(l) do print(k,v) end
+
+	--------------------
+	local wait, done;
+	local last_item = 0;
+	local r, l = new(function (item)
+		assert(type(item) == "number")
+		assert((item == last_item + 1) or item == 3);
+		last_item = item;
+		if item == 3 then
+			wait, done = async.waiter();
+			wait();
+		end
+	end);
+	
+	r:run(1);
+	assert(r.state == "ready");
+	r:run(2);
+	assert(r.state == "ready");
+	
+	local dones = {};
+	r:run(3);
+	assert(r.state == "waiting");
+	r:run(3);
+	assert(r.state == "waiting");
+	r:run(3);
+	assert(r.state == "waiting");
+	r:run(4);
+	assert(r.state == "waiting");
+
+	for i = 1, 3 do
+		done();
+		if i < 3 then
+			assert(r.state == "waiting");
+		end
+	end
+
+	assert(r.state == "ready");
+	--for k, v in ipairs(l) do print(k,v) end
+
+	--------------------
+	local wait, done;
+	local last_item = 0;
+	local r, l = new(function (item)
+		assert(type(item) == "number")
+		assert((item == last_item + 1) or item == 3);
+		last_item = item;
+		if item == 3 then
+			wait, done = async.waiter();
+			wait();
+		end
+	end);
+	
+	r:run(1);
+	assert(r.state == "ready");
+	r:run(2);
+	assert(r.state == "ready");
+	
+	local dones = {};
+	r:run(3);
+	assert(r.state == "waiting");
+	r:run(3);
+	assert(r.state == "waiting");
+
+	for i = 1, 2 do
+		done();
+		if i < 2 then
+			assert(r.state == "waiting");
+		end
+	end
+
+	assert(r.state == "ready");
+	r:run(4);
+	assert(r.state == "ready");
+
+	assert(r.state == "ready");
+	--for k, v in ipairs(l) do print(k,v) end
+
+	-- Now with multiple runners
+	--------------------
+	local wait1, done1;
+	local last_item1 = 0;
+	local r1, l1 = new(function (item)
+		assert(type(item) == "number")
+		assert((item == last_item1 + 1) or item == 3);
+		last_item1 = item;
+		if item == 3 then
+			wait1, done1 = async.waiter();
+			wait1();
+		end
+	end, "r1");
+
+	local wait2, done2;
+	local last_item2 = 0;
+	local r2, l2 = new(function (item)
+		assert(type(item) == "number")
+		assert((item == last_item2 + 1) or item == 3);
+		last_item2 = item;
+		if item == 3 then
+			wait2, done2 = async.waiter();
+			wait2();
+		end
+	end, "r2");
+	
+	r1:run(1);
+	assert(r1.state == "ready");
+	r1:run(2);
+	assert(r1.state == "ready");
+	
+	local dones = {};
+	r1:run(3);
+	assert(r1.state == "waiting");
+	r1:run(3);
+	assert(r1.state == "waiting");
+
+	r2:run(1);
+	assert(r1.state == "waiting");
+	assert(r2.state == "ready");
+
+	r2:run(2);
+	assert(r1.state == "waiting");
+	assert(r2.state == "ready");
+
+	r2:run(3);
+	assert(r1.state == "waiting");
+	assert(r2.state == "waiting");
+	done2();
+
+	r2:run(3);
+	assert(r1.state == "waiting");
+	assert(r2.state == "waiting");
+	done2();
+
+	r2:run(2);
+	assert(r1.state == "waiting");
+	assert(r2.state == "ready");
+
+	for i = 1, 2 do
+		done1();
+		if i < 2 then
+			assert(r1.state == "waiting");
+		end
+	end
+
+	assert(r1.state == "ready");
+	r1:run(4);
+	assert(r1.state == "ready");
+
+	assert(r1.state == "ready");
+	--for k, v in ipairs(l1) do print(k,v) end
+	
+
+	--------------------
+	local wait1, done1;
+	local last_item1 = 0;
+	local r1, l1 = new(function (item)
+		assert(type(item) == "number")
+		assert((item == last_item1 + 1) or item == 3);
+		last_item1 = item;
+		if item == 3 then
+			wait1, done1 = async.waiter();
+			wait1();
+		end
+	end, "r1");
+
+	local wait2, done2;
+	local last_item2 = 0;
+	local r2, l2 = new(function (item)
+		assert(type(item) == "number")
+		assert((item == last_item2 + 1) or item == 3);
+		last_item2 = item;
+		if item == 3 then
+			wait2, done2 = async.waiter();
+			wait2();
+		end
+	end, "r2");
+	
+	r1:run(1);
+	assert(r1.state == "ready");
+	r1:run(2);
+	assert(r1.state == "ready");
+	
+	r1:run(5);
+	assert(r1.state == "ready");
+
+	local dones = {};
+	r1:run(3);
+	assert(r1.state == "waiting");
+	r1:run(5); -- Will error, when we get to it
+	assert(r1.state == "waiting");
+	r1:run(3);
+	assert(r1.state == "waiting");
+
+	r2:run(1);
+	assert(r1.state == "waiting");
+	assert(r2.state == "ready");
+
+	r2:run(2);
+	assert(r1.state == "waiting");
+	assert(r2.state == "ready");
+
+	r2:run(3);
+	assert(r1.state == "waiting");
+	assert(r2.state == "waiting");
+	done2();
+
+	r2:run(3);
+	assert(r1.state == "waiting");
+	assert(r2.state == "waiting");
+	done2();
+
+	r2:run(2);
+	assert(r1.state == "waiting");
+	assert(r2.state == "ready");
+
+	for i = 1, 2 do
+		done1();
+		if i < 2 then
+			assert_equal(r1.state, "waiting");
+		end
+	end
+
+	assert(r1.state == "ready");
+	r1:run(4);
+	assert(r1.state == "ready");
+
+	assert(r1.state == "ready");
+	--for k, v in ipairs(l1) do print(k,v) end
+	
+	
+end
diff --git a/tests/test_util_xml.lua b/tests/test_util_xml.lua
index ba44da19..56d3f7b9 100644
--- a/tests/test_util_xml.lua
+++ b/tests/test_util_xml.lua
@@ -9,4 +9,5 @@ function parse(parse)
 ]]
 	local stanza = parse(x);
 	assert_equal(stanza.tags[2].attr.xmlns, "b");
+	assert_equal(stanza.tags[2].namespaces["a"], "b");
 end