@ryanfw/prompt-orchestration-pipeline 1.2.7 → 1.2.9

package/src/ui/components/__tests__/DAGGrid.test.tsx

@@ -117,3 +117,95 @@ test("DAGGrid opens task details from keyboard interaction and renders connector
   expect(view.getByText("Build · done")).toBeTruthy();
   expect(view.container.querySelectorAll('path[marker-end="url(#dag-grid-arrow)"]').length).toBe(1);
 });
+
+test("DAGGrid does not render the restart badge when restartCount is absent or zero", () => {
+  installMatchMedia();
+
+  const view = render(
+    <DAGGrid
+      items={[makeItem({ id: "build", title: "Build", status: "pending" })]}
+      jobId="job-1"
+      taskById={{ build: makeTaskState({ name: "build", state: "pending" }) }}
+      pipelineTasks={["build"]}
+      filesByTypeForItem={() => emptyFiles}
+      geometryAdapter={geometryAdapter}
+    />,
+  );
+
+  expect(view.container.querySelector('[data-role="restart-badge"]')).toBeNull();
+
+  const viewZero = render(
+    <DAGGrid
+      items={[makeItem({ id: "build", title: "Build", status: "pending", restartCount: 0 })]}
+      jobId="job-1"
+      taskById={{ build: makeTaskState({ name: "build", state: "pending" }) }}
+      pipelineTasks={["build"]}
+      filesByTypeForItem={() => emptyFiles}
+      geometryAdapter={geometryAdapter}
+    />,
+  );
+
+  expect(viewZero.container.querySelector('[data-role="restart-badge"]')).toBeNull();
+});
+
+test("DAGGrid renders the restart badge with singular label when restartCount is 1", () => {
+  installMatchMedia();
+
+  const view = render(
+    <DAGGrid
+      items={[makeItem({ id: "build", title: "Build", status: "pending", restartCount: 1 })]}
+      jobId="job-1"
+      taskById={{ build: makeTaskState({ name: "build", state: "pending" }) }}
+      pipelineTasks={["build"]}
+      filesByTypeForItem={() => emptyFiles}
+      geometryAdapter={geometryAdapter}
+    />,
+  );
+
+  const badge = view.container.querySelector('[data-role="restart-badge"]');
+  expect(badge).not.toBeNull();
+  expect(badge!.getAttribute("aria-label")).toBe("Restarted 1 time");
+  expect(badge!.getAttribute("title")).toBe("Restarted 1 time");
+  expect(badge!.textContent).toBe("↻ 1");
+});
+
+test("DAGGrid renders the restart badge with plural label when restartCount is greater than 1", () => {
+  installMatchMedia();
+
+  const view = render(
+    <DAGGrid
+      items={[makeItem({ id: "build", title: "Build", status: "pending", restartCount: 5 })]}
+      jobId="job-1"
+      taskById={{ build: makeTaskState({ name: "build", state: "pending" }) }}
+      pipelineTasks={["build"]}
+      filesByTypeForItem={() => emptyFiles}
+      geometryAdapter={geometryAdapter}
+    />,
+  );
+
+  const badge = view.container.querySelector('[data-role="restart-badge"]');
+  expect(badge).not.toBeNull();
+  expect(badge!.getAttribute("aria-label")).toBe("Restarted 5 times");
+  expect(badge!.textContent).toBe("↻ 5");
+});
+
+test("DAGGrid renders the restart badge as the first child of the card header", () => {
+  installMatchMedia();
+
+  const view = render(
+    <DAGGrid
+      items={[makeItem({ id: "build", title: "Build", status: "pending", restartCount: 2 })]}
+      jobId="job-1"
+      taskById={{ build: makeTaskState({ name: "build", state: "pending" }) }}
+      pipelineTasks={["build"]}
+      filesByTypeForItem={() => emptyFiles}
+      geometryAdapter={geometryAdapter}
+    />,
+  );
+
+  const header = view.container.querySelector('[data-role="card-header"]');
+  expect(header).not.toBeNull();
+  const firstChild = header!.firstElementChild;
+  expect(firstChild).not.toBeNull();
+  expect(firstChild!.getAttribute("data-role")).toBe("restart-badge");
+});

package/src/ui/components/__tests__/JobDetail.test.tsx

@@ -0,0 +1,62 @@
+import "./test-dom";
+
+import { render } from "@testing-library/react";
+import { afterEach, expect, mock, test } from "bun:test";
+
+import type { DagItem, JobDetail as JobDetailType, PipelineType } from "../types";
+
+const capturedItems: DagItem[][] = [];
+
+mock.module("../DAGGrid", () => ({
+  default: ({ items }: { items: DagItem[] }) => {
+    capturedItems.push(items);
+    return null;
+  },
+}));
+
+const JobDetail = (await import("../JobDetail")).default;
+
+afterEach(() => {
+  document.body.innerHTML = "";
+  capturedItems.length = 0;
+});
+
+function makeJob(restartCount: number): JobDetailType {
+  return {
+    id: "job-1",
+    name: "Job One",
+    status: "running",
+    tasks: {
+      build: {
+        name: "build",
+        state: "running",
+        startedAt: 1_000,
+        files: { artifacts: [], logs: [], tmp: [] },
+        restartCount,
+      },
+    },
+    pipeline: { tasks: ["build"] },
+    current: "build",
+  };
+}
+
+const pipeline: PipelineType = {
+  name: "Test",
+  slug: "test",
+  description: "",
+  tasks: [{ name: "build" }],
+};
+
+test("JobDetail re-emits dag item when only restartCount changes", () => {
+  const view = render(<JobDetail job={makeJob(1)} pipeline={pipeline} />);
+  expect(capturedItems.length).toBeGreaterThan(0);
+  const firstItems = capturedItems[capturedItems.length - 1] as DagItem[];
+  const firstBuild = firstItems.find((item) => item.id === "build");
+  expect(firstBuild?.restartCount).toBe(1);
+
+  view.rerender(<JobDetail job={makeJob(2)} pipeline={pipeline} />);
+  const lastItems = capturedItems[capturedItems.length - 1] as DagItem[];
+  const secondBuild = lastItems.find((item) => item.id === "build");
+  expect(secondBuild?.restartCount).toBe(2);
+  expect(secondBuild).not.toBe(firstBuild);
+});

package/src/ui/components/types.ts

@@ -43,6 +43,7 @@ export interface TaskStateObject {
   artifacts?: string[];
   tokenUsage?: Record<string, unknown>;
   attempts?: number;
+  restartCount?: number;
   executionTimeMs?: number;
 }
 
@@ -105,6 +106,7 @@ export interface DagItem {
   body: string | null;
   startedAt: string | number;
   endedAt: string | number | null;
+  restartCount?: number;
 }
 
 export interface PipelineTask {