mirror of
https://github.com/samiyev/puaros.git
synced 2025-12-27 23:06:54 +05:00
0f2ed5b301
- Add ISessionStorage interface and RedisSessionStorage implementation - Add ContextManager for token budget and compression - Add StartSession, HandleMessage, UndoChange use cases - Update CHANGELOG and TODO documentation - 88 new tests (1174 total), 97.73% coverage
548 lines
21 KiB
TypeScript
548 lines
21 KiB
TypeScript
import { describe, it, expect, vi, beforeEach } from "vitest"
|
|
import {
|
|
RunTestsTool,
|
|
type RunTestsResult,
|
|
type TestRunner,
|
|
} from "../../../../../src/infrastructure/tools/run/RunTestsTool.js"
|
|
import type { ToolContext } from "../../../../../src/domain/services/ITool.js"
|
|
import type { IStorage } from "../../../../../src/domain/services/IStorage.js"
|
|
|
|
function createMockStorage(): IStorage {
|
|
return {
|
|
getFile: vi.fn(),
|
|
setFile: vi.fn(),
|
|
deleteFile: vi.fn(),
|
|
getAllFiles: vi.fn().mockResolvedValue(new Map()),
|
|
getFileCount: vi.fn().mockResolvedValue(0),
|
|
getAST: vi.fn(),
|
|
setAST: vi.fn(),
|
|
deleteAST: vi.fn(),
|
|
getAllASTs: vi.fn().mockResolvedValue(new Map()),
|
|
getMeta: vi.fn(),
|
|
setMeta: vi.fn(),
|
|
deleteMeta: vi.fn(),
|
|
getAllMetas: vi.fn().mockResolvedValue(new Map()),
|
|
getSymbolIndex: vi.fn().mockResolvedValue(new Map()),
|
|
setSymbolIndex: vi.fn(),
|
|
getDepsGraph: vi.fn().mockResolvedValue({ imports: new Map(), importedBy: new Map() }),
|
|
setDepsGraph: vi.fn(),
|
|
getProjectConfig: vi.fn(),
|
|
setProjectConfig: vi.fn(),
|
|
connect: vi.fn(),
|
|
disconnect: vi.fn(),
|
|
isConnected: vi.fn().mockReturnValue(true),
|
|
clear: vi.fn(),
|
|
} as unknown as IStorage
|
|
}
|
|
|
|
function createMockContext(storage?: IStorage): ToolContext {
|
|
return {
|
|
projectRoot: "/test/project",
|
|
storage: storage ?? createMockStorage(),
|
|
requestConfirmation: vi.fn().mockResolvedValue(true),
|
|
onProgress: vi.fn(),
|
|
}
|
|
}
|
|
|
|
type ExecResult = { stdout: string; stderr: string }
|
|
type ExecFn = (command: string, options: Record<string, unknown>) => Promise<ExecResult>
|
|
|
|
function createMockExec(options: {
|
|
stdout?: string
|
|
stderr?: string
|
|
error?: Error & { code?: number; stdout?: string; stderr?: string }
|
|
}): ExecFn {
|
|
return vi.fn().mockImplementation(() => {
|
|
if (options.error) {
|
|
return Promise.reject(options.error)
|
|
}
|
|
return Promise.resolve({
|
|
stdout: options.stdout ?? "",
|
|
stderr: options.stderr ?? "",
|
|
})
|
|
})
|
|
}
|
|
|
|
function createMockFsAccess(existingFiles: string[]): typeof import("fs/promises").access {
|
|
return vi.fn().mockImplementation((filePath: string) => {
|
|
for (const file of existingFiles) {
|
|
if (filePath.endsWith(file)) {
|
|
return Promise.resolve()
|
|
}
|
|
}
|
|
return Promise.reject(new Error("ENOENT"))
|
|
})
|
|
}
|
|
|
|
function createMockFsReadFile(
|
|
packageJson?: Record<string, unknown>,
|
|
): typeof import("fs/promises").readFile {
|
|
return vi.fn().mockImplementation((filePath: string) => {
|
|
if (filePath.endsWith("package.json") && packageJson) {
|
|
return Promise.resolve(JSON.stringify(packageJson))
|
|
}
|
|
return Promise.reject(new Error("ENOENT"))
|
|
})
|
|
}
|
|
|
|
describe("RunTestsTool", () => {
|
|
let tool: RunTestsTool
|
|
|
|
beforeEach(() => {
|
|
tool = new RunTestsTool()
|
|
})
|
|
|
|
describe("metadata", () => {
|
|
it("should have correct name", () => {
|
|
expect(tool.name).toBe("run_tests")
|
|
})
|
|
|
|
it("should have correct category", () => {
|
|
expect(tool.category).toBe("run")
|
|
})
|
|
|
|
it("should not require confirmation", () => {
|
|
expect(tool.requiresConfirmation).toBe(false)
|
|
})
|
|
|
|
it("should have correct parameters", () => {
|
|
expect(tool.parameters).toHaveLength(3)
|
|
expect(tool.parameters[0].name).toBe("path")
|
|
expect(tool.parameters[1].name).toBe("filter")
|
|
expect(tool.parameters[2].name).toBe("watch")
|
|
})
|
|
|
|
it("should have description", () => {
|
|
expect(tool.description).toContain("test")
|
|
expect(tool.description).toContain("vitest")
|
|
})
|
|
})
|
|
|
|
describe("validateParams", () => {
|
|
it("should return null for empty params", () => {
|
|
expect(tool.validateParams({})).toBeNull()
|
|
})
|
|
|
|
it("should return null for valid params", () => {
|
|
expect(tool.validateParams({ path: "src", filter: "login", watch: true })).toBeNull()
|
|
})
|
|
|
|
it("should return error for invalid path", () => {
|
|
expect(tool.validateParams({ path: 123 })).toContain("path")
|
|
})
|
|
|
|
it("should return error for invalid filter", () => {
|
|
expect(tool.validateParams({ filter: 123 })).toContain("filter")
|
|
})
|
|
|
|
it("should return error for invalid watch", () => {
|
|
expect(tool.validateParams({ watch: "yes" })).toContain("watch")
|
|
})
|
|
})
|
|
|
|
describe("detectTestRunner", () => {
|
|
it("should detect vitest from config file", async () => {
|
|
const fsAccess = createMockFsAccess(["vitest.config.ts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(undefined, fsAccess, fsReadFile)
|
|
|
|
const runner = await toolWithMocks.detectTestRunner("/test/project")
|
|
|
|
expect(runner).toBe("vitest")
|
|
})
|
|
|
|
it("should detect vitest from .js config", async () => {
|
|
const fsAccess = createMockFsAccess(["vitest.config.js"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(undefined, fsAccess, fsReadFile)
|
|
|
|
const runner = await toolWithMocks.detectTestRunner("/test/project")
|
|
|
|
expect(runner).toBe("vitest")
|
|
})
|
|
|
|
it("should detect vitest from .mts config", async () => {
|
|
const fsAccess = createMockFsAccess(["vitest.config.mts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(undefined, fsAccess, fsReadFile)
|
|
|
|
const runner = await toolWithMocks.detectTestRunner("/test/project")
|
|
|
|
expect(runner).toBe("vitest")
|
|
})
|
|
|
|
it("should detect jest from config file", async () => {
|
|
const fsAccess = createMockFsAccess(["jest.config.js"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(undefined, fsAccess, fsReadFile)
|
|
|
|
const runner = await toolWithMocks.detectTestRunner("/test/project")
|
|
|
|
expect(runner).toBe("jest")
|
|
})
|
|
|
|
it("should detect vitest from devDependencies", async () => {
|
|
const fsAccess = createMockFsAccess([])
|
|
const fsReadFile = createMockFsReadFile({
|
|
devDependencies: { vitest: "^1.0.0" },
|
|
})
|
|
const toolWithMocks = new RunTestsTool(undefined, fsAccess, fsReadFile)
|
|
|
|
const runner = await toolWithMocks.detectTestRunner("/test/project")
|
|
|
|
expect(runner).toBe("vitest")
|
|
})
|
|
|
|
it("should detect jest from devDependencies", async () => {
|
|
const fsAccess = createMockFsAccess([])
|
|
const fsReadFile = createMockFsReadFile({
|
|
devDependencies: { jest: "^29.0.0" },
|
|
})
|
|
const toolWithMocks = new RunTestsTool(undefined, fsAccess, fsReadFile)
|
|
|
|
const runner = await toolWithMocks.detectTestRunner("/test/project")
|
|
|
|
expect(runner).toBe("jest")
|
|
})
|
|
|
|
it("should detect mocha from devDependencies", async () => {
|
|
const fsAccess = createMockFsAccess([])
|
|
const fsReadFile = createMockFsReadFile({
|
|
devDependencies: { mocha: "^10.0.0" },
|
|
})
|
|
const toolWithMocks = new RunTestsTool(undefined, fsAccess, fsReadFile)
|
|
|
|
const runner = await toolWithMocks.detectTestRunner("/test/project")
|
|
|
|
expect(runner).toBe("mocha")
|
|
})
|
|
|
|
it("should detect npm test script as fallback", async () => {
|
|
const fsAccess = createMockFsAccess([])
|
|
const fsReadFile = createMockFsReadFile({
|
|
scripts: { test: "node test.js" },
|
|
})
|
|
const toolWithMocks = new RunTestsTool(undefined, fsAccess, fsReadFile)
|
|
|
|
const runner = await toolWithMocks.detectTestRunner("/test/project")
|
|
|
|
expect(runner).toBe("npm")
|
|
})
|
|
|
|
it("should return null when no runner found", async () => {
|
|
const fsAccess = createMockFsAccess([])
|
|
const fsReadFile = createMockFsReadFile({})
|
|
const toolWithMocks = new RunTestsTool(undefined, fsAccess, fsReadFile)
|
|
|
|
const runner = await toolWithMocks.detectTestRunner("/test/project")
|
|
|
|
expect(runner).toBeNull()
|
|
})
|
|
})
|
|
|
|
describe("buildCommand", () => {
|
|
describe("vitest", () => {
|
|
it("should build basic vitest command", () => {
|
|
const cmd = tool.buildCommand("vitest")
|
|
expect(cmd).toBe("npx vitest run")
|
|
})
|
|
|
|
it("should build vitest with path", () => {
|
|
const cmd = tool.buildCommand("vitest", "src/tests")
|
|
expect(cmd).toBe("npx vitest run src/tests")
|
|
})
|
|
|
|
it("should build vitest with filter", () => {
|
|
const cmd = tool.buildCommand("vitest", undefined, "login")
|
|
expect(cmd).toBe('npx vitest run -t "login"')
|
|
})
|
|
|
|
it("should build vitest with watch", () => {
|
|
const cmd = tool.buildCommand("vitest", undefined, undefined, true)
|
|
expect(cmd).toBe("npx vitest")
|
|
})
|
|
|
|
it("should build vitest with all options", () => {
|
|
const cmd = tool.buildCommand("vitest", "src", "login", true)
|
|
expect(cmd).toBe('npx vitest src -t "login"')
|
|
})
|
|
})
|
|
|
|
describe("jest", () => {
|
|
it("should build basic jest command", () => {
|
|
const cmd = tool.buildCommand("jest")
|
|
expect(cmd).toBe("npx jest")
|
|
})
|
|
|
|
it("should build jest with path", () => {
|
|
const cmd = tool.buildCommand("jest", "src/tests")
|
|
expect(cmd).toBe("npx jest src/tests")
|
|
})
|
|
|
|
it("should build jest with filter", () => {
|
|
const cmd = tool.buildCommand("jest", undefined, "login")
|
|
expect(cmd).toBe('npx jest -t "login"')
|
|
})
|
|
|
|
it("should build jest with watch", () => {
|
|
const cmd = tool.buildCommand("jest", undefined, undefined, true)
|
|
expect(cmd).toBe("npx jest --watch")
|
|
})
|
|
})
|
|
|
|
describe("mocha", () => {
|
|
it("should build basic mocha command", () => {
|
|
const cmd = tool.buildCommand("mocha")
|
|
expect(cmd).toBe("npx mocha")
|
|
})
|
|
|
|
it("should build mocha with path", () => {
|
|
const cmd = tool.buildCommand("mocha", "test/")
|
|
expect(cmd).toBe("npx mocha test/")
|
|
})
|
|
|
|
it("should build mocha with filter", () => {
|
|
const cmd = tool.buildCommand("mocha", undefined, "login")
|
|
expect(cmd).toBe('npx mocha --grep "login"')
|
|
})
|
|
|
|
it("should build mocha with watch", () => {
|
|
const cmd = tool.buildCommand("mocha", undefined, undefined, true)
|
|
expect(cmd).toBe("npx mocha --watch")
|
|
})
|
|
})
|
|
|
|
describe("npm", () => {
|
|
it("should build basic npm test command", () => {
|
|
const cmd = tool.buildCommand("npm")
|
|
expect(cmd).toBe("npm test")
|
|
})
|
|
|
|
it("should build npm test with path", () => {
|
|
const cmd = tool.buildCommand("npm", "src/tests")
|
|
expect(cmd).toBe("npm test -- src/tests")
|
|
})
|
|
|
|
it("should build npm test with filter", () => {
|
|
const cmd = tool.buildCommand("npm", undefined, "login")
|
|
expect(cmd).toBe('npm test -- "login"')
|
|
})
|
|
})
|
|
})
|
|
|
|
describe("execute", () => {
|
|
describe("no runner detected", () => {
|
|
it("should return error when no runner found", async () => {
|
|
const execFn = createMockExec({})
|
|
const fsAccess = createMockFsAccess([])
|
|
const fsReadFile = createMockFsReadFile({})
|
|
const toolWithMocks = new RunTestsTool(execFn, fsAccess, fsReadFile)
|
|
const ctx = createMockContext()
|
|
|
|
const result = await toolWithMocks.execute({}, ctx)
|
|
|
|
expect(result.success).toBe(false)
|
|
expect(result.error).toContain("No test runner detected")
|
|
})
|
|
})
|
|
|
|
describe("successful tests", () => {
|
|
it("should return success when tests pass", async () => {
|
|
const execFn = createMockExec({
|
|
stdout: "All tests passed",
|
|
stderr: "",
|
|
})
|
|
const fsAccess = createMockFsAccess(["vitest.config.ts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(execFn, fsAccess, fsReadFile)
|
|
const ctx = createMockContext()
|
|
|
|
const result = await toolWithMocks.execute({}, ctx)
|
|
|
|
expect(result.success).toBe(true)
|
|
const data = result.data as RunTestsResult
|
|
expect(data.passed).toBe(true)
|
|
expect(data.exitCode).toBe(0)
|
|
expect(data.runner).toBe("vitest")
|
|
expect(data.stdout).toContain("All tests passed")
|
|
})
|
|
|
|
it("should include command in result", async () => {
|
|
const execFn = createMockExec({})
|
|
const fsAccess = createMockFsAccess(["vitest.config.ts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(execFn, fsAccess, fsReadFile)
|
|
const ctx = createMockContext()
|
|
|
|
const result = await toolWithMocks.execute({}, ctx)
|
|
|
|
expect(result.success).toBe(true)
|
|
const data = result.data as RunTestsResult
|
|
expect(data.command).toBe("npx vitest run")
|
|
})
|
|
|
|
it("should include duration in result", async () => {
|
|
const execFn = createMockExec({})
|
|
const fsAccess = createMockFsAccess(["vitest.config.ts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(execFn, fsAccess, fsReadFile)
|
|
const ctx = createMockContext()
|
|
|
|
const result = await toolWithMocks.execute({}, ctx)
|
|
|
|
expect(result.success).toBe(true)
|
|
const data = result.data as RunTestsResult
|
|
expect(data.durationMs).toBeGreaterThanOrEqual(0)
|
|
})
|
|
})
|
|
|
|
describe("failing tests", () => {
|
|
it("should return success=true but passed=false for test failures", async () => {
|
|
const error = Object.assign(new Error("Tests failed"), {
|
|
code: 1,
|
|
stdout: "1 test failed",
|
|
stderr: "AssertionError",
|
|
})
|
|
const execFn = createMockExec({ error })
|
|
const fsAccess = createMockFsAccess(["vitest.config.ts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(execFn, fsAccess, fsReadFile)
|
|
const ctx = createMockContext()
|
|
|
|
const result = await toolWithMocks.execute({}, ctx)
|
|
|
|
expect(result.success).toBe(true)
|
|
const data = result.data as RunTestsResult
|
|
expect(data.passed).toBe(false)
|
|
expect(data.exitCode).toBe(1)
|
|
expect(data.stdout).toContain("1 test failed")
|
|
expect(data.stderr).toContain("AssertionError")
|
|
})
|
|
})
|
|
|
|
describe("with options", () => {
|
|
it("should pass path to command", async () => {
|
|
const execFn = createMockExec({})
|
|
const fsAccess = createMockFsAccess(["vitest.config.ts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(execFn, fsAccess, fsReadFile)
|
|
const ctx = createMockContext()
|
|
|
|
const result = await toolWithMocks.execute({ path: "src/tests" }, ctx)
|
|
|
|
expect(result.success).toBe(true)
|
|
const data = result.data as RunTestsResult
|
|
expect(data.command).toContain("src/tests")
|
|
})
|
|
|
|
it("should pass filter to command", async () => {
|
|
const execFn = createMockExec({})
|
|
const fsAccess = createMockFsAccess(["vitest.config.ts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(execFn, fsAccess, fsReadFile)
|
|
const ctx = createMockContext()
|
|
|
|
const result = await toolWithMocks.execute({ filter: "login" }, ctx)
|
|
|
|
expect(result.success).toBe(true)
|
|
const data = result.data as RunTestsResult
|
|
expect(data.command).toContain('-t "login"')
|
|
})
|
|
|
|
it("should pass watch option", async () => {
|
|
const execFn = createMockExec({})
|
|
const fsAccess = createMockFsAccess(["vitest.config.ts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(execFn, fsAccess, fsReadFile)
|
|
const ctx = createMockContext()
|
|
|
|
const result = await toolWithMocks.execute({ watch: true }, ctx)
|
|
|
|
expect(result.success).toBe(true)
|
|
const data = result.data as RunTestsResult
|
|
expect(data.command).toBe("npx vitest")
|
|
expect(data.command).not.toContain("run")
|
|
})
|
|
})
|
|
|
|
describe("error handling", () => {
|
|
it("should handle timeout", async () => {
|
|
const error = new Error("Command timed out")
|
|
const execFn = createMockExec({ error })
|
|
const fsAccess = createMockFsAccess(["vitest.config.ts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(execFn, fsAccess, fsReadFile)
|
|
const ctx = createMockContext()
|
|
|
|
const result = await toolWithMocks.execute({}, ctx)
|
|
|
|
expect(result.success).toBe(false)
|
|
expect(result.error).toContain("timed out")
|
|
})
|
|
|
|
it("should handle generic errors", async () => {
|
|
const error = new Error("Something went wrong")
|
|
const execFn = createMockExec({ error })
|
|
const fsAccess = createMockFsAccess(["vitest.config.ts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(execFn, fsAccess, fsReadFile)
|
|
const ctx = createMockContext()
|
|
|
|
const result = await toolWithMocks.execute({}, ctx)
|
|
|
|
expect(result.success).toBe(false)
|
|
expect(result.error).toBe("Something went wrong")
|
|
})
|
|
})
|
|
|
|
describe("exec options", () => {
|
|
it("should run in project root", async () => {
|
|
const execFn = createMockExec({})
|
|
const fsAccess = createMockFsAccess(["vitest.config.ts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(execFn, fsAccess, fsReadFile)
|
|
const ctx = createMockContext()
|
|
ctx.projectRoot = "/my/project"
|
|
|
|
await toolWithMocks.execute({}, ctx)
|
|
|
|
expect(execFn).toHaveBeenCalledWith(
|
|
expect.any(String),
|
|
expect.objectContaining({ cwd: "/my/project" }),
|
|
)
|
|
})
|
|
|
|
it("should set CI environment variable", async () => {
|
|
const execFn = createMockExec({})
|
|
const fsAccess = createMockFsAccess(["vitest.config.ts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(execFn, fsAccess, fsReadFile)
|
|
const ctx = createMockContext()
|
|
|
|
await toolWithMocks.execute({}, ctx)
|
|
|
|
expect(execFn).toHaveBeenCalledWith(
|
|
expect.any(String),
|
|
expect.objectContaining({
|
|
env: expect.objectContaining({ CI: "true" }),
|
|
}),
|
|
)
|
|
})
|
|
})
|
|
|
|
describe("call id", () => {
|
|
it("should generate unique call id", async () => {
|
|
const execFn = createMockExec({})
|
|
const fsAccess = createMockFsAccess(["vitest.config.ts"])
|
|
const fsReadFile = createMockFsReadFile()
|
|
const toolWithMocks = new RunTestsTool(execFn, fsAccess, fsReadFile)
|
|
const ctx = createMockContext()
|
|
|
|
const result = await toolWithMocks.execute({}, ctx)
|
|
|
|
expect(result.callId).toMatch(/^run_tests-\d+$/)
|
|
})
|
|
})
|
|
})
|
|
})
|