util.iterators: join: Work even with only a single iterator in the chain
authorMatthew Wild <mwild1@gmail.com>
Thu, 06 Oct 2022 18:34:40 +0100
changeset 12748 e894677359e5
parent 12747 19113f232423
child 12749 2cbf0e9314ff
util.iterators: join: Work even with only a single iterator in the chain
spec/util_iterators_spec.lua
util/iterators.lua
--- a/spec/util_iterators_spec.lua	Thu Oct 06 16:00:39 2022 +0100
+++ b/spec/util_iterators_spec.lua	Thu Oct 06 18:34:40 2022 +0100
@@ -10,6 +10,14 @@
 			end
 			assert.same(output, expect);
 		end);
+		it("should work with only a single iterator", function ()
+			local expect = { "a", "b", "c" };
+			local output = {};
+			for x in iter.join(iter.values({"a", "b", "c"})) do
+				table.insert(output, x);
+			end
+			assert.same(output, expect);
+		end);
 	end);
 
 	describe("sorted_pairs", function ()
--- a/util/iterators.lua	Thu Oct 06 16:00:39 2022 +0100
+++ b/util/iterators.lua	Thu Oct 06 18:34:40 2022 +0100
@@ -240,7 +240,8 @@
 end
 
 function it.join(f, s, var)
-	return setmetatable({ {f, s, var} }, join_mt);
+	local t = setmetatable({ {f, s, var} }, join_mt);
+	return t, { t, 1 };
 end
 
 return it;