Ekdohibs
/
zig
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix issue #6303: iterating empty PriorityQueue crashes

master
Literally Void 2020-09-09 20:49:49 -07:00 committed by Veikka Tuominen
parent 749417a1f3
commit 78baa16da0
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lib/std/priority_queue.zig
View File

@ -195,7 +195,7 @@ pub fn PriorityQueue(comptime T: type) type {
count: usize,
pub fn next(it: *Iterator) ?T {
if (it.count > it.queue.len - 1) return null;
if (it.count >= it.queue.len) return null;
const out = it.count;
it.count += 1;
return it.queue.items[out];
@ -428,3 +428,12 @@ test "std.PriorityQueue: remove at index" {
expectEqual(queue.remove(), 3);
expectEqual(queue.removeOrNull(), null);
}
test "std.PriorityQueue: iterator while empty" {
var queue = PQ.init(testing.allocator, lessThan);
defer queue.deinit();
var it = queue.iterator();
expectEqual(it.next(), null);
}