test: address review feedback for anonymous function traceback test (#1665)

ChouUn · claude · ChouUn · commit d61f863c0b92 · 2026-03-01T06:41:40.000+08:00
Use distinct sourcemap values and explicit &lt;main.ts:N&gt; assertions
per reviewer request.

Co-Authored-By: Claude Opus 4.6 &lt;noreply@anthropic.com&gt;
diff --git a/test/unit/error.spec.ts b/test/unit/error.spec.ts
@@ -350,40 +350,47 @@ test("sourceMapTraceback maps anonymous function locations in .lua files (#1665)
     // Nested IIFEs produce <file.lua:N> anonymous function notation in traceback.
     // Old pattern (%S+)%.lua:(%d+) captures "<main" from "<main.lua:4>",
     // failing the sourcemap lookup. Fix: ([^%s<]+) excludes "<".
-    //
-    // Compiled Lua for the nested IIFEs below:
-    //   line 3: (function()
-    //   line 4:     (function()
-    //   line 5:         print(debug.traceback())
-    //   line 6:     end)(nil)
-    //   line 7: end)(nil)
+
+    // mapping is copied from the emitted Lua, not invented.
+    const mapping = `{["5"] = 1,["6"] = 2,["7"] = 3,["8"] = 4,["9"] = 3,["10"] = 2,["11"] = 1}`;
+
+    // Test harness executes via luaL_dostring (chunk names are [string "..."]), so we mock a file-based traceback.
     const fakeTraceback = [
         "stack traceback:",
-        "\tmain.lua:5: in function <main.lua:4>",
-        "\tmain.lua:6: in function <main.lua:3>",
+        "\tmain.lua:8: in function <main.lua:7>",
+        "\tmain.lua:7: in function <main.lua:6>",
         "\t[C]: in ?",
     ].join("\n");
 
-    const result = util.testFunction`
+    const builder = util.testFunction`
         return (() => {
             return (() => {
-                return debug.traceback();
+                return (debug.traceback as (this: void) => string)();
             })();
         })();
     `
-        .setLuaHeader(
-            `__TS__sourcemap = { ["main.lua"] = {["3"] = 7, ["4"] = 8, ["5"] = 9, ["6"] = 8} }\n` +
-                `local __real_tb = debug.traceback\n` +
-                `debug.traceback = function() return ${JSON.stringify(fakeTraceback)} end`
-        )
-        .setOptions({ sourceMapTraceback: true })
-        .getLuaExecutionResult();
-
-    // Regular line references should be mapped
-    expect(result).toContain("main.ts:9");
-    expect(result).toContain("main.ts:8");
-    // Anonymous function definitions <main.lua:4> and <main.lua:3> should also be mapped
+        // Inject sourcemap for "main.lua" and mock debug.traceback to return file-based frames.
+        .setLuaHeader(`
+            __TS__sourcemap = { ["main.lua"] = ${mapping} };
+            local __real_tb = debug.traceback
+            debug.traceback = function() return ${JSON.stringify(fakeTraceback)} end
+        `)
+        .setOptions({ sourceMapTraceback: true });
+
+    const lua = builder.getMainLuaCodeChunk();
+    // Sanity check: emitted code registers the same mapping literal we inject above.
+    expect(lua).toContain(`__TS__SourceMapTraceBack(debug.getinfo(1).short_src, ${mapping});`);
+
+    const result = builder.getLuaExecutionResult();
+    expect(result).toEqual(expect.any(String));
+    // Both `main.lua:N` and `<main.lua:N>` frames should be rewritten using the sourcemap.
     expect(result).not.toContain("main.lua");
+    // Regular line references
+    expect(result).toContain("\tmain.ts:4:");
+    expect(result).toContain("\tmain.ts:3:");
+    // Anonymous function references must keep <> format
+    expect(result).toContain("<main.ts:3>");
+    expect(result).toContain("<main.ts:2>");
 });
 
 util.testEachVersion(
