Maintain test to traceback info including module files

Author: yairchu

frontend/src/utils/__tests__/traceback.test.ts

@@ -6,27 +6,34 @@ import { extractAllTracebackInfo } from "../traceback";
 describe("traceback", () => {
   test("extracts cell-link", () => {
     const errors = extractAllTracebackInfo(Tracebacks.raw);
-    expect(errors).toMatchInlineSnapshot(`
+    expect(errors[0].isCell).toBe(false);
+    expect(errors[0].src.endsWith("marimo/_runtime/executor.py")).toBe(true);
+    expect(errors.slice(1)).toMatchInlineSnapshot(`
       [
         {
-          "cellId": "Hbol",
+          "isCell": true,
           "lineNumber": 4,
+          "src": "Hbol",
         },
         {
-          "cellId": "Hbol",
+          "isCell": true,
           "lineNumber": 2,
+          "src": "Hbol",
         },
       ]
     `);
   });
 
   test("extracts cell-link from assertion", () => {
     const info = extractAllTracebackInfo(Tracebacks.assertion);
-    expect(info).toMatchInlineSnapshot(`
+    expect(info[0].isCell).toBe(false);
+    expect(info[0].src.endsWith("marimo/_runtime/executor.py")).toBe(true);
+    expect(info.slice(1)).toMatchInlineSnapshot(`
       [
         {
-          "cellId": "Hbol",
+          "isCell": true,
           "lineNumber": 1,
+          "src": "Hbol",
         },
       ]
     `);

frontend/src/utils/traceback.ts

@@ -112,11 +112,8 @@ export function extractAllTracebackInfo(traceback: string): TracebackInfo[] {
     replace: (domNode) => {
       const info = getTracebackInfo(domNode);
       if (info) {
-        infos.push({
-          cellId: info.cellId,
-          lineNumber: info.lineNumber,
-        });
-        return `${info.cellId}:${info.lineNumber}`;
+        infos.push(info);
+        return `${info.src}:${info.lineNumber}`;
       }
     },
   });