mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-01-07 00:32:28 +08:00
6ce7e3378b
- Create tests/e2e/lib/helpers.ts with shared SSE mock functions - Add proper error UI assertions to error-handling.spec.ts - Remove waitForTimeout calls in favor of real assertions - Update 6 test files to use shared helpers
89 lines
2.5 KiB
TypeScript
89 lines
2.5 KiB
TypeScript
/**
|
|
* Shared test helpers for E2E tests
|
|
*/
|
|
|
|
/**
|
|
* Creates a mock SSE response for the chat API
|
|
* Format matches AI SDK UI message stream protocol
|
|
*/
|
|
export function createMockSSEResponse(
|
|
xml: string,
|
|
text: string,
|
|
toolName = "display_diagram",
|
|
) {
|
|
const messageId = `msg_${Date.now()}`
|
|
const toolCallId = `call_${Date.now()}`
|
|
const textId = `text_${Date.now()}`
|
|
|
|
const events = [
|
|
{ type: "start", messageId },
|
|
{ type: "text-start", id: textId },
|
|
{ type: "text-delta", id: textId, delta: text },
|
|
{ type: "text-end", id: textId },
|
|
{ type: "tool-input-start", toolCallId, toolName },
|
|
{ type: "tool-input-available", toolCallId, toolName, input: { xml } },
|
|
{
|
|
type: "tool-output-available",
|
|
toolCallId,
|
|
output: "Successfully displayed the diagram",
|
|
},
|
|
{ type: "finish" },
|
|
]
|
|
|
|
return (
|
|
events.map((e) => `data: ${JSON.stringify(e)}\n\n`).join("") +
|
|
"data: [DONE]\n\n"
|
|
)
|
|
}
|
|
|
|
/**
|
|
* Creates a text-only SSE response (no tool call)
|
|
*/
|
|
export function createTextOnlyResponse(text: string) {
|
|
const messageId = `msg_${Date.now()}`
|
|
const textId = `text_${Date.now()}`
|
|
|
|
const events = [
|
|
{ type: "start", messageId },
|
|
{ type: "text-start", id: textId },
|
|
{ type: "text-delta", id: textId, delta: text },
|
|
{ type: "text-end", id: textId },
|
|
{ type: "finish" },
|
|
]
|
|
|
|
return (
|
|
events.map((e) => `data: ${JSON.stringify(e)}\n\n`).join("") +
|
|
"data: [DONE]\n\n"
|
|
)
|
|
}
|
|
|
|
/**
|
|
* Creates a mock SSE response with a tool error
|
|
*/
|
|
export function createToolErrorResponse(text: string, errorMessage: string) {
|
|
const messageId = `msg_${Date.now()}`
|
|
const toolCallId = `call_${Date.now()}`
|
|
const textId = `text_${Date.now()}`
|
|
|
|
const events = [
|
|
{ type: "start", messageId },
|
|
{ type: "text-start", id: textId },
|
|
{ type: "text-delta", id: textId, delta: text },
|
|
{ type: "text-end", id: textId },
|
|
{ type: "tool-input-start", toolCallId, toolName: "display_diagram" },
|
|
{
|
|
type: "tool-input-available",
|
|
toolCallId,
|
|
toolName: "display_diagram",
|
|
input: { xml: "<invalid>" },
|
|
},
|
|
{ type: "tool-output-error", toolCallId, error: errorMessage },
|
|
{ type: "finish" },
|
|
]
|
|
|
|
return (
|
|
events.map((e) => `data: ${JSON.stringify(e)}\n\n`).join("") +
|
|
"data: [DONE]\n\n"
|
|
)
|
|
}
|