98 lines
2.7 KiB
TypeScript
98 lines
2.7 KiB
TypeScript
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
import { PermissionKey, SystemRole } from "@capakraken/shared";
|
|
|
|
vi.mock("@capakraken/application", async (importOriginal) => {
|
|
const actual = await importOriginal<typeof import("@capakraken/application")>();
|
|
return {
|
|
...actual,
|
|
approveEstimateVersion: vi.fn(),
|
|
cloneEstimate: vi.fn(),
|
|
createEstimateExport: vi.fn(),
|
|
createEstimatePlanningHandoff: vi.fn(),
|
|
createEstimateRevision: vi.fn(),
|
|
getDashboardBudgetForecast: vi.fn().mockResolvedValue([]),
|
|
getDashboardPeakTimes: vi.fn().mockResolvedValue([]),
|
|
getEstimateById: vi.fn(),
|
|
listAssignmentBookings: vi.fn().mockResolvedValue([]),
|
|
submitEstimateVersion: vi.fn(),
|
|
updateEstimateDraft: vi.fn(),
|
|
};
|
|
});
|
|
|
|
import { getEstimateById } from "@capakraken/application";
|
|
|
|
import { executeTool } from "../router/assistant-tools.js";
|
|
import {
|
|
createToolContext,
|
|
resetEstimateToolMocks,
|
|
} from "./assistant-tools-estimate-test-helpers.js";
|
|
|
|
describe("assistant estimate read tools", () => {
|
|
beforeEach(() => {
|
|
resetEstimateToolMocks();
|
|
});
|
|
|
|
it("returns stable read errors for missing estimates and versions", async () => {
|
|
vi.mocked(getEstimateById).mockResolvedValueOnce(null as never);
|
|
|
|
const detailCtx = createToolContext(
|
|
{},
|
|
{
|
|
userRole: SystemRole.CONTROLLER,
|
|
permissions: [PermissionKey.VIEW_COSTS],
|
|
},
|
|
);
|
|
|
|
const detailResult = await executeTool(
|
|
"get_estimate_detail",
|
|
JSON.stringify({ estimateId: "missing_estimate" }),
|
|
detailCtx,
|
|
);
|
|
|
|
expect(JSON.parse(detailResult.content)).toEqual({
|
|
error: "Estimate not found with the given criteria.",
|
|
});
|
|
|
|
const versionsCtx = createToolContext(
|
|
{
|
|
estimate: {
|
|
findUnique: vi.fn().mockResolvedValue(null),
|
|
},
|
|
},
|
|
{ userRole: SystemRole.CONTROLLER },
|
|
);
|
|
|
|
const versionsResult = await executeTool(
|
|
"list_estimate_versions",
|
|
JSON.stringify({ estimateId: "missing_estimate" }),
|
|
versionsCtx,
|
|
);
|
|
|
|
expect(JSON.parse(versionsResult.content)).toEqual({
|
|
error: "Estimate not found with the given criteria.",
|
|
});
|
|
|
|
const snapshotCtx = createToolContext(
|
|
{
|
|
estimate: {
|
|
findUnique: vi.fn().mockResolvedValue(null),
|
|
},
|
|
},
|
|
{
|
|
userRole: SystemRole.CONTROLLER,
|
|
permissions: [PermissionKey.VIEW_COSTS],
|
|
},
|
|
);
|
|
|
|
const snapshotResult = await executeTool(
|
|
"get_estimate_version_snapshot",
|
|
JSON.stringify({ estimateId: "missing_estimate", versionId: "missing_version" }),
|
|
snapshotCtx,
|
|
);
|
|
|
|
expect(JSON.parse(snapshotResult.content)).toEqual({
|
|
error: "Estimate version not found with the given criteria.",
|
|
});
|
|
});
|
|
});
|