fix: Modify test mock data

This commit is contained in:
Vincent 2025-11-08 09:57:48 -08:00
commit 12332932b0
2 changed files with 15 additions and 12 deletions

View file

@ -510,12 +510,13 @@ describe("useConnection", () => {
requestedSchema: {
type: "object",
properties: {
name: {
$ref: "#/properties/nameDef",
},
nameDef: {
source: {
type: "string",
title: "Name",
minLength: 1,
title: "A Connectable Node",
},
target: {
$ref: "#/properties/source",
},
},
},
@ -529,9 +530,10 @@ describe("useConnection", () => {
expect(mockProtocolOnMessage).toHaveBeenCalledTimes(1);
const message = mockProtocolOnMessage.mock.calls[0][0];
expect(message.params.requestedSchema.properties.name).toEqual({
expect(message.params.requestedSchema.properties.target).toEqual({
type: "string",
title: "Name",
minLength: 1,
title: "A Connectable Node",
});
});
});

View file

@ -302,7 +302,7 @@ export function formatFieldLabel(key: string): string {
}
/**
* Resolves $ref references in a JSON-RPC message's requestedSchema
* Resolves `$ref` references in a JSON-RPC "elicitation/create" message's `requestedSchema` field
* @param message The JSON-RPC message that may contain $ref references
* @returns A new message with resolved $ref references, or the original message if no resolution is needed
*/
@ -325,10 +325,11 @@ export function resolveRefsInMessage(message: JSONRPCMessage): JSONRPCMessage {
...requestedSchema,
properties: Object.fromEntries(
Object.entries(requestedSchema.properties).map(
([key, propSchema]) => [
key,
resolveRef(propSchema, requestedSchema),
],
([key, propSchema]) => {
const resolved = resolveRef(propSchema, requestedSchema);
const normalized = normalizeUnionType(resolved);
return [key, normalized];
},
),
),
},