prosody
3a1efc0532cb

Find changesets by keywords by author, files, the commit message, revision number or hash, or revset expression.

Software / code / prosody

Changeset

8605:3a1efc0532cb

util.async: Add tests
author Matthew Wild <mwild1@gmail.com>
date Fri, 16 Mar 2018 17:13:57 +0000
parents 8604:1c8c7fd259c8
children 8606:b6b1f0f9b381
files spec/util_async_spec.lua
diffstat 1 files changed, 315 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/spec/util_async_spec.lua	Fri Mar 16 17:13:57 2018 +0000
@@ -0,0 +1,315 @@
+local async = require "util.async";
+
+describe("util.async", function()
+	local debug = false;
+	local print = print;
+	if debug then
+		require "util.logger".add_simple_sink(print);
+	else
+		print = function () end
+	end
+	local function new(func, name)
+		local log = {};
+		return async.runner(func, setmetatable({}, {
+			__index = function (_, event)
+				return function (runner, err)
+					print(name, "event", event, err)
+					print "--"
+					table.insert(log, { event = event, err = err });
+				end;
+			end;
+		})), log;
+	end
+	describe("#runner", function()
+		it("should work", function()			
+			local r, l = new(function (item) assert(type(item) == "number") end);
+			r:run(1);
+			r:run(2);
+		end);
+	end);
+	describe("#waiter", function()
+		it("should work", function ()
+			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
+		end);
+		
+		it("should work", function ()
+			--------------------
+			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
+		end);
+		it("should work", function ()
+			--------------------
+			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
+		end);
+		it("should work", function ()
+			--------------------
+			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
+		end);
+		it("should work with multiple runners in parallel", function ()
+			-- 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(4);
+			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
+		end);
+		it("should work work with multiple runners in parallel", function ()
+			--------------------
+			local wait1, done1;
+			local last_item1 = 0;
+			local r1, l1 = new(function (item)
+				print("r1 processing ", 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)
+				print("r2 processing ", item);
+				assert.is_number(item);
+				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.equal(r1.state, "ready");
+			r1:run(2);
+			assert.equal(r1.state, "ready");
+
+			r1:run(5);
+			assert.equal(r1.state, "ready");
+
+			local dones = {};
+			r1:run(3);
+			assert.equal(r1.state, "waiting");
+			r1:run(5); -- Will error, when we get to it
+			assert.equal(r1.state, "waiting");
+			done1();
+			assert.equal(r1.state, "ready");
+			r1:run(3);
+			assert.equal(r1.state, "waiting");
+
+			r2:run(1);
+			assert.equal(r1.state, "waiting");
+			assert.equal(r2.state, "ready");
+
+			r2:run(2);
+			assert.equal(r1.state, "waiting");
+			assert.equal(r2.state, "ready");
+
+			r2:run(3);
+			assert.equal(r1.state, "waiting");
+			assert.equal(r2.state, "waiting");
+
+			done2();
+			assert.equal(r1.state, "waiting");
+			assert.equal(r2.state, "ready");
+
+			r2:run(3);
+			assert.equal(r1.state, "waiting");
+			assert.equal(r2.state, "waiting");
+
+			done2();
+			assert.equal(r1.state, "waiting");
+			assert.equal(r2.state, "ready");
+
+			r2:run(4);
+			assert.equal(r1.state, "waiting");
+			assert.equal(r2.state, "ready");
+
+			done1();
+		
+			assert.equal(r1.state, "ready");
+			r1:run(4);
+			assert.equal(r1.state, "ready");
+
+			assert.equal(r1.state, "ready");
+			--for k, v in ipairs(l1) do print(k,v) end
+		end);
+	end);
+end);