shouldWarnOnOrphanedUserRepair,

} from "./attempt.prompt-helpers.js";

import { composeSystemPromptWithHookContext } from "./attempt.thread-helpers.js";

import {

decodeHtmlEntitiesInObject,

wrapStreamFnRepairMalformedToolCallArguments,

} from "./attempt.tool-call-argument-repair.js";

import { wrapStreamFnRepairMalformedToolCallArguments } from "./attempt.tool-call-argument-repair.js";

import {

wrapStreamFnSanitizeMalformedToolCalls,

wrapStreamFnTrimToolCallNames,

});

});

describe("decodeHtmlEntitiesInObject", () => {

it("decodes HTML entities in string values", () => {

const result = decodeHtmlEntitiesInObject(

"source .env && psql "$DB" -c <query>",

);

expect(result).toBe('source .env && psql "$DB" -c <query>');

});

it("recursively decodes nested objects", () => {

const input = {

command: "cd ~/dev &amp;&amp; npm run build",

args: ["--flag=&quot;value&quot;", "&lt;input&gt;"],

nested: { deep: "a &amp; b" },

};

const result = decodeHtmlEntitiesInObject(input) as Record<string, unknown>;

expect(result.command).toBe("cd ~/dev && npm run build");

expect((result.args as string[])[0]).toBe('--flag="value"');

expect((result.args as string[])[1]).toBe("<input>");

expect((result.nested as Record<string, string>).deep).toBe("a & b");

});

it("passes through non-string primitives unchanged", () => {

expect(decodeHtmlEntitiesInObject(42)).toBe(42);

expect(decodeHtmlEntitiesInObject(null)).toBe(null);

expect(decodeHtmlEntitiesInObject(true)).toBe(true);

expect(decodeHtmlEntitiesInObject(undefined)).toBe(undefined);

});

it("returns strings without entities unchanged", () => {

const input = "plain string with no entities";

expect(decodeHtmlEntitiesInObject(input)).toBe(input);

});

it("decodes numeric character references", () => {

expect(decodeHtmlEntitiesInObject("&#39;hello&#39;")).toBe("'hello'");

expect(decodeHtmlEntitiesInObject("&#x27;world&#x27;")).toBe("'world'");

});

});

describe("prependSystemPromptAddition", () => {

it("prepends context-engine addition to the system prompt", () => {

const result = prependSystemPromptAddition({