# HG changeset patch
# User Matthew Wild <mwild1@gmail.com>
# Date 1521412986 0
# Node ID e88744fa09859711f603a41733c16137832014d6
# Parent  24b59a62acce29702df26e345efa141a6fa93655
util.async: Add some more tests for wait/done

diff -r 24b59a62acce -r e88744fa0985 spec/util_async_spec.lua
--- a/spec/util_async_spec.lua	Sun Mar 18 12:05:38 2018 +0000
+++ b/spec/util_async_spec.lua	Sun Mar 18 22:43:06 2018 +0000
@@ -534,6 +534,46 @@
 			--for k, v in ipairs(l1) do print(k,v) end
 		end);
 
+		it("should support multiple done() calls", function ()
+			local processed_item;
+			local wait, done;
+			local rf = spy.new(function (item)
+				wait, done = async.waiter(4);
+				wait();
+				processed_item = item;
+			end);
+			local r = async.runner(rf, mock_watchers());
+			r:run("test");
+			for i = 1, 3 do
+				done();
+				assert.equal(r.state, "waiting");
+				assert.is_nil(processed_item);
+			end
+			done();
+			assert.equal(r.state, "ready");
+			assert.equal(processed_item, "test");
+			assert.spy(r.watchers.error).was_not.called();
+		end);
+
+		it("should not allow done() to be called more than specified", function ()
+			local processed_item;
+			local wait, done;
+			local rf = spy.new(function (item)
+				wait, done = async.waiter(4);
+				wait();
+				processed_item = item;
+			end);
+			local r = async.runner(rf, mock_watchers());
+			r:run("test");
+			for i = 1, 4 do
+				done();
+			end
+			assert.has_error(done);;
+			assert.equal(r.state, "ready");
+			assert.equal(processed_item, "test");
+			assert.spy(r.watchers.error).was_not.called();
+		end);
+
 		it("should allow done() to be called before wait()", function ()
 			local processed_item;
 			local rf = spy.new(function (item)