@elench/testkit 0.1.81 → 0.1.83

package/lib/cli/tui/run-tree-state.test.mjs

@@ -0,0 +1,324 @@
+import { describe, expect, it } from "vitest";
+import { createRunTreeState } from "./run-tree-state.mjs";
+
+function makePlans(suites = []) {
+  return [
+    {
+      config: { name: "api" },
+      skipped: false,
+      suites: suites.length > 0 ? suites : [
+        {
+          name: "users",
+          type: "integration",
+          displayType: "int",
+          framework: "k6",
+          files: ["tests/api/users.int.testkit.ts", "tests/api/users-create.int.testkit.ts"],
+        },
+        {
+          name: "auth",
+          type: "integration",
+          displayType: "int",
+          framework: "k6",
+          files: ["tests/api/auth.int.testkit.ts"],
+        },
+      ],
+    },
+  ];
+}
+
+describe("run-tree-state", () => {
+  it("builds tree hierarchy from service plans", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+    const snap = state.getSnapshot();
+
+    expect(snap.services).toHaveLength(1);
+    expect(snap.services[0].name).toBe("api");
+    expect(snap.services[0].skipped).toBe(false);
+    expect(snap.services[0].types).toHaveLength(1);
+    expect(snap.services[0].types[0].type).toBe("int");
+    expect(snap.services[0].types[0].suites).toHaveLength(2);
+    expect(snap.services[0].types[0].suites[0].groupLabel).toBe("Users");
+    expect(snap.services[0].types[0].suites[0].visibleFiles).toHaveLength(2);
+    expect(snap.services[0].types[0].suites[1].groupLabel).toBe("Auth");
+  });
+
+  it("marks file running", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+    state.markFileRunning("api", "int:users", "tests/api/users.int.testkit.ts");
+    const snap = state.getSnapshot();
+    const file = snap.services[0].types[0].suites[0].visibleFiles.find(
+      (f) => f.path === "tests/api/users.int.testkit.ts"
+    );
+    expect(file.status).toBe("running");
+  });
+
+  it("collapses suite when all files pass", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+    state.setTotalFileCount(3);
+
+    state.markFileFinished(
+      { serviceName: "api", type: "integration", displayType: "int", framework: "k6", suiteName: "users", file: "tests/api/users.int.testkit.ts" },
+      { failed: false, durationMs: 1200 }
+    );
+    state.markFileFinished(
+      { serviceName: "api", type: "integration", displayType: "int", framework: "k6", suiteName: "users", file: "tests/api/users-create.int.testkit.ts" },
+      { failed: false, durationMs: 800 }
+    );
+
+    const snap = state.getSnapshot();
+    const suite = snap.services[0].types[0].suites[0];
+    expect(suite.collapsed).toBe(true);
+    expect(suite.collapseStatus).toBe("all_passed");
+    expect(suite.visibleFiles).toHaveLength(0);
+    expect(suite.totalDurationMs).toBe(2000);
+    expect(snap.completedCount).toBe(2);
+  });
+
+  it("collapses suite when all files skipped", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+
+    state.markFileFinished(
+      { serviceName: "api", type: "integration", displayType: "int", framework: "k6", suiteName: "users", file: "tests/api/users.int.testkit.ts" },
+      { status: "skipped", reason: "ci skip", durationMs: 0 }
+    );
+    state.markFileFinished(
+      { serviceName: "api", type: "integration", displayType: "int", framework: "k6", suiteName: "users", file: "tests/api/users-create.int.testkit.ts" },
+      { status: "skipped", reason: "ci skip", durationMs: 0 }
+    );
+
+    const snap = state.getSnapshot();
+    const suite = snap.services[0].types[0].suites[0];
+    expect(suite.collapsed).toBe(true);
+    expect(suite.collapseStatus).toBe("all_skipped");
+  });
+
+  it("keeps suite expanded when a file fails", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+
+    state.markFileFinished(
+      { serviceName: "api", type: "integration", displayType: "int", framework: "k6", suiteName: "users", file: "tests/api/users.int.testkit.ts" },
+      { failed: false, durationMs: 1200 }
+    );
+    state.markFileFinished(
+      { serviceName: "api", type: "integration", displayType: "int", framework: "k6", suiteName: "users", file: "tests/api/users-create.int.testkit.ts" },
+      { failed: true, error: "assertion failed", durationMs: 500, failureDetails: [] }
+    );
+
+    const snap = state.getSnapshot();
+    const suite = snap.services[0].types[0].suites[0];
+    expect(suite.collapsed).toBe(false);
+    // Only failed files visible when no running/pending
+    expect(suite.visibleFiles).toHaveLength(1);
+    expect(suite.visibleFiles[0].status).toBe("failed");
+    expect(suite.passedCount).toBe(1);
+  });
+
+  it("keeps suite expanded while files are running", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+
+    state.markFileRunning("api", "int:users", "tests/api/users.int.testkit.ts");
+    state.markFileFinished(
+      { serviceName: "api", type: "integration", displayType: "int", framework: "k6", suiteName: "users", file: "tests/api/users-create.int.testkit.ts" },
+      { failed: false, durationMs: 800 }
+    );
+
+    const snap = state.getSnapshot();
+    const suite = snap.services[0].types[0].suites[0];
+    expect(suite.collapsed).toBe(false);
+    expect(suite.visibleFiles).toHaveLength(2);
+  });
+
+  it("type collapses when all suites are collapsed", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+
+    // Pass all files in both suites
+    state.markFileFinished(
+      { serviceName: "api", type: "integration", displayType: "int", framework: "k6", suiteName: "users", file: "tests/api/users.int.testkit.ts" },
+      { failed: false, durationMs: 1000 }
+    );
+    state.markFileFinished(
+      { serviceName: "api", type: "integration", displayType: "int", framework: "k6", suiteName: "users", file: "tests/api/users-create.int.testkit.ts" },
+      { failed: false, durationMs: 500 }
+    );
+    state.markFileFinished(
+      { serviceName: "api", type: "integration", displayType: "int", framework: "k6", suiteName: "auth", file: "tests/api/auth.int.testkit.ts" },
+      { failed: false, durationMs: 2000 }
+    );
+
+    const snap = state.getSnapshot();
+    const type = snap.services[0].types[0];
+    expect(type.collapsed).toBe(true);
+  });
+
+  it("tracks completed and total counts", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+    state.setTotalFileCount(3);
+
+    expect(state.getSnapshot().totalCount).toBe(3);
+    expect(state.getSnapshot().completedCount).toBe(0);
+
+    state.markFileFinished(
+      { serviceName: "api", type: "integration", displayType: "int", framework: "k6", suiteName: "users", file: "tests/api/users.int.testkit.ts" },
+      { failed: false, durationMs: 100 }
+    );
+    expect(state.getSnapshot().completedCount).toBe(1);
+  });
+
+  it("handles skipped services", () => {
+    const state = createRunTreeState();
+    state.initFromPlans([{ config: { name: "web" }, skipped: true, suites: [] }]);
+    const snap = state.getSnapshot();
+    expect(snap.services[0].skipped).toBe(true);
+  });
+
+  it("markServiceSkipped updates existing service", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+    state.markServiceSkipped("api", "no matching types");
+    const snap = state.getSnapshot();
+    expect(snap.services[0].skipped).toBe(true);
+    expect(snap.services[0].skipReason).toBe("no matching types");
+  });
+
+  it("finish populates summaryData", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+
+    const mockResults = [
+      {
+        name: "api",
+        failed: false,
+        skipped: false,
+        suiteCount: 2,
+        completedSuiteCount: 2,
+        failedSuiteCount: 0,
+        skippedSuiteCount: 0,
+        totalFileCount: 3,
+        passedFileCount: 3,
+        failedFileCount: 0,
+        skippedFileCount: 0,
+        notRunFileCount: 0,
+        suites: [],
+        errors: [],
+      },
+    ];
+
+    state.finish(mockResults, 5000, null);
+    const snap = state.getSnapshot();
+    expect(snap.finished).toBe(true);
+    expect(snap.summaryData.result).toBe("PASSED");
+    expect(snap.summaryData.rows.length).toBeGreaterThanOrEqual(7);
+  });
+
+  it("subscribe fires callback on state mutations", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+    let callCount = 0;
+    state.subscribe(() => { callCount += 1; });
+
+    state.setTotalFileCount(3);
+    state.setPhase("running");
+    expect(callCount).toBe(2);
+  });
+
+  it("markPlannedSkip sets file status and increments completed", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+    state.setTotalFileCount(3);
+
+    state.markPlannedSkip({ serviceName: "api", file: "tests/api/users.int.testkit.ts", reason: "ci skip" });
+    const snap = state.getSnapshot();
+    const file = snap.services[0].types[0].suites[0].visibleFiles.find(
+      (f) => f.path === "tests/api/users.int.testkit.ts"
+    );
+    expect(file.status).toBe("skipped");
+    expect(snap.completedCount).toBe(1);
+  });
+
+  it("markRuntimeError marks file failed", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+
+    state.markRuntimeError(
+      { serviceName: "api", file: "tests/api/users.int.testkit.ts" },
+      "process crashed"
+    );
+    const snap = state.getSnapshot();
+    const file = snap.services[0].types[0].suites[0].visibleFiles.find(
+      (f) => f.path === "tests/api/users.int.testkit.ts"
+    );
+    expect(file.status).toBe("failed");
+    expect(file.error).toBe("process crashed");
+  });
+
+  it("tracks the selected failure after a file fails", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+
+    state.markFileFinished(
+      {
+        serviceName: "api",
+        type: "integration",
+        displayType: "int",
+        framework: "k6",
+        suiteName: "users",
+        file: "tests/api/users-create.int.testkit.ts",
+      },
+      { failed: true, error: "assertion failed", durationMs: 500, failureDetails: [] }
+    );
+
+    const snap = state.getSnapshot();
+    expect(snap.failures).toHaveLength(1);
+    expect(snap.selectedFailure.filePath).toBe("tests/api/users-create.int.testkit.ts");
+  });
+
+  it("cycles through failures", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+
+    state.markRuntimeError(
+      { serviceName: "api", file: "tests/api/users.int.testkit.ts" },
+      "process crashed"
+    );
+    state.markRuntimeError(
+      { serviceName: "api", file: "tests/api/auth.int.testkit.ts" },
+      "another crash"
+    );
+
+    const first = state.getSnapshot().selectedFailure.filePath;
+    state.selectNextFailure();
+    const second = state.getSnapshot().selectedFailure.filePath;
+    expect(second).not.toBe(first);
+    state.selectPreviousFailure();
+    expect(state.getSnapshot().selectedFailure.filePath).toBe(first);
+  });
+
+  it("records investigation transcript state", () => {
+    const state = createRunTreeState();
+    state.initFromPlans(makePlans());
+    state.markRuntimeError(
+      { serviceName: "api", file: "tests/api/users.int.testkit.ts" },
+      "process crashed"
+    );
+
+    state.beginInvestigation({ provider: "codex", userMessage: "Investigate this failure" });
+    state.appendAgentEvent({ type: "start" });
+    state.appendAgentEvent({ type: "status", message: "Inspecting repository" });
+    state.appendAgentEvent({ type: "delta", text: "Likely root cause." });
+    state.completeAgentSession({ finalText: "Likely root cause.", exitCode: 0 });
+
+    const snap = state.getSnapshot();
+    expect(snap.mode).toBe("investigating");
+    expect(snap.agentSession.status).toBe("complete");
+    expect(snap.agentSession.entries.some((entry) => entry.kind === "assistant")).toBe(true);
+    expect(snap.agentSession.finalText).toContain("Likely root cause");
+  });
+});