@arii/boomtick-mcp 0.4.2 → 0.6.0

package/src/tools/contract.ts

@@ -847,6 +847,30 @@ export const CodeReviewResultJsonSchema = {
           "default": null,
           "title": "Line"
         },
+        "startLine": {
+          "anyOf": [
+            {
+              "type": "integer"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Startline"
+        },
+        "endLine": {
+          "anyOf": [
+            {
+              "type": "integer"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Endline"
+        },
         "snippet": {
           "anyOf": [
             {
@@ -882,6 +906,18 @@ export const CodeReviewResultJsonSchema = {
           ],
           "default": null,
           "title": "Counterexample"
+        },
+        "suggestion": {
+          "anyOf": [
+            {
+              "type": "string"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Suggestion"
         }
       },
       "required": [
@@ -1077,6 +1113,30 @@ export const CodeReviewStateJsonSchema = {
           "default": null,
           "title": "Line"
         },
+        "startLine": {
+          "anyOf": [
+            {
+              "type": "integer"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Startline"
+        },
+        "endLine": {
+          "anyOf": [
+            {
+              "type": "integer"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Endline"
+        },
         "snippet": {
           "anyOf": [
             {
@@ -1112,6 +1172,18 @@ export const CodeReviewStateJsonSchema = {
           ],
           "default": null,
           "title": "Counterexample"
+        },
+        "suggestion": {
+          "anyOf": [
+            {
+              "type": "string"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Suggestion"
         }
       },
       "required": [
@@ -1597,6 +1669,39 @@ export const GetPackageScriptsInputJsonSchema = {
   "type": "object"
 } as const;
 
+export const GetPrCommitsInputSchema = z.object({ "prNumber": z.number().int().describe("The number of the PR to list commits for."), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
+
+export type GetPrCommitsInput = z.infer<typeof GetPrCommitsInputSchema>;
+
+export const GetPrCommitsInputJsonSchema = {
+  "properties": {
+    "prNumber": {
+      "description": "The number of the PR to list commits for.",
+      "title": "Prnumber",
+      "type": "integer"
+    },
+    "repo": {
+      "anyOf": [
+        {
+          "pattern": "^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$",
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "description": "The target repository override (e.g. org/repo).",
+      "title": "Repo"
+    }
+  },
+  "required": [
+    "prNumber"
+  ],
+  "title": "GetPrCommitsInput",
+  "type": "object"
+} as const;
+
 export const GetPrDiffInputSchema = z.object({ "prNumber": z.number().int().describe("The number of the pull request to get the diff for."), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
 
 export type GetPrDiffInput = z.infer<typeof GetPrDiffInputSchema>;
@@ -1663,6 +1768,39 @@ export const GetPrInputJsonSchema = {
   "type": "object"
 } as const;
 
+export const GetPrReviewsInputSchema = z.object({ "prNumber": z.number().int().describe("The number of the PR to view reviews for."), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
+
+export type GetPrReviewsInput = z.infer<typeof GetPrReviewsInputSchema>;
+
+export const GetPrReviewsInputJsonSchema = {
+  "properties": {
+    "prNumber": {
+      "description": "The number of the PR to view reviews for.",
+      "title": "Prnumber",
+      "type": "integer"
+    },
+    "repo": {
+      "anyOf": [
+        {
+          "pattern": "^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$",
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "description": "The target repository override (e.g. org/repo).",
+      "title": "Repo"
+    }
+  },
+  "required": [
+    "prNumber"
+  ],
+  "title": "GetPrReviewsInput",
+  "type": "object"
+} as const;
+
 export const GetRouteMapInputSchema = z.object({ "includeStatic": z.union([z.boolean(), z.null()]).describe("Whether to include static assets in the route map.").default(null) });
 
 export type GetRouteMapInput = z.infer<typeof GetRouteMapInputSchema>;
@@ -1687,6 +1825,57 @@ export const GetRouteMapInputJsonSchema = {
   "type": "object"
 } as const;
 
+export const GetWorkflowRunsInputSchema = z.object({ "branch": z.union([z.string(), z.null()]).describe("Filter workflow runs by branch name.").default(null), "event": z.union([z.string(), z.null()]).describe("Filter workflow runs by event (e.g. push, pull_request).").default(null), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
+
+export type GetWorkflowRunsInput = z.infer<typeof GetWorkflowRunsInputSchema>;
+
+export const GetWorkflowRunsInputJsonSchema = {
+  "properties": {
+    "branch": {
+      "anyOf": [
+        {
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "description": "Filter workflow runs by branch name.",
+      "title": "Branch"
+    },
+    "event": {
+      "anyOf": [
+        {
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "description": "Filter workflow runs by event (e.g. push, pull_request).",
+      "title": "Event"
+    },
+    "repo": {
+      "anyOf": [
+        {
+          "pattern": "^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$",
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "description": "The target repository override (e.g. org/repo).",
+      "title": "Repo"
+    }
+  },
+  "title": "GetWorkflowRunsInput",
+  "type": "object"
+} as const;
+
 export const HealthCheckInputSchema = z.object({ "checkDeep": z.boolean().describe("Whether to perform a deep health check including external dependencies.").default(false) });
 
 export type HealthCheckInput = z.infer<typeof HealthCheckInputSchema>;
@@ -2342,7 +2531,7 @@ export const ReadAgentContextInputJsonSchema = {
   "type": "object"
 } as const;
 
-export const ReadCiLogsInputSchema = z.object({ "prNumber": z.number().int().describe("The number of the pull request to read logs for."), "all": z.boolean().describe("Include logs for successful runs (default: false).").default(false) });
+export const ReadCiLogsInputSchema = z.object({ "prNumber": z.number().int().describe("The number of the pull request to read logs for."), "all": z.boolean().describe("Include logs for successful runs (default: false).").default(false), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
 
 export type ReadCiLogsInput = z.infer<typeof ReadCiLogsInputSchema>;
 
@@ -2358,6 +2547,20 @@ export const ReadCiLogsInputJsonSchema = {
       "description": "Include logs for successful runs (default: false).",
       "title": "All",
       "type": "boolean"
+    },
+    "repo": {
+      "anyOf": [
+        {
+          "pattern": "^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$",
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "description": "The target repository override (e.g. org/repo).",
+      "title": "Repo"
     }
   },
   "required": [
@@ -2538,7 +2741,7 @@ export const ReadPRCommentsResponseJsonSchema = {
   "type": "object"
 } as const;
 
-export const RepoLogsInputSchema = z.object({ "prNumber": z.number().int().describe("The number of the pull request to stream logs for."), "grep": z.union([z.string(), z.null()]).describe("Optional pattern to filter log lines.").default(null) });
+export const RepoLogsInputSchema = z.object({ "prNumber": z.number().int().describe("The number of the pull request to stream logs for."), "grep": z.union([z.string(), z.null()]).describe("Optional pattern to filter log lines.").default(null), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
 
 export type RepoLogsInput = z.infer<typeof RepoLogsInputSchema>;
 
@@ -2561,6 +2764,20 @@ export const RepoLogsInputJsonSchema = {
       "default": null,
       "description": "Optional pattern to filter log lines.",
       "title": "Grep"
+    },
+    "repo": {
+      "anyOf": [
+        {
+          "pattern": "^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$",
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "description": "The target repository override (e.g. org/repo).",
+      "title": "Repo"
     }
   },
   "required": [
@@ -2615,7 +2832,7 @@ export const ReviewCommentJsonSchema = {
   "type": "object"
 } as const;
 
-export const ReviewFindingSchema = z.object({ "id": z.string(), "file": z.string(), "issue": z.string(), "status": z.enum(["open","resolved"]), "confidence": z.enum(["high","medium","low"]), "severity": z.union([z.enum(["HIGH","MEDIUM","LOW"]), z.null()]).default(null), "line": z.union([z.number().int(), z.null()]).default(null), "snippet": z.union([z.string(), z.null()]).default(null), "fixSummary": z.union([z.string(), z.null()]).default(null), "counterexample": z.union([z.string(), z.null()]).default(null) });
+export const ReviewFindingSchema = z.object({ "id": z.string(), "file": z.string(), "issue": z.string(), "status": z.enum(["open","resolved"]), "confidence": z.enum(["high","medium","low"]), "severity": z.union([z.enum(["HIGH","MEDIUM","LOW"]), z.null()]).default(null), "line": z.union([z.number().int(), z.null()]).default(null), "startLine": z.union([z.number().int(), z.null()]).default(null), "endLine": z.union([z.number().int(), z.null()]).default(null), "snippet": z.union([z.string(), z.null()]).default(null), "fixSummary": z.union([z.string(), z.null()]).default(null), "counterexample": z.union([z.string(), z.null()]).default(null), "suggestion": z.union([z.string(), z.null()]).default(null) });
 
 export type ReviewFinding = z.infer<typeof ReviewFindingSchema>;
 
@@ -2679,6 +2896,30 @@ export const ReviewFindingJsonSchema = {
       "default": null,
       "title": "Line"
     },
+    "startLine": {
+      "anyOf": [
+        {
+          "type": "integer"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "title": "Startline"
+    },
+    "endLine": {
+      "anyOf": [
+        {
+          "type": "integer"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "title": "Endline"
+    },
     "snippet": {
       "anyOf": [
         {
@@ -2714,6 +2955,18 @@ export const ReviewFindingJsonSchema = {
       ],
       "default": null,
       "title": "Counterexample"
+    },
+    "suggestion": {
+      "anyOf": [
+        {
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "title": "Suggestion"
     }
   },
   "required": [

package/src/tools/github.checkout_branch.ts

@@ -7,7 +7,7 @@ export const CheckoutBranchInputSchema = z.object({
 });
 
 export async function checkoutBranchHandler(args: z.infer<typeof CheckoutBranchInputSchema>) {
-  const result = await runCommand("td-cli", ["gh", "checkout", args.branch], { cwd: args.worktreePath });
+  const result = await runCommand("td", ["gh", "checkout", args.branch], { cwd: args.worktreePath });
 
   if (result.exitCode !== 0) {
     throw new Error(`Failed to checkout branch ${args.branch}: ${result.stderr}`);

package/src/tools/github.comment_triage_summary.ts

@@ -7,7 +7,7 @@ export const CommentTriageSummaryInputSchema = z.object({
 });
 
 export async function commentTriageSummaryHandler(args: z.infer<typeof CommentTriageSummaryInputSchema>) {
-  const result = await runCommand("td-cli", [
+  const result = await runCommand("td", [
     "gh",
     "post-comment",
     "--pr", args.prNumber.toString(),

package/src/tools/github.create_issue.test.ts

@@ -23,11 +23,11 @@ describe("github.create_issue", () => {
       stderr: "",
       exitCode: 0,
       durationMs: 10,
-      command: "td-cli gh create-issue --title 'Test Issue' --body 'Test Body'"
+      command: "td gh create-issue --title 'Test Issue' --body 'Test Body'"
     });
 
     const result = await createIssueHandler({ title: "Test Issue", body: "Test Body", file: null, repo: null });
-    expect(shell.runCommand).toHaveBeenCalledWith("td-cli", [
+    expect(shell.runCommand).toHaveBeenCalledWith("td", [
       "gh",
       "create-issue",
       "--title",
@@ -58,7 +58,7 @@ describe("github.create_issue", () => {
       stderr: "",
       exitCode: 0,
       durationMs: 10,
-      command: "td-cli gh create-issue --title 'Cross Repo Issue' --body 'Body' --repo 'org/other-repo'"
+      command: "td gh create-issue --title 'Cross Repo Issue' --body 'Body' --repo 'org/other-repo'"
     });
 
     const result = await createIssueHandler({
@@ -68,7 +68,7 @@ describe("github.create_issue", () => {
       repo: "org/other-repo"
     });
 
-    expect(shell.runCommand).toHaveBeenCalledWith("td-cli", [
+    expect(shell.runCommand).toHaveBeenCalledWith("td", [
       "gh",
       "create-issue",
       "--title",
@@ -88,7 +88,7 @@ describe("github.create_issue", () => {
       stderr: "Auth failed",
       exitCode: 1,
       durationMs: 10,
-      command: "td-cli gh create-issue"
+      command: "td gh create-issue"
     });
 
     await expect(createIssueHandler({ title: "Test Issue", body: "Test Body", file: null, repo: null })).rejects.toThrow("Failed to create issue: Auth failed");
@@ -105,7 +105,7 @@ describe("github.create_issue", () => {
       stderr: "",
       exitCode: 0,
       durationMs: 10,
-      command: "td-cli gh create-issue"
+      command: "td gh create-issue"
     });
 
     await expect(createIssueHandler({ title: "Test Issue", body: "Test Body", file: null, repo: null })).rejects.toThrow("Failed to create issue: Repo not found");

package/src/tools/github.create_issue.ts

@@ -15,7 +15,7 @@ export async function createIssueHandler(args: z.input<typeof CreateIssueInputSc
   if (params.file) cmdArgs.push("--file", params.file);
   cmdArgs.push("--repo", targetRepo);
 
-  const result = await runCommand("td-cli", cmdArgs);
+  const result = await runCommand("td", cmdArgs);
 
   if (result.exitCode !== 0) {
     throw new Error(`Failed to create issue: ${sanitizeError(result.stderr)}`);

package/src/tools/github.create_pull_request.test.ts

@@ -21,7 +21,7 @@ describe("github.create_pull_request", () => {
       stderr: "",
       exitCode: 0,
       durationMs: 10,
-      command: "td-cli gh create-pr --title 'PR' --body 'Body' --head 'feature' --base 'main'"
+      command: "td gh create-pr --title 'PR' --body 'Body' --head 'feature' --base 'main'"
     });
 
     const result = await createPullRequestHandler({
@@ -50,7 +50,7 @@ describe("github.create_pull_request", () => {
       stderr: "",
       exitCode: 0,
       durationMs: 10,
-      command: "td-cli gh create-pr --title 'PR' --body 'Body' --head 'feature' --base 'main' --repo 'org/other-repo'"
+      command: "td gh create-pr --title 'PR' --body 'Body' --head 'feature' --base 'main' --repo 'org/other-repo'"
     });
 
     const result = await createPullRequestHandler({
@@ -62,7 +62,7 @@ describe("github.create_pull_request", () => {
       repo: "org/other-repo"
     });
 
-    expect(shell.runCommand).toHaveBeenCalledWith("td-cli", [
+    expect(shell.runCommand).toHaveBeenCalledWith("td", [
       "gh",
       "create-pr",
       "--title",
@@ -91,7 +91,7 @@ describe("github.create_pull_request", () => {
       stderr: "",
       exitCode: 0,
       durationMs: 10,
-      command: "td-cli gh create-pr"
+      command: "td gh create-pr"
     });
 
     await expect(createPullRequestHandler({

package/src/tools/github.create_pull_request.ts

@@ -23,7 +23,7 @@ export async function createPullRequestHandler(args: z.input<typeof CreatePullRe
     tdArgs.push("--draft");
   }
 
-  const result = await runCommand("td-cli", tdArgs);
+  const result = await runCommand("td", tdArgs);
 
   if (result.exitCode !== 0) {
     // Attempt to parse stdout as JSON to see if there's a structured error message

package/src/tools/github.get_merge_conflict_files.ts

@@ -1,31 +1,21 @@
 import { z } from "zod";
-import { runCommand } from "../lib/shell.js";
+import { getMergeConflictFiles } from "../lib/git.js";
 
 export const GetMergeConflictFilesInputSchema = z.object({
-  prNumber: z.number(),
+  prNumber: z.number().optional(),
   baseBranch: z.string().optional().default("main"),
+  headRef: z.string().optional().default("HEAD"),
 });
 
 export async function getMergeConflictFilesHandler(args: z.infer<typeof GetMergeConflictFilesInputSchema>) {
-  const result = await runCommand("td-cli", [
-    "gh", "merge-conflicts", args.prNumber.toString(),
-    "--base", args.baseBranch || "main"
-  ]);
-
-  if (result.exitCode !== 0) {
-    throw new Error(`Failed to get merge conflicts: ${result.stderr}`);
-  }
-
-  const output = JSON.parse(result.stdout);
-  if (output.status === "error") {
-    throw new Error(`Failed to get merge conflicts: ${output.message}`);
-  }
+  const baseBranch = args.baseBranch || "main";
+  const headRef = args.headRef || "HEAD";
+  const conflictFiles = await getMergeConflictFiles(baseBranch, headRef);
 
   return {
-    prNumber: output.prNumber,
-    baseBranch: output.baseBranch,
-    headRef: output.headRef,
-    conflictFiles: output.conflictFiles,
-    commandLog: output.commandLog
+    prNumber: args.prNumber,
+    baseBranch,
+    headRef,
+    conflictFiles,
   };
 }

package/src/tools/github.get_pr.ts

@@ -12,7 +12,7 @@ export async function getPrHandler(args: z.infer<typeof GetPrInputSchema>) {
   const params = GetPrInputSchema.parse(args);
   const targetRepo = resolveRepo(params.repo);
 
-  const result = await runCommand("td-cli", ["gh", "view", params.prNumber.toString(), "--repo", targetRepo]);
+  const result = await runCommand("td", ["gh", "view", params.prNumber.toString(), "--repo", targetRepo]);
 
   if (result.exitCode !== 0) {
     throw new Error(`Failed to get PR details: ${sanitizeError(result.stderr)}`);

package/src/tools/github.get_pr_commits.ts

@@ -0,0 +1,27 @@
+import { z } from "zod";
+import { runCommand } from "../lib/shell.js";
+import { resolveRepo } from "../lib/git.js";
+import { sanitizeError } from "../lib/error_utils.js";
+
+const GetPrCommitsInputSchema = z.object({
+  prNumber: z.number().describe("The number of the PR to list commits for."),
+  repo: z.string().optional().describe("The target repository override (e.g. org/repo)."),
+});
+
+export async function getPrCommitsHandler(args: z.infer<typeof GetPrCommitsInputSchema>) {
+  const params = GetPrCommitsInputSchema.parse(args);
+  const targetRepo = resolveRepo(params.repo);
+
+  const result = await runCommand("td", ["gh", "list-pr-commits", "--pr-number", params.prNumber.toString(), "--repo", targetRepo]);
+
+  if (result.exitCode !== 0) {
+    throw new Error(`Failed to list PR commits: ${sanitizeError(result.stderr)}`);
+  }
+
+  const output = JSON.parse(result.stdout);
+  if (output.status === "error") {
+    throw new Error(`Failed to list PR commits: ${output.message}`);
+  }
+
+  return { commits: output.commits };
+}

package/src/tools/github.get_pr_diff.ts

@@ -11,7 +11,7 @@ export async function getPrDiffHandler(args: z.infer<typeof GetPrDiffInputSchema
   const params = GetPrDiffInputSchema.parse(args);
   const targetRepo = resolveRepo(params.repo);
 
-  const result = await runCommand("td-cli", [
+  const result = await runCommand("td", [
     "gh", "pr-diff", params.prNumber.toString(), "--repo", targetRepo
   ]);
 

package/src/tools/github.get_pr_reviews.test.ts

@@ -0,0 +1,110 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+import { getPrReviewsHandler } from "./github.get_pr_reviews.js";
+import { getPrCommitsHandler } from "./github.get_pr_commits.js";
+import { getWorkflowRunsHandler } from "./repo.get_workflow_runs.js";
+import * as shell from "../lib/shell.js";
+
+vi.mock("../lib/shell.js", () => ({
+  runCommand: vi.fn(),
+}));
+
+describe("MCP GitHub tools", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  describe("getPrReviewsHandler", () => {
+    it("should fetch PR reviews via td gh read-pr-reviews", async () => {
+      vi.mocked(shell.runCommand).mockResolvedValueOnce({
+        stdout: JSON.stringify({
+          status: "success",
+          reviews: [{ id: 1, state: "APPROVED", body: "LGTM" }],
+          review_count: 1,
+        }),
+        stderr: "",
+        exitCode: 0,
+        durationMs: 10,
+        command: "td gh read-pr-reviews",
+      });
+
+      const res = await getPrReviewsHandler({ prNumber: 819, repo: "arii/boomtick" });
+
+      expect(shell.runCommand).toHaveBeenCalledWith("td", [
+        "gh",
+        "read-pr-reviews",
+        "--pr-number",
+        "819",
+        "--repo",
+        "arii/boomtick",
+      ]);
+      expect(res).toEqual({
+        reviews: [{ id: 1, state: "APPROVED", body: "LGTM" }],
+        reviewCount: 1,
+      });
+    });
+  });
+
+  describe("getPrCommitsHandler", () => {
+    it("should list PR commits via td gh list-pr-commits", async () => {
+      vi.mocked(shell.runCommand).mockResolvedValueOnce({
+        stdout: JSON.stringify({
+          status: "success",
+          commits: [{ sha: "abc1234", author: "Dev", message: "feat: test" }],
+        }),
+        stderr: "",
+        exitCode: 0,
+        durationMs: 10,
+        command: "td gh list-pr-commits",
+      });
+
+      const res = await getPrCommitsHandler({ prNumber: 819, repo: "arii/boomtick" });
+
+      expect(shell.runCommand).toHaveBeenCalledWith("td", [
+        "gh",
+        "list-pr-commits",
+        "--pr-number",
+        "819",
+        "--repo",
+        "arii/boomtick",
+      ]);
+      expect(res).toEqual({
+        commits: [{ sha: "abc1234", author: "Dev", message: "feat: test" }],
+      });
+    });
+  });
+
+  describe("getWorkflowRunsHandler", () => {
+    it("should fetch workflow runs via td repo workflow-runs", async () => {
+      vi.mocked(shell.runCommand).mockResolvedValueOnce({
+        stdout: JSON.stringify({
+          status: "success",
+          workflow_runs: [{ id: 10, name: "CI" }],
+        }),
+        stderr: "",
+        exitCode: 0,
+        durationMs: 10,
+        command: "td repo workflow-runs",
+      });
+
+      const res = await getWorkflowRunsHandler({
+        branch: "main",
+        event: "push",
+        repo: "arii/boomtick",
+      });
+
+      expect(shell.runCommand).toHaveBeenCalledWith("td", [
+        "repo",
+        "workflow-runs",
+        "--repo",
+        "arii/boomtick",
+        "--branch",
+        "main",
+        "--event",
+        "push",
+      ]);
+      expect(res).toEqual({
+        workflowRuns: [{ id: 10, name: "CI" }],
+      });
+    });
+  });
+});