mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-01-09 17:52:30 +08:00
test: add comprehensive E2E tests for all major features
- Error handling tests (API errors, rate limits, network timeout, truncated XML) - Multi-turn conversation tests (sequential requests, history preservation) - File upload tests (upload button, file preview, sending with message) - Theme switching tests (dark mode toggle, persistence, system preference) - Language switching tests (EN/JA/ZH, persistence, locale URLs) - Iframe interaction tests (draw.io loading, toolbar, diagram rendering) - Copy/paste tests (chat input, XML input, special characters) - History restore tests (new chat, persistence, browser navigation)
This commit is contained in:
190
tests/e2e/multi-turn.spec.ts
Normal file
190
tests/e2e/multi-turn.spec.ts
Normal file
@@ -0,0 +1,190 @@
|
||||
import { expect, test } from "@playwright/test"
|
||||
|
||||
// Helper to create SSE response
|
||||
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"
|
||||
)
|
||||
}
|
||||
|
||||
// Helper for text-only response
|
||||
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"
|
||||
)
|
||||
}
|
||||
|
||||
test.describe("Multi-turn Conversation", () => {
|
||||
test("handles multiple diagram requests in sequence", async ({ page }) => {
|
||||
let requestCount = 0
|
||||
await page.route("**/api/chat", async (route) => {
|
||||
requestCount++
|
||||
const xml =
|
||||
requestCount === 1
|
||||
? `<mxCell id="box1" value="First" style="rounded=1;" vertex="1" parent="1"><mxGeometry x="100" y="100" width="100" height="40" as="geometry"/></mxCell>`
|
||||
: `<mxCell id="box2" value="Second" style="rounded=1;" vertex="1" parent="1"><mxGeometry x="100" y="200" width="100" height="40" as="geometry"/></mxCell>`
|
||||
await route.fulfill({
|
||||
status: 200,
|
||||
contentType: "text/event-stream",
|
||||
body: createMockSSEResponse(
|
||||
xml,
|
||||
`Creating diagram ${requestCount}...`,
|
||||
),
|
||||
})
|
||||
})
|
||||
|
||||
await page.goto("/", { waitUntil: "networkidle" })
|
||||
await page
|
||||
.locator("iframe")
|
||||
.waitFor({ state: "visible", timeout: 30000 })
|
||||
|
||||
const chatInput = page.locator('textarea[aria-label="Chat input"]')
|
||||
await expect(chatInput).toBeVisible({ timeout: 10000 })
|
||||
|
||||
// First request
|
||||
await chatInput.fill("Draw first box")
|
||||
await chatInput.press("ControlOrMeta+Enter")
|
||||
await expect(page.locator('text="Creating diagram 1..."')).toBeVisible({
|
||||
timeout: 15000,
|
||||
})
|
||||
|
||||
// Second request
|
||||
await chatInput.fill("Draw second box")
|
||||
await chatInput.press("ControlOrMeta+Enter")
|
||||
await expect(page.locator('text="Creating diagram 2..."')).toBeVisible({
|
||||
timeout: 15000,
|
||||
})
|
||||
|
||||
// Both messages should be visible
|
||||
await expect(page.locator('text="Draw first box"')).toBeVisible()
|
||||
await expect(page.locator('text="Draw second box"')).toBeVisible()
|
||||
})
|
||||
|
||||
test("preserves conversation history", async ({ page }) => {
|
||||
let requestCount = 0
|
||||
await page.route("**/api/chat", async (route) => {
|
||||
requestCount++
|
||||
const request = route.request()
|
||||
const body = JSON.parse(request.postData() || "{}")
|
||||
|
||||
// Verify messages array grows with each request
|
||||
if (requestCount === 2) {
|
||||
// Second request should have previous messages
|
||||
expect(body.messages?.length).toBeGreaterThan(1)
|
||||
}
|
||||
|
||||
await route.fulfill({
|
||||
status: 200,
|
||||
contentType: "text/event-stream",
|
||||
body: createTextOnlyResponse(`Response ${requestCount}`),
|
||||
})
|
||||
})
|
||||
|
||||
await page.goto("/", { waitUntil: "networkidle" })
|
||||
await page
|
||||
.locator("iframe")
|
||||
.waitFor({ state: "visible", timeout: 30000 })
|
||||
|
||||
const chatInput = page.locator('textarea[aria-label="Chat input"]')
|
||||
await expect(chatInput).toBeVisible({ timeout: 10000 })
|
||||
|
||||
// First message
|
||||
await chatInput.fill("Hello")
|
||||
await chatInput.press("ControlOrMeta+Enter")
|
||||
await expect(page.locator('text="Response 1"')).toBeVisible({
|
||||
timeout: 15000,
|
||||
})
|
||||
|
||||
// Second message (should include history)
|
||||
await chatInput.fill("Follow up question")
|
||||
await chatInput.press("ControlOrMeta+Enter")
|
||||
await expect(page.locator('text="Response 2"')).toBeVisible({
|
||||
timeout: 15000,
|
||||
})
|
||||
})
|
||||
|
||||
test("can continue after a text-only response", async ({ page }) => {
|
||||
let requestCount = 0
|
||||
await page.route("**/api/chat", async (route) => {
|
||||
requestCount++
|
||||
if (requestCount === 1) {
|
||||
// First: text-only explanation
|
||||
await route.fulfill({
|
||||
status: 200,
|
||||
contentType: "text/event-stream",
|
||||
body: createTextOnlyResponse(
|
||||
"I understand. Let me explain the architecture first.",
|
||||
),
|
||||
})
|
||||
} else {
|
||||
// Second: diagram generation
|
||||
const xml = `<mxCell id="arch" value="Architecture" style="rounded=1;" vertex="1" parent="1"><mxGeometry x="100" y="100" width="120" height="50" as="geometry"/></mxCell>`
|
||||
await route.fulfill({
|
||||
status: 200,
|
||||
contentType: "text/event-stream",
|
||||
body: createMockSSEResponse(xml, "Here is the diagram:"),
|
||||
})
|
||||
}
|
||||
})
|
||||
|
||||
await page.goto("/", { waitUntil: "networkidle" })
|
||||
await page
|
||||
.locator("iframe")
|
||||
.waitFor({ state: "visible", timeout: 30000 })
|
||||
|
||||
const chatInput = page.locator('textarea[aria-label="Chat input"]')
|
||||
await expect(chatInput).toBeVisible({ timeout: 10000 })
|
||||
|
||||
// Ask for explanation first
|
||||
await chatInput.fill("Explain the architecture")
|
||||
await chatInput.press("ControlOrMeta+Enter")
|
||||
await expect(
|
||||
page.locator(
|
||||
'text="I understand. Let me explain the architecture first."',
|
||||
),
|
||||
).toBeVisible({ timeout: 15000 })
|
||||
|
||||
// Then ask for diagram
|
||||
await chatInput.fill("Now show it as a diagram")
|
||||
await chatInput.press("ControlOrMeta+Enter")
|
||||
await expect(page.locator('text="Complete"')).toBeVisible({
|
||||
timeout: 15000,
|
||||
})
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user