Commit 5138856a72fc009ec799cb935d9117e3bd72f16a | zig

author	Luuk de Gram <luuk@degram.dev> 2022-05-14 21:39:09+02:00
committer	Jakub Konka <kubkon@jakubkonka.com> 2022-05-15 09:30:59+02:00
log	5138856a72fc009ec799cb935d9117e3bd72f16a
tree	fb4713ece697eebf87c57acde6c4bc50f6627471
parent	1bdcbd18ae1f312748c9909db98532a8dfd006eb

test harness: Set filename on error return

While calling `next` an error can occur while parsing the file.
However, we don't set the filename that is currently being processed, until `next` completed successfully.
This means that for invalid test names, the wrong filename was being displayed in the panic message.
The fix is to retrieve the correct filename when an error occurs and then setting the filename appropriately.

1 files changed, 22 insertions(+), 6 deletions(-)

src
- test.zig

src/test.zig+22-6

 @@ -398,6 +398,8 @@ const TestIterator = struct {
     start: usize = 0,
     end: usize = 0,
     filenames: []const []const u8,
+    /// reset on each call to `next`
+    index: usize = 0,
     const Error = error{InvalidIncrementalTestIndex};
 @@ -416,12 +418,12 @@ const TestIterator = struct {
         }
         const remaining = it.filenames[it.end..];
-        var i: usize = 0;
+        it.index = 0;
-        while (i < remaining.len - 1) : (i += 1) {
+        while (it.index < remaining.len - 1) : (it.index += 1) {
             // First, check if this file is part of an incremental update sequence
             // Split filename into "<base_name>.<index>.<file_ext>"
-            const prev_parts = getTestFileNameParts(remaining[i]);
+            const prev_parts = getTestFileNameParts(remaining[it.index]);
-            const new_parts = getTestFileNameParts(remaining[i + 1]);
+            const new_parts = getTestFileNameParts(remaining[it.index + 1]);
             // If base_name and file_ext match, these files are in the same test sequence
             // and the new one should be the incremented version of the previous test
 @@ -441,13 +443,22 @@ const TestIterator = struct {
                 if (new_parts.test_index != null and new_parts.test_index.? != 0)
                     return error.InvalidIncrementalTestIndex;
-                it.end += i + 1;
+                it.end += it.index + 1;
                 break;
             }
         } else {
             it.end += remaining.len;
         }
     }
+    /// In the event of an `error.InvalidIncrementalTestIndex`, this function can
+    /// be used to find the current filename that was being processed.
+    /// Asserts the iterator hasn't reached the end.
+    fn currentFilename(it: TestIterator) []const u8 {
+        assert(it.end != it.filenames.len);
+        const remaining = it.filenames[it.end..];
+        return remaining[it.index + 1];
+    }
 };
 /// For a filename in the format "<filename>.X.<ext>" or "<filename>.<ext>", returns
 @@ -1051,7 +1062,8 @@ pub const TestContext = struct {
         sortTestFilenames(filenames.items);
         var test_it = TestIterator{ .filenames = filenames.items };
-        while (try test_it.next()) |batch| {
+        while (test_it.next()) |maybe_batch| {
+            const batch = maybe_batch orelse break;
             const strategy: TestStrategy = if (batch.len > 1) .incremental else .independent;
             var cases = std.ArrayList(usize).init(ctx.arena);
 @@ -1133,6 +1145,10 @@ pub const TestContext = struct {
                     }
                 }
             }
+        } else |err| {
+            // make sure the current file is set to the file that produced an error
+            current_file.* = test_it.currentFilename();
+            return err;
         }
     }