test(vscode): add code-action-provider tests for action set and command IDs

2026-09-15 12:50:50 +08:00 · 2026-02-20 13:40:57 +01:00
parent bc3bb46aa0
commit 1bad5eb067
1 changed files with 125 additions and 0 deletions
@@ -0,0 +1,125 @@
+import { describe, it, expect, mock } from "bun:test"
+
+const makeAction = (title: string, kind: string) => ({ title, kind })
+const makeKind = (value: string) => ({
+  value,
+  append: (v: string) => makeKind(`${value}.${v}`),
+})
+
+const QuickFix = makeKind("quickfix")
+const RefactorRewrite = makeKind("refactor.rewrite")
+
+const mockVscode = {
+  CodeAction: class {
+    command?: { command: string; title: string }
+    isPreferred?: boolean
+    constructor(
+      public title: string,
+      public kind: { value: string },
+    ) {}
+  },
+  CodeActionKind: {
+    QuickFix,
+    RefactorRewrite,
+  },
+}
+
+mock.module("vscode", () => mockVscode)
+
+const { KiloCodeActionProvider } = await import("../../src/services/code-actions/code-action-provider")
+
+const provider = new KiloCodeActionProvider()
+
+function makeRange(isEmpty: boolean) {
+  return { isEmpty }
+}
+
+function makeContext(diagnosticCount: number) {
+  return { diagnostics: Array.from({ length: diagnosticCount }) }
+}
+
+describe("KiloCodeActionProvider", () => {
+  describe("provideCodeActions", () => {
+    it("returns empty array when range is empty", () => {
+      const result = provider.provideCodeActions({} as never, makeRange(true) as never, makeContext(0) as never)
+      expect(result).toEqual([])
+    })
+
+    it("returns empty array when range is empty even with diagnostics", () => {
+      const result = provider.provideCodeActions({} as never, makeRange(true) as never, makeContext(3) as never)
+      expect(result).toEqual([])
+    })
+
+    describe("non-empty range, no diagnostics", () => {
+      it("returns Add, Explain, Improve actions", () => {
+        const result = provider.provideCodeActions({} as never, makeRange(false) as never, makeContext(0) as never)
+        const titles = result.map((a) => a.title)
+        expect(titles).toContain("Add to Kilo Code")
+        expect(titles).toContain("Explain with Kilo Code")
+        expect(titles).toContain("Improve with Kilo Code")
+      })
+
+      it("does not include Fix action", () => {
+        const result = provider.provideCodeActions({} as never, makeRange(false) as never, makeContext(0) as never)
+        expect(result.map((a) => a.title)).not.toContain("Fix with Kilo Code")
+      })
+
+      it("returns exactly 3 actions", () => {
+        const result = provider.provideCodeActions({} as never, makeRange(false) as never, makeContext(0) as never)
+        expect(result).toHaveLength(3)
+      })
+
+      it("uses correct command IDs", () => {
+        const result = provider.provideCodeActions({} as never, makeRange(false) as never, makeContext(0) as never)
+        const commands = result.map((a) => a.command?.command)
+        expect(commands).toContain("kilo-code.new.addToContext")
+        expect(commands).toContain("kilo-code.new.explainCode")
+        expect(commands).toContain("kilo-code.new.improveCode")
+      })
+
+      it("no action is preferred", () => {
+        const result = provider.provideCodeActions({} as never, makeRange(false) as never, makeContext(0) as never)
+        expect(result.every((a) => !a.isPreferred)).toBe(true)
+      })
+    })
+
+    describe("non-empty range, with diagnostics", () => {
+      it("returns Add and Fix actions", () => {
+        const result = provider.provideCodeActions({} as never, makeRange(false) as never, makeContext(2) as never)
+        const titles = result.map((a) => a.title)
+        expect(titles).toContain("Add to Kilo Code")
+        expect(titles).toContain("Fix with Kilo Code")
+      })
+
+      it("does not include Explain or Improve actions", () => {
+        const result = provider.provideCodeActions({} as never, makeRange(false) as never, makeContext(1) as never)
+        const titles = result.map((a) => a.title)
+        expect(titles).not.toContain("Explain with Kilo Code")
+        expect(titles).not.toContain("Improve with Kilo Code")
+      })
+
+      it("returns exactly 2 actions", () => {
+        const result = provider.provideCodeActions({} as never, makeRange(false) as never, makeContext(1) as never)
+        expect(result).toHaveLength(2)
+      })
+
+      it("Fix action is preferred", () => {
+        const result = provider.provideCodeActions({} as never, makeRange(false) as never, makeContext(1) as never)
+        const fix = result.find((a) => a.title === "Fix with Kilo Code")
+        expect(fix?.isPreferred).toBe(true)
+      })
+
+      it("Fix action uses QuickFix kind", () => {
+        const result = provider.provideCodeActions({} as never, makeRange(false) as never, makeContext(1) as never)
+        const fix = result.find((a) => a.title === "Fix with Kilo Code")
+        expect(fix?.kind.value).toBe("quickfix")
+      })
+
+      it("uses correct Fix command ID", () => {
+        const result = provider.provideCodeActions({} as never, makeRange(false) as never, makeContext(1) as never)
+        const fix = result.find((a) => a.title === "Fix with Kilo Code")
+        expect(fix?.command?.command).toBe("kilo-code.new.fixCode")
+      })
+    })
+  })
+})