diegosouzapw--omniroute/tests/unit/translator-openai-to-cursor.test.ts

import test from "node:test";
import assert from "node:assert/strict";

const { buildCursorRequest } =
  await import("../../open-sse/translator/request/openai-to-cursor.ts");

test("OpenAI -> Cursor rewrites system prompts and preserves assistant tool calls", () => {
  const result = buildCursorRequest(
    "gpt-4o",
    {
      messages: [
        { role: "system", content: "Rules" },
        { role: "user", content: "Hello" },
        {
          role: "assistant",
          content: "Working",
          tool_calls: [
            {
              id: "call_1",
              type: "function",
              function: { name: "read_file", arguments: '{"path":"/tmp/a"}' },
              index: 0,
            },
          ],
        },
      ],
    },
    false,
    null
  );

  assert.deepEqual(result.messages[0], {
    role: "user",
    content: "[System Instructions]\nRules",
  });
  assert.deepEqual(result.messages[1], {
    role: "user",
    content: "Hello",
  });
  assert.deepEqual(result.messages[2], {
    role: "assistant",
    content: "Working",
    tool_calls: [
      {
        id: "call_1",
        type: "function",
        function: { name: "read_file", arguments: '{"path":"/tmp/a"}' },
      },
    ],
  });
});

test("OpenAI -> Cursor converts tool_result blocks into sanitized XML user content", () => {
  const result = buildCursorRequest(
    "gpt-4o",
    {
      messages: [
        {
          role: "assistant",
          content: [{ type: "tool_use", id: "call_1\nnoise", name: "read_file", input: {} }],
        },
        {
          role: "user",
          content: [
            { type: "text", text: "Hello" },
            {
              type: "tool_result",
              tool_use_id: "call_1",
              content: [{ type: "text", text: "done\u0000" }],
            },
          ],
        },
      ],
    },
    false,
    null
  );

  assert.equal(result.messages.length, 2);
  assert.equal(result.messages[1].role, "user");
  assert.match(result.messages[1].content, /Hello/);
  assert.match(result.messages[1].content, /<tool_name>read_file<\/tool_name>/);
  assert.match(result.messages[1].content, /<tool_call_id>call_1<\/tool_call_id>/);
  assert.match(result.messages[1].content, /<result>done<\/result>/);
  assert.equal(result.messages[1].content.includes("\u0000"), false);
});

test("OpenAI -> Cursor converts assistant tool_use blocks into assistant tool_calls", () => {
  const result = buildCursorRequest(
    "gpt-4o",
    {
      messages: [
        {
          role: "assistant",
          content: [
            { type: "text", text: "Calling a tool" },
            { type: "tool_use", id: "call_2", name: "weather", input: { city: "Tokyo" } },
          ],
        },
      ],
    },
    false,
    null
  );

  assert.deepEqual(result.messages, [
    {
      role: "assistant",
      content: "Calling a tool",
      tool_calls: [
        {
          id: "call_2",
          type: "function",
          function: { name: "weather", arguments: '{"city":"Tokyo"}' },
        },
      ],
    },
  ]);
});

test("OpenAI -> Cursor converts tool role messages using remembered tool metadata", () => {
  const result = buildCursorRequest(
    "gpt-4o",
    {
      messages: [
        {
          role: "assistant",
          tool_calls: [
            {
              id: "call_3",
              type: "function",
              function: { name: "search_docs", arguments: "{}" },
            },
          ],
        },
        { role: "tool", tool_call_id: "call_3", content: "found it" },
      ],
    },
    false,
    null
  );

  assert.equal(result.messages[1].role, "user");
  assert.match(result.messages[1].content, /<tool_name>search_docs<\/tool_name>/);
  assert.match(result.messages[1].content, /<result>found it<\/result>/);
});

test("OpenAI -> Cursor preserves image_url parts so vision input survives", () => {
  const dataUri = "data:image/png;base64,iVBORw0KGgo=";
  const result = buildCursorRequest(
    "gpt-5.2",
    {
      messages: [
        {
          role: "user",
          content: [
            { type: "text", text: "what color?" },
            { type: "image_url", image_url: { url: dataUri } },
            { type: "image_url", image_url: { url: "https://example.com/a.png" } },
          ],
        },
      ],
    },
    false,
    null
  );

  // Content is kept as an OpenAI array: a leading text part + the image parts.
  assert.equal(result.messages.length, 1);
  assert.equal(result.messages[0].role, "user");
  const content = result.messages[0].content;
  assert.ok(Array.isArray(content), "content preserved as array");
  assert.deepEqual(content[0], { type: "text", text: "what color?" });
  assert.deepEqual(content[1], { type: "image_url", image_url: { url: dataUri } });
  assert.deepEqual(content[2], {
    type: "image_url",
    image_url: { url: "https://example.com/a.png" },
  });
});

test("OpenAI -> Cursor accepts shorthand image_url string form", () => {
  const result = buildCursorRequest(
    "gpt-5.2",
    {
      messages: [
        {
          role: "user",
          content: [{ type: "image_url", image_url: "data:image/png;base64,AAAA" }],
        },
      ],
    },
    false,
    null
  );
  const content = result.messages[0].content;
  assert.ok(Array.isArray(content));
  // No text part (none supplied) — just the normalized image part.
  assert.deepEqual(content, [
    { type: "image_url", image_url: { url: "data:image/png;base64,AAAA" } },
  ]);
});