refactor(server): canonicalize service API (#31049)

packages/tui/test/prompt/traits.test.ts

@@ -0,0 +1,25 @@
+import { describe, expect, test } from "bun:test"
+import { computePromptTraits } from "../../src/prompt/traits"
+
+describe("computePromptTraits", () => {
+  test("normal mode without autocomplete only captures tab", () => {
+    const traits = computePromptTraits({ mode: "normal", autocompleteVisible: false })
+    expect(traits.capture).toEqual(["tab"])
+    expect(traits.suspend).toBeUndefined()
+    expect(traits.status).toBeUndefined()
+  })
+
+  test("normal mode with autocomplete captures navigation keys", () => {
+    const traits = computePromptTraits({ mode: "normal", autocompleteVisible: true })
+    expect(traits.capture).toEqual(["escape", "navigate", "submit", "tab"])
+    expect(traits.suspend).toBeUndefined()
+    expect(traits.status).toBeUndefined()
+  })
+
+  test("shell mode disables capture and labels the prompt without suspending", () => {
+    const traits = computePromptTraits({ mode: "shell", autocompleteVisible: false })
+    expect(traits.capture).toBeUndefined()
+    expect(traits.suspend).toBeUndefined()
+    expect(traits.status).toBe("SHELL")
+  })
+})