116 lines
4.7 KiB
TypeScript
116 lines
4.7 KiB
TypeScript
import { expect, test } from "@playwright/test"
|
|
import {
|
|
assistantMessage,
|
|
partUpdated,
|
|
setupTimeline,
|
|
status,
|
|
textPart,
|
|
userMessage,
|
|
} from "../performance/timeline-stability/fixture"
|
|
|
|
test("keeps one connection open while delivering multiple events", async ({ page }) => {
|
|
const timeline = await setupTimeline(page)
|
|
|
|
const first = await timeline.transport.send(partUpdated(textPart("prt_transport_first", "first event")))
|
|
const second = await timeline.transport.send(partUpdated(textPart("prt_transport_second", "second event")))
|
|
|
|
await timeline.waitForPart("prt_transport_first")
|
|
await timeline.waitForPart("prt_transport_second")
|
|
expect(first.connectionID).toBe(second.connectionID)
|
|
expect(await timeline.transport.connections()).toHaveLength(1)
|
|
expect(await timeline.transport.acknowledgements()).toHaveLength(2)
|
|
})
|
|
|
|
test("delivers a burst from one stream chunk", async ({ page }) => {
|
|
const timeline = await setupTimeline(page)
|
|
const acknowledgements = await timeline.transport.burst([
|
|
partUpdated(textPart("prt_transport_burst_a", "burst a")),
|
|
partUpdated(textPart("prt_transport_burst_b", "burst b")),
|
|
])
|
|
|
|
await timeline.waitForPart("prt_transport_burst_a")
|
|
await timeline.waitForPart("prt_transport_burst_b")
|
|
expect(acknowledgements.map((item) => item.chunkCount)).toEqual([1, 1])
|
|
expect(new Set(acknowledgements.map((item) => item.deliveryID)).size).toBe(2)
|
|
})
|
|
|
|
test("parses split JSON and a split multibyte code point", async ({ page }) => {
|
|
const timeline = await setupTimeline(page)
|
|
const payload = partUpdated(textPart("prt_transport_split", "split snowman \u2603\u2603\u2603"))
|
|
const encoded = new TextEncoder().encode(`data: ${JSON.stringify(payload)}\n\n`)
|
|
const snowman = new TextEncoder().encode("\u2603")[0]!
|
|
const multibyte = encoded.indexOf(snowman)
|
|
|
|
const acknowledgement = await timeline.transport.split(payload, [9, multibyte + 1, multibyte + 2])
|
|
|
|
await timeline.waitForPart("prt_transport_split")
|
|
await expect(page.locator('[data-timeline-part-id="prt_transport_split"]')).toContainText(
|
|
"split snowman \u2603\u2603\u2603",
|
|
)
|
|
expect(acknowledgement.chunkCount).toBe(4)
|
|
})
|
|
|
|
test("delivers server heartbeat without mutating the timeline", async ({ page }) => {
|
|
const timeline = await setupTimeline(page, {
|
|
messages: [userMessage(), assistantMessage([textPart("prt_transport_steady", "steady")])],
|
|
})
|
|
const before = await page.locator("[data-timeline-row]").allTextContents()
|
|
|
|
await timeline.transport.heartbeat()
|
|
await timeline.settle()
|
|
|
|
expect(await page.locator("[data-timeline-row]").allTextContents()).toEqual(before)
|
|
expect(await timeline.transport.connections()).toHaveLength(1)
|
|
})
|
|
|
|
test("reconnects after a clean close", async ({ page }) => {
|
|
const timeline = await setupTimeline(page, { eventRetry: 10 })
|
|
const first = await timeline.transport.waitForConnection()
|
|
|
|
await timeline.transport.close()
|
|
const second = await timeline.transport.waitForConnection({ after: first.id })
|
|
await timeline.transport.send(partUpdated(textPart("prt_transport_close", "after close")))
|
|
|
|
await timeline.waitForPart("prt_transport_close")
|
|
expect(second.id).toBeGreaterThan(first.id)
|
|
expect((await timeline.transport.connections())[0]?.endedBy).toBe("close")
|
|
})
|
|
|
|
test("reconnects after a stream error", async ({ page }) => {
|
|
const timeline = await setupTimeline(page, { eventRetry: 10 })
|
|
const first = await timeline.transport.waitForConnection()
|
|
|
|
await timeline.transport.error("contract failure")
|
|
const second = await timeline.transport.waitForConnection({ after: first.id })
|
|
await timeline.transport.send(status("busy"))
|
|
|
|
await expect.poll(async () => (await timeline.transport.connections()).length).toBe(2)
|
|
expect(second.id).toBeGreaterThan(first.id)
|
|
expect((await timeline.transport.connections())[0]?.endedBy).toBe("error")
|
|
})
|
|
|
|
test("records event IDs and reconnect Last-Event-ID headers", async ({ page }) => {
|
|
const timeline = await setupTimeline(page, { eventRetry: 10 })
|
|
const first = await timeline.transport.send(partUpdated(textPart("prt_transport_id", "event with id")), {
|
|
id: "timeline-event-7",
|
|
})
|
|
await timeline.waitForPart("prt_transport_id")
|
|
|
|
await timeline.transport.error("retry with event id")
|
|
const connection = await timeline.transport.waitForConnection({ after: first.connectionID })
|
|
|
|
expect(first.eventID).toBe("timeline-event-7")
|
|
expect(connection.headers["last-event-id"]).toBe("timeline-event-7")
|
|
})
|
|
|
|
test("passes through non-event fetches", async ({ page }) => {
|
|
const timeline = await setupTimeline(page)
|
|
|
|
const health = await page.evaluate(async () => {
|
|
const response = await fetch("/global/health")
|
|
return response.json()
|
|
})
|
|
|
|
expect(health).toEqual({ healthy: true })
|
|
expect(await timeline.transport.connections()).toHaveLength(1)
|
|
})
|