mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-11 12:09:53 +08:00
46 lines
1.6 KiB
TypeScript
46 lines
1.6 KiB
TypeScript
|
import { cleanup, render } from "@testing-library/react"
|
||
|
|
import { afterEach, describe, expect, it } from "vitest"
|
||
|
|
import { ToolCallCard } from "@/components/chat/ToolCallCard"
|
||
|
|
|
||
|
|
afterEach(cleanup)
|
||
|
|
|
||
|
|
const dict = {
|
||
|
|
tools: { complete: "Complete" },
|
||
|
|
chat: { copied: "Copied", failedToCopy: "Failed", copyResponse: "Copy" },
|
||
|
|
}
|
||
|
|
|
||
|
|
describe("ToolCallCard", () => {
|
||
|
|
it("shows streamed operations without crashing on broken entries", () => {
|
||
|
|
// A partly streamed or invalid edit_diagram input
|
||
|
|
const operations = [
|
||
|
|
null,
|
||
|
|
{ operation: {} },
|
||
|
|
{ operation: "add", cell_id: {} },
|
||
|
|
// JSON can hold an object that does not turn into text
|
||
|
|
JSON.parse('{"operation":"add","cell_id":{"toString":null}}'),
|
||
|
|
{ operation: "add", cell_id: "2", new_xml: {} },
|
||
|
|
{ operation: "update", cell_id: "3", new_xml: '<mxCell id="3"/>' },
|
||
|
|
]
|
||
|
|
const { container } = render(
|
||
|
|
<ToolCallCard
|
||
|
|
part={
|
||
|
|
{
|
||
|
|
type: "tool-edit_diagram",
|
||
|
|
toolCallId: "t1",
|
||
|
|
state: "input-streaming",
|
||
|
|
input: { operations },
|
||
|
|
} as any
|
||
|
|
}
|
||
|
|
expandedTools={{ t1: true }}
|
||
|
|
setExpandedTools={() => {}}
|
||
|
|
onCopy={() => {}}
|
||
|
|
copiedToolCallId={null}
|
||
|
|
copyFailedToolCallId={null}
|
||
|
|
dict={dict}
|
||
|
|
/>,
|
||
|
|
)
|
||
|
|
expect(container.textContent).toContain("cell_id: 3")
|
||
|
|
expect(container.textContent).toContain('<mxCell id="3"/>')
|
||
|
|
})
|
||
|
|
})
|