util.queue: Add 'consume()' convenience iterator

This commit is contained in:
Matthew Wild 2019-03-23 08:47:55 +00:00
parent 95314bb2be
commit a274eacbbc
2 changed files with 40 additions and 0 deletions

View file

@ -100,4 +100,41 @@ describe("util.queue", function()
end);
end);
describe("consume()", function ()
it("should work", function ()
local q = queue.new(10);
for i = 1, 5 do
q:push(i);
end
local c = 0;
for i in q:consume() do
assert(i == c + 1);
assert(q:count() == (5-i));
c = i;
end
end);
it("should work even if items are pushed in the loop", function ()
local q = queue.new(10);
for i = 1, 5 do
q:push(i);
end
local c = 0;
for i in q:consume() do
assert(i == c + 1);
if c < 3 then
assert(q:count() == (5-i));
else
assert(q:count() == (6-i));
end
c = i;
if c == 3 then
q:push(6);
end
end
assert.equal(c, 6);
end);
end);
end);