feat(tmux): add session-kill utility with runner tests
This commit is contained in:
@@ -0,0 +1,63 @@
|
|||||||
|
import { beforeEach, describe, expect, it, mock } from "bun:test"
|
||||||
|
|
||||||
|
import type { TmuxCommandResult } from "../runner"
|
||||||
|
|
||||||
|
const sessionKillSpecifier = import.meta.resolve("./session-kill")
|
||||||
|
const environmentSpecifier = import.meta.resolve("./environment")
|
||||||
|
const loggerSpecifier = import.meta.resolve("../../logger")
|
||||||
|
const runnerSpecifier = import.meta.resolve("../runner")
|
||||||
|
const tmuxPathResolverSpecifier = import.meta.resolve("../../../tools/interactive-bash/tmux-path-resolver")
|
||||||
|
|
||||||
|
const runTmuxCommandMock = mock(async (): Promise<TmuxCommandResult> => ({
|
||||||
|
success: true,
|
||||||
|
output: "",
|
||||||
|
stdout: "",
|
||||||
|
stderr: "",
|
||||||
|
exitCode: 0,
|
||||||
|
}))
|
||||||
|
const isInsideTmuxMock = mock((): boolean => true)
|
||||||
|
const getTmuxPathMock = mock(async (): Promise<string | undefined> => "sh")
|
||||||
|
const logMock = mock(() => undefined)
|
||||||
|
|
||||||
|
async function loadKillTmuxSessionIfExists(): Promise<typeof import("./session-kill").killTmuxSessionIfExists> {
|
||||||
|
const module = await import(`${sessionKillSpecifier}?test=${crypto.randomUUID()}`)
|
||||||
|
return module.killTmuxSessionIfExists
|
||||||
|
}
|
||||||
|
|
||||||
|
function registerModuleMocks(): void {
|
||||||
|
mock.module(environmentSpecifier, () => ({ isInsideTmux: isInsideTmuxMock }))
|
||||||
|
mock.module(loggerSpecifier, () => ({ log: logMock }))
|
||||||
|
mock.module(runnerSpecifier, () => ({ runTmuxCommand: runTmuxCommandMock }))
|
||||||
|
mock.module(tmuxPathResolverSpecifier, () => ({ getTmuxPath: getTmuxPathMock }))
|
||||||
|
}
|
||||||
|
|
||||||
|
describe("killTmuxSessionIfExists runner integration", () => {
|
||||||
|
beforeEach(() => {
|
||||||
|
registerModuleMocks()
|
||||||
|
runTmuxCommandMock.mockClear()
|
||||||
|
isInsideTmuxMock.mockClear()
|
||||||
|
getTmuxPathMock.mockClear()
|
||||||
|
logMock.mockClear()
|
||||||
|
|
||||||
|
runTmuxCommandMock
|
||||||
|
.mockResolvedValueOnce({ success: true, output: "", stdout: "", stderr: "", exitCode: 0 })
|
||||||
|
.mockResolvedValueOnce({ success: true, output: "", stdout: "", stderr: "", exitCode: 0 })
|
||||||
|
isInsideTmuxMock.mockReturnValue(true)
|
||||||
|
getTmuxPathMock.mockResolvedValue("sh")
|
||||||
|
})
|
||||||
|
|
||||||
|
it("#given session exists #when killTmuxSessionIfExists called #then delegates has-session and kill-session to shared runner", async () => {
|
||||||
|
// given
|
||||||
|
const killTmuxSessionIfExists = await loadKillTmuxSessionIfExists()
|
||||||
|
|
||||||
|
// when
|
||||||
|
const result = await killTmuxSessionIfExists("omo-agents")
|
||||||
|
|
||||||
|
// then
|
||||||
|
expect(result).toBe(true)
|
||||||
|
expect(runTmuxCommandMock.mock.calls).toEqual([
|
||||||
|
["sh", ["has-session", "-t", "omo-agents"]],
|
||||||
|
["sh", ["kill-session", "-t", "omo-agents"]],
|
||||||
|
])
|
||||||
|
})
|
||||||
|
})
|
||||||
@@ -1,134 +1,84 @@
|
|||||||
import { beforeEach, describe, expect, it, mock } from "bun:test"
|
import { beforeEach, describe, expect, it, mock } from "bun:test"
|
||||||
|
|
||||||
type KillTmuxSessionIfExists = typeof import("./session-kill").killTmuxSessionIfExists
|
import type { TmuxCommandResult } from "../runner"
|
||||||
|
|
||||||
type SpawnCall = {
|
|
||||||
command: string[]
|
|
||||||
options: {
|
|
||||||
stdout?: string
|
|
||||||
stderr?: string
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type FakeSubprocess = {
|
|
||||||
exited: Promise<number>
|
|
||||||
stdout: ReadableStream<Uint8Array>
|
|
||||||
stderr: ReadableStream<Uint8Array>
|
|
||||||
}
|
|
||||||
|
|
||||||
const spawnCalls: SpawnCall[] = []
|
|
||||||
const queuedProcesses: FakeSubprocess[] = []
|
|
||||||
|
|
||||||
function createStream(chunks: string[] = []): ReadableStream<Uint8Array> {
|
|
||||||
const textEncoder = new TextEncoder()
|
|
||||||
|
|
||||||
return new ReadableStream({
|
|
||||||
start(controller) {
|
|
||||||
for (const chunk of chunks) {
|
|
||||||
controller.enqueue(textEncoder.encode(chunk))
|
|
||||||
}
|
|
||||||
|
|
||||||
controller.close()
|
|
||||||
},
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
function createProcess(exitCode: number, output: { stdout?: string[]; stderr?: string[] } = {}): FakeSubprocess {
|
|
||||||
return {
|
|
||||||
exited: Promise.resolve(exitCode),
|
|
||||||
stdout: createStream(output.stdout),
|
|
||||||
stderr: createStream(output.stderr),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const spawnMock = mock((command: string[], options: { stdout?: string; stderr?: string } = {}) => {
|
|
||||||
spawnCalls.push({ command, options })
|
|
||||||
|
|
||||||
const process = queuedProcesses.shift()
|
|
||||||
if (!process) {
|
|
||||||
throw new Error(`No fake subprocess configured for ${command.join(" ")}`)
|
|
||||||
}
|
|
||||||
|
|
||||||
return process
|
|
||||||
})
|
|
||||||
|
|
||||||
const isInsideTmuxMock = mock((): boolean => true)
|
|
||||||
const getTmuxPathMock = mock(async (): Promise<string | undefined> => "tmux")
|
|
||||||
const logMock = mock(() => undefined)
|
|
||||||
|
|
||||||
const sessionKillSpecifier = import.meta.resolve("./session-kill")
|
const sessionKillSpecifier = import.meta.resolve("./session-kill")
|
||||||
const environmentSpecifier = import.meta.resolve("./environment")
|
const environmentSpecifier = import.meta.resolve("./environment")
|
||||||
const loggerSpecifier = import.meta.resolve("../../logger")
|
const loggerSpecifier = import.meta.resolve("../../logger")
|
||||||
const spawnProcessSpecifier = import.meta.resolve("./spawn-process")
|
const runnerSpecifier = import.meta.resolve("../runner")
|
||||||
const tmuxPathResolverSpecifier = import.meta.resolve("../../../tools/interactive-bash/tmux-path-resolver")
|
const tmuxPathResolverSpecifier = import.meta.resolve("../../../tools/interactive-bash/tmux-path-resolver")
|
||||||
|
|
||||||
async function loadKillTmuxSessionIfExists(): Promise<typeof KillTmuxSessionIfExists> {
|
const runTmuxCommandMock = mock(async (): Promise<TmuxCommandResult> => ({
|
||||||
|
success: true,
|
||||||
|
output: "",
|
||||||
|
stdout: "",
|
||||||
|
stderr: "",
|
||||||
|
exitCode: 0,
|
||||||
|
}))
|
||||||
|
const isInsideTmuxMock = mock((): boolean => true)
|
||||||
|
const getTmuxPathMock = mock(async (): Promise<string | undefined> => "tmux")
|
||||||
|
const logMock = mock(() => undefined)
|
||||||
|
|
||||||
|
async function loadKillTmuxSessionIfExists(): Promise<typeof import("./session-kill").killTmuxSessionIfExists> {
|
||||||
const module = await import(`${sessionKillSpecifier}?test=${crypto.randomUUID()}`)
|
const module = await import(`${sessionKillSpecifier}?test=${crypto.randomUUID()}`)
|
||||||
return module.killTmuxSessionIfExists
|
return module.killTmuxSessionIfExists
|
||||||
}
|
}
|
||||||
|
|
||||||
function registerModuleMocks(): void {
|
function registerModuleMocks(): void {
|
||||||
mock.module(spawnProcessSpecifier, () => ({ spawn: spawnMock }))
|
|
||||||
mock.module(environmentSpecifier, () => ({ isInsideTmux: isInsideTmuxMock }))
|
mock.module(environmentSpecifier, () => ({ isInsideTmux: isInsideTmuxMock }))
|
||||||
mock.module(tmuxPathResolverSpecifier, () => ({ getTmuxPath: getTmuxPathMock }))
|
|
||||||
mock.module(loggerSpecifier, () => ({ log: logMock }))
|
mock.module(loggerSpecifier, () => ({ log: logMock }))
|
||||||
|
mock.module(runnerSpecifier, () => ({ runTmuxCommand: runTmuxCommandMock }))
|
||||||
|
mock.module(tmuxPathResolverSpecifier, () => ({ getTmuxPath: getTmuxPathMock }))
|
||||||
}
|
}
|
||||||
|
|
||||||
describe("killTmuxSessionIfExists", () => {
|
describe("killTmuxSessionIfExists", () => {
|
||||||
beforeEach(() => {
|
beforeEach(() => {
|
||||||
registerModuleMocks()
|
registerModuleMocks()
|
||||||
spawnCalls.length = 0
|
runTmuxCommandMock.mockClear()
|
||||||
queuedProcesses.length = 0
|
|
||||||
spawnMock.mockClear()
|
|
||||||
isInsideTmuxMock.mockClear()
|
isInsideTmuxMock.mockClear()
|
||||||
getTmuxPathMock.mockClear()
|
getTmuxPathMock.mockClear()
|
||||||
logMock.mockClear()
|
logMock.mockClear()
|
||||||
|
|
||||||
isInsideTmuxMock.mockImplementation((): boolean => true)
|
runTmuxCommandMock.mockResolvedValue({
|
||||||
getTmuxPathMock.mockImplementation(async (): Promise<string | undefined> => "tmux")
|
success: true,
|
||||||
|
output: "",
|
||||||
|
stdout: "",
|
||||||
|
stderr: "",
|
||||||
|
exitCode: 0,
|
||||||
|
})
|
||||||
|
isInsideTmuxMock.mockReturnValue(true)
|
||||||
|
getTmuxPathMock.mockResolvedValue("tmux")
|
||||||
})
|
})
|
||||||
|
|
||||||
it("#given omo-agents session exists #when killTmuxSessionIfExists called #then kill-session invoked and returns true", async () => {
|
it("#given omo-agents session exists #when killTmuxSessionIfExists called #then kill-session invoked and returns true", async () => {
|
||||||
// given
|
// given
|
||||||
const killTmuxSessionIfExists = await loadKillTmuxSessionIfExists()
|
const killTmuxSessionIfExists = await loadKillTmuxSessionIfExists()
|
||||||
queuedProcesses.push(createProcess(0), createProcess(0, { stdout: ["killed"], stderr: [] }))
|
|
||||||
|
|
||||||
// when
|
// when
|
||||||
const result = await killTmuxSessionIfExists("omo-agents")
|
const result = await killTmuxSessionIfExists("omo-agents")
|
||||||
|
|
||||||
// then
|
// then
|
||||||
expect(result).toBe(true)
|
expect(result).toBe(true)
|
||||||
expect(spawnCalls).toEqual([
|
expect(runTmuxCommandMock.mock.calls).toEqual([
|
||||||
{
|
["tmux", ["has-session", "-t", "omo-agents"]],
|
||||||
command: ["tmux", "has-session", "-t", "omo-agents"],
|
["tmux", ["kill-session", "-t", "omo-agents"]],
|
||||||
options: { stdout: "ignore", stderr: "ignore" },
|
|
||||||
},
|
|
||||||
{
|
|
||||||
command: ["tmux", "kill-session", "-t", "omo-agents"],
|
|
||||||
options: { stdout: "pipe", stderr: "pipe" },
|
|
||||||
},
|
|
||||||
])
|
])
|
||||||
})
|
})
|
||||||
|
|
||||||
it("#given omo-agents session does NOT exist (has-session exits non-zero) #when killTmuxSessionIfExists called #then NO kill-session invocation and returns false", async () => {
|
it("#given omo-agents session does NOT exist #when killTmuxSessionIfExists called #then NO kill-session invocation and returns false", async () => {
|
||||||
// given
|
// given
|
||||||
const killTmuxSessionIfExists = await loadKillTmuxSessionIfExists()
|
const killTmuxSessionIfExists = await loadKillTmuxSessionIfExists()
|
||||||
queuedProcesses.push(createProcess(1))
|
runTmuxCommandMock.mockResolvedValueOnce({ success: false, output: "", stdout: "", stderr: "", exitCode: 1 })
|
||||||
|
|
||||||
// when
|
// when
|
||||||
const result = await killTmuxSessionIfExists("omo-agents")
|
const result = await killTmuxSessionIfExists("omo-agents")
|
||||||
|
|
||||||
// then
|
// then
|
||||||
expect(result).toBe(false)
|
expect(result).toBe(false)
|
||||||
expect(spawnCalls).toEqual([
|
expect(runTmuxCommandMock.mock.calls).toEqual([["tmux", ["has-session", "-t", "omo-agents"]]])
|
||||||
{
|
|
||||||
command: ["tmux", "has-session", "-t", "omo-agents"],
|
|
||||||
options: { stdout: "ignore", stderr: "ignore" },
|
|
||||||
},
|
|
||||||
])
|
|
||||||
})
|
})
|
||||||
|
|
||||||
it("#given not inside tmux #when killTmuxSessionIfExists called #then returns false without any spawn", async () => {
|
it("#given not inside tmux #when killTmuxSessionIfExists called #then returns false without runner calls", async () => {
|
||||||
// given
|
// given
|
||||||
const killTmuxSessionIfExists = await loadKillTmuxSessionIfExists()
|
const killTmuxSessionIfExists = await loadKillTmuxSessionIfExists()
|
||||||
isInsideTmuxMock.mockReturnValue(false)
|
isInsideTmuxMock.mockReturnValue(false)
|
||||||
@@ -138,11 +88,10 @@ describe("killTmuxSessionIfExists", () => {
|
|||||||
|
|
||||||
// then
|
// then
|
||||||
expect(result).toBe(false)
|
expect(result).toBe(false)
|
||||||
expect(spawnCalls).toHaveLength(0)
|
expect(runTmuxCommandMock).not.toHaveBeenCalled()
|
||||||
expect(getTmuxPathMock).toHaveBeenCalledTimes(0)
|
|
||||||
})
|
})
|
||||||
|
|
||||||
it("#given tmux not found #when killTmuxSessionIfExists called #then returns false without spawn", async () => {
|
it("#given tmux not found #when killTmuxSessionIfExists called #then returns false without runner calls", async () => {
|
||||||
// given
|
// given
|
||||||
const killTmuxSessionIfExists = await loadKillTmuxSessionIfExists()
|
const killTmuxSessionIfExists = await loadKillTmuxSessionIfExists()
|
||||||
getTmuxPathMock.mockResolvedValue(undefined)
|
getTmuxPathMock.mockResolvedValue(undefined)
|
||||||
@@ -152,22 +101,21 @@ describe("killTmuxSessionIfExists", () => {
|
|||||||
|
|
||||||
// then
|
// then
|
||||||
expect(result).toBe(false)
|
expect(result).toBe(false)
|
||||||
expect(spawnCalls).toHaveLength(0)
|
expect(runTmuxCommandMock).not.toHaveBeenCalled()
|
||||||
})
|
})
|
||||||
|
|
||||||
it("#given kill-session itself fails (e.g., race between has-session and kill) #when killTmuxSessionIfExists called #then returns false but does not throw", async () => {
|
it("#given kill-session itself fails #when killTmuxSessionIfExists called #then returns false but does not throw", async () => {
|
||||||
// given
|
// given
|
||||||
const killTmuxSessionIfExists = await loadKillTmuxSessionIfExists()
|
const killTmuxSessionIfExists = await loadKillTmuxSessionIfExists()
|
||||||
queuedProcesses.push(
|
runTmuxCommandMock
|
||||||
createProcess(0),
|
.mockResolvedValueOnce({ success: true, output: "", stdout: "", stderr: "", exitCode: 0 })
|
||||||
createProcess(1, { stdout: [], stderr: ["no session"] }),
|
.mockResolvedValueOnce({ success: false, output: "", stdout: "", stderr: "no session", exitCode: 1 })
|
||||||
)
|
|
||||||
|
|
||||||
// when
|
// when
|
||||||
const result = await killTmuxSessionIfExists("omo-agents")
|
const result = await killTmuxSessionIfExists("omo-agents")
|
||||||
|
|
||||||
// then
|
// then
|
||||||
expect(result).toBe(false)
|
expect(result).toBe(false)
|
||||||
expect(spawnCalls).toHaveLength(2)
|
expect(runTmuxCommandMock).toHaveBeenCalledTimes(2)
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|||||||
@@ -1,13 +1,9 @@
|
|||||||
async function readStream(stream: ReadableStream<Uint8Array> | null | undefined): Promise<string> {
|
|
||||||
return stream ? new Response(stream).text() : ""
|
|
||||||
}
|
|
||||||
|
|
||||||
export async function killTmuxSessionIfExists(sessionName: string): Promise<boolean> {
|
export async function killTmuxSessionIfExists(sessionName: string): Promise<boolean> {
|
||||||
const [{ log }, { isInsideTmux }, { getTmuxPath }, { spawn }] = await Promise.all([
|
const [{ log }, { isInsideTmux }, { getTmuxPath }, { runTmuxCommand }] = await Promise.all([
|
||||||
import("../../logger"),
|
import("../../logger"),
|
||||||
import("./environment"),
|
import("./environment"),
|
||||||
import("../../../tools/interactive-bash/tmux-path-resolver"),
|
import("../../../tools/interactive-bash/tmux-path-resolver"),
|
||||||
import("./spawn-process"),
|
import("../runner"),
|
||||||
])
|
])
|
||||||
|
|
||||||
if (!isInsideTmux()) {
|
if (!isInsideTmux()) {
|
||||||
@@ -21,28 +17,21 @@ export async function killTmuxSessionIfExists(sessionName: string): Promise<bool
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
const hasSessionProcess = spawn([tmux, "has-session", "-t", sessionName], {
|
const hasSessionResult = await runTmuxCommand(tmux, ["has-session", "-t", sessionName])
|
||||||
stdout: "ignore",
|
|
||||||
stderr: "ignore",
|
|
||||||
})
|
|
||||||
|
|
||||||
if ((await hasSessionProcess.exited) !== 0) {
|
if (hasSessionResult.exitCode !== 0) {
|
||||||
log("[killTmuxSessionIfExists] SKIP: session not found", { sessionName })
|
log("[killTmuxSessionIfExists] SKIP: session not found", { sessionName })
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
const killSessionProcess = spawn([tmux, "kill-session", "-t", sessionName], {
|
const killSessionResult = await runTmuxCommand(tmux, ["kill-session", "-t", sessionName])
|
||||||
stdout: "pipe",
|
|
||||||
stderr: "pipe",
|
|
||||||
})
|
|
||||||
const [, stderr, exitCode] = await Promise.all([
|
|
||||||
readStream(killSessionProcess.stdout),
|
|
||||||
readStream(killSessionProcess.stderr),
|
|
||||||
killSessionProcess.exited,
|
|
||||||
])
|
|
||||||
|
|
||||||
if (exitCode !== 0) {
|
if (killSessionResult.exitCode !== 0) {
|
||||||
log("[killTmuxSessionIfExists] FAILED", { sessionName, exitCode, stderr: stderr.trim() })
|
log("[killTmuxSessionIfExists] FAILED", {
|
||||||
|
sessionName,
|
||||||
|
exitCode: killSessionResult.exitCode,
|
||||||
|
stderr: killSessionResult.stderr.trim(),
|
||||||
|
})
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user