@marimo-team/frontend 0.24.3-dev24 → 0.24.3-dev26

package/src/core/cells/__tests__/cell.test.ts

@@ -220,6 +220,67 @@ describe("transitionCell serialization", () => {
 });
 
 describe("transitionCell stdin resolution", () => {
+  describe.each([
+    {
+      name: "interrupted",
+      message: {
+        output: {
+          channel: "marimo-error",
+          mimetype: "application/vnd.marimo+error",
+          data: [{ type: "interruption" }],
+          timestamp: 1,
+        },
+      } as CellMessage,
+    },
+    {
+      name: "idle",
+      message: { status: "idle" } as CellMessage,
+    },
+  ])("when $name", ({ name, message }) => {
+    it.each(["text/plain", "text/password"] as const)(
+      "preserves a processed %s prompt",
+      (mimetype) => {
+        const prompt: OutputMessage = {
+          channel: "stdin",
+          mimetype,
+          data: "Enter a value: ",
+          timestamp: 0,
+        };
+        const waiting = transitionCell(
+          createCellRuntimeState({ status: "running" }),
+          { console: prompt } as CellMessage,
+        );
+        const resolved = transitionCell(waiting, message);
+
+        expect(resolved.interrupted).toBe(name === "interrupted");
+        expect(resolved.consoleOutputs[0]).toMatchObject({
+          ...prompt,
+          response: "",
+        });
+      },
+    );
+
+    it.each(["text/plain", "text/password"] as const)(
+      "preserves an existing %s response",
+      (mimetype) => {
+        const prompt = {
+          channel: "stdin" as const,
+          mimetype,
+          data: "Enter a value: ",
+          timestamp: 0,
+          response: "submitted value",
+        };
+        const answered = createCellRuntimeState({
+          status: "running",
+          consoleOutputs: [prompt],
+        });
+        const resolved = transitionCell(answered, message);
+
+        expect(resolved.consoleOutputs).toEqual([prompt]);
+      },
+    );
+  });
+
   function stdinOutput(): OutputMessage {
     return {
       channel: "stdin",

package/src/core/cells/cell.ts

@@ -126,7 +126,15 @@ export function transitionCell(
     nextCell.debuggerActive = false;
     consoleOutputs = consoleOutputs.map((output) => {
       if (output.channel === "stdin") {
-        return { ...output, response: output.response ?? "" };
+        // StatefulOutputMessage exposes data through a getter, which object
+        // spread would omit.
+        return {
+          channel: output.channel,
+          mimetype: output.mimetype,
+          data: output.data,
+          timestamp: output.timestamp,
+          response: output.response ?? "",
+        };
       }
       return output;
     });