1 files changed, 8 insertions, 3 deletions
diff --git a/libpanto/src/tool_registry.zig b/libpanto/src/tool_registry.zig
index 35aff02..ec3dcbf 100644
--- a/libpanto/src/tool_registry.zig
+++ b/libpanto/src/tool_registry.zig
@@ -344,10 +344,10 @@ const TestTool = struct {
         .deinit = deinit,
     };
 
-    fn invoke(ctx: *anyopaque, input: []const u8, allocator: Allocator) anyerror![]u8 {
+    fn invoke(ctx: *anyopaque, input: []const u8, allocator: Allocator) anyerror![]tool_mod.ResultPart {
         const self: *TestTool = @ptrCast(@alignCast(ctx));
         self.invocations += 1;
-        return try allocator.dupe(u8, input);
+        return tool_mod.textResult(allocator, input);
     }
 
     fn deinit(ctx: *anyopaque, allocator: Allocator) void {
@@ -438,7 +438,12 @@ const TestSource = struct {
                 results[i] = .{ .err = e };
                 continue;
             };
-            results[i] = .{ .ok = buf };
+            results[i] = .{
+                .ok = tool_mod.ownedTextResult(allocator, buf) catch |e| {
+                    results[i] = .{ .err = e };
+                    continue;
+                },
+            };
         }
     }