mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-01-09 17:52:30 +08:00
test: add Vitest and Playwright testing infrastructure
- Add Vitest for unit tests (39 tests) - cached-responses.test.ts - ai-providers.test.ts - chat-helpers.test.ts - utils.test.ts - Add Playwright for E2E tests (3 smoke tests) - Homepage load - Japanese locale - Settings dialog - Add CI workflow (.github/workflows/test.yml) - Add vitest.config.mts and playwright.config.ts - Update .gitignore for test artifacts
This commit is contained in:
dayuan.jiang
246
tests/unit/chat-helpers.test.ts
Normal file
246
tests/unit/chat-helpers.test.ts
Normal file
@@ -0,0 +1,246 @@
|
||||
// @vitest-environment node
|
||||
import { describe, expect, it } from "vitest"
|
||||
|
||||
// Test the helper functions from chat route
|
||||
// These are re-implemented here for testing since they're not exported
|
||||
|
||||
const MAX_FILE_SIZE = 2 * 1024 * 1024 // 2MB
|
||||
const MAX_FILES = 5
|
||||
|
||||
function validateFileParts(messages: any[]): {
|
||||
valid: boolean
|
||||
error?: string
|
||||
} {
|
||||
const lastMessage = messages[messages.length - 1]
|
||||
const fileParts =
|
||||
lastMessage?.parts?.filter((p: any) => p.type === "file") || []
|
||||
|
||||
if (fileParts.length > MAX_FILES) {
|
||||
return {
|
||||
valid: false,
|
||||
error: `Too many files. Maximum ${MAX_FILES} allowed.`,
|
||||
}
|
||||
}
|
||||
|
||||
for (const filePart of fileParts) {
|
||||
if (filePart.url?.startsWith("data:")) {
|
||||
const base64Data = filePart.url.split(",")[1]
|
||||
if (base64Data) {
|
||||
const sizeInBytes = Math.ceil((base64Data.length * 3) / 4)
|
||||
if (sizeInBytes > MAX_FILE_SIZE) {
|
||||
return {
|
||||
valid: false,
|
||||
error: `File exceeds ${MAX_FILE_SIZE / 1024 / 1024}MB limit.`,
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return { valid: true }
|
||||
}
|
||||
|
||||
function isMinimalDiagram(xml: string): boolean {
|
||||
const stripped = xml.replace(/\s/g, "")
|
||||
return !stripped.includes('id="2"')
|
||||
}
|
||||
|
||||
function replaceHistoricalToolInputs(messages: any[]): any[] {
|
||||
return messages.map((msg) => {
|
||||
if (msg.role !== "assistant" || !Array.isArray(msg.content)) {
|
||||
return msg
|
||||
}
|
||||
const replacedContent = msg.content
|
||||
.map((part: any) => {
|
||||
if (part.type === "tool-call") {
|
||||
const toolName = part.toolName
|
||||
if (
|
||||
!part.input ||
|
||||
typeof part.input !== "object" ||
|
||||
Object.keys(part.input).length === 0
|
||||
) {
|
||||
return null
|
||||
}
|
||||
if (
|
||||
toolName === "display_diagram" ||
|
||||
toolName === "edit_diagram"
|
||||
) {
|
||||
return {
|
||||
...part,
|
||||
input: {
|
||||
placeholder:
|
||||
"[XML content replaced - see current diagram XML in system context]",
|
||||
},
|
||||
}
|
||||
}
|
||||
}
|
||||
return part
|
||||
})
|
||||
.filter(Boolean)
|
||||
return { ...msg, content: replacedContent }
|
||||
})
|
||||
}
|
||||
|
||||
describe("validateFileParts", () => {
|
||||
it("returns valid for no files", () => {
|
||||
const messages = [
|
||||
{ role: "user", parts: [{ type: "text", text: "hello" }] },
|
||||
]
|
||||
expect(validateFileParts(messages)).toEqual({ valid: true })
|
||||
})
|
||||
|
||||
it("returns valid for files under limit", () => {
|
||||
const smallBase64 = btoa("x".repeat(100))
|
||||
const messages = [
|
||||
{
|
||||
role: "user",
|
||||
parts: [
|
||||
{
|
||||
type: "file",
|
||||
url: `data:image/png;base64,${smallBase64}`,
|
||||
},
|
||||
],
|
||||
},
|
||||
]
|
||||
expect(validateFileParts(messages)).toEqual({ valid: true })
|
||||
})
|
||||
|
||||
it("returns error for too many files", () => {
|
||||
const messages = [
|
||||
{
|
||||
role: "user",
|
||||
parts: Array(6)
|
||||
.fill(null)
|
||||
.map(() => ({
|
||||
type: "file",
|
||||
url: "data:image/png;base64,abc",
|
||||
})),
|
||||
},
|
||||
]
|
||||
const result = validateFileParts(messages)
|
||||
expect(result.valid).toBe(false)
|
||||
expect(result.error).toContain("Too many files")
|
||||
})
|
||||
|
||||
it("returns error for file exceeding size limit", () => {
|
||||
// Create base64 that decodes to > 2MB
|
||||
const largeBase64 = btoa("x".repeat(3 * 1024 * 1024))
|
||||
const messages = [
|
||||
{
|
||||
role: "user",
|
||||
parts: [
|
||||
{
|
||||
type: "file",
|
||||
url: `data:image/png;base64,${largeBase64}`,
|
||||
},
|
||||
],
|
||||
},
|
||||
]
|
||||
const result = validateFileParts(messages)
|
||||
expect(result.valid).toBe(false)
|
||||
expect(result.error).toContain("exceeds")
|
||||
})
|
||||
})
|
||||
|
||||
describe("isMinimalDiagram", () => {
|
||||
it("returns true for empty diagram", () => {
|
||||
const xml = '<mxCell id="0"/><mxCell id="1" parent="0"/>'
|
||||
expect(isMinimalDiagram(xml)).toBe(true)
|
||||
})
|
||||
|
||||
it("returns false for diagram with content", () => {
|
||||
const xml =
|
||||
'<mxCell id="0"/><mxCell id="1" parent="0"/><mxCell id="2" value="Hello"/>'
|
||||
expect(isMinimalDiagram(xml)).toBe(false)
|
||||
})
|
||||
|
||||
it("handles whitespace correctly", () => {
|
||||
const xml = ' <mxCell id="0"/> <mxCell id="1" parent="0"/> '
|
||||
expect(isMinimalDiagram(xml)).toBe(true)
|
||||
})
|
||||
})
|
||||
|
||||
describe("replaceHistoricalToolInputs", () => {
|
||||
it("replaces display_diagram tool inputs with placeholder", () => {
|
||||
const messages = [
|
||||
{
|
||||
role: "assistant",
|
||||
content: [
|
||||
{
|
||||
type: "tool-call",
|
||||
toolName: "display_diagram",
|
||||
input: { xml: "<mxCell...>" },
|
||||
},
|
||||
],
|
||||
},
|
||||
]
|
||||
const result = replaceHistoricalToolInputs(messages)
|
||||
expect(result[0].content[0].input.placeholder).toContain(
|
||||
"XML content replaced",
|
||||
)
|
||||
})
|
||||
|
||||
it("replaces edit_diagram tool inputs with placeholder", () => {
|
||||
const messages = [
|
||||
{
|
||||
role: "assistant",
|
||||
content: [
|
||||
{
|
||||
type: "tool-call",
|
||||
toolName: "edit_diagram",
|
||||
input: { operations: [] },
|
||||
},
|
||||
],
|
||||
},
|
||||
]
|
||||
const result = replaceHistoricalToolInputs(messages)
|
||||
expect(result[0].content[0].input.placeholder).toContain(
|
||||
"XML content replaced",
|
||||
)
|
||||
})
|
||||
|
||||
it("removes tool calls with invalid inputs", () => {
|
||||
const messages = [
|
||||
{
|
||||
role: "assistant",
|
||||
content: [
|
||||
{
|
||||
type: "tool-call",
|
||||
toolName: "display_diagram",
|
||||
input: {},
|
||||
},
|
||||
{
|
||||
type: "tool-call",
|
||||
toolName: "display_diagram",
|
||||
input: null,
|
||||
},
|
||||
],
|
||||
},
|
||||
]
|
||||
const result = replaceHistoricalToolInputs(messages)
|
||||
expect(result[0].content).toHaveLength(0)
|
||||
})
|
||||
|
||||
it("preserves non-assistant messages", () => {
|
||||
const messages = [{ role: "user", content: "hello" }]
|
||||
const result = replaceHistoricalToolInputs(messages)
|
||||
expect(result).toEqual(messages)
|
||||
})
|
||||
|
||||
it("preserves other tool calls", () => {
|
||||
const messages = [
|
||||
{
|
||||
role: "assistant",
|
||||
content: [
|
||||
{
|
||||
type: "tool-call",
|
||||
toolName: "other_tool",
|
||||
input: { foo: "bar" },
|
||||
},
|
||||
],
|
||||
},
|
||||
]
|
||||
const result = replaceHistoricalToolInputs(messages)
|
||||
expect(result[0].content[0].input).toEqual({ foo: "bar" })
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user