OmniRoute/tests/unit/cli-completion-dynamic.test.ts

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

test("completion.mjs pode ser importado sem erro", async () => {
  const mod = await import("../../bin/cli/commands/completion.mjs");
  assert.equal(typeof mod.registerCompletion, "function");
  assert.equal(typeof mod.runCompletionCommand, "function");
});

test("runCompletionCommand bash retorna 0 e string não-vazia", async () => {
  const chunks: string[] = [];
  const orig = process.stdout.write.bind(process.stdout);
  (process.stdout as any).write = (c: string | Uint8Array) => {
    if (typeof c === "string") chunks.push(c);
    return true;
  };
  try {
    const { runCompletionCommand } = await import("../../bin/cli/commands/completion.mjs");
    const code = await runCompletionCommand("bash");
    assert.equal(code, 0);
  } finally {
    (process.stdout as any).write = orig;
  }
  const out = chunks.join("");
  assert.ok(out.includes("omniroute"), "bash script should mention omniroute");
  assert.ok(out.includes("_omniroute"), "bash script should define _omniroute function");
});

test("runCompletionCommand zsh contém compdef", async () => {
  const chunks: string[] = [];
  const orig = process.stdout.write.bind(process.stdout);
  (process.stdout as any).write = (c: string | Uint8Array) => {
    if (typeof c === "string") chunks.push(c);
    return true;
  };
  try {
    const { runCompletionCommand } = await import("../../bin/cli/commands/completion.mjs");
    const code = await runCompletionCommand("zsh");
    assert.equal(code, 0);
  } finally {
    (process.stdout as any).write = orig;
  }
  const out = chunks.join("");
  assert.ok(out.includes("compdef"), "zsh script should contain compdef");
});

test("runCompletionCommand fish retorna 0", async () => {
  const chunks: string[] = [];
  const orig = process.stdout.write.bind(process.stdout);
  (process.stdout as any).write = (c: string | Uint8Array) => {
    if (typeof c === "string") chunks.push(c);
    return true;
  };
  try {
    const { runCompletionCommand } = await import("../../bin/cli/commands/completion.mjs");
    const code = await runCompletionCommand("fish");
    assert.equal(code, 0);
  } finally {
    (process.stdout as any).write = orig;
  }
  const out = chunks.join("");
  assert.ok(out.includes("omniroute"), "fish script should mention omniroute");
});

test("runCompletionCommand shell inválido retorna 1", async () => {
  const orig = process.stderr.write.bind(process.stderr);
  (process.stderr as any).write = () => true;
  try {
    const { runCompletionCommand } = await import("../../bin/cli/commands/completion.mjs");
    const code = await runCompletionCommand("powershell" as any);
    assert.equal(code, 1);
  } finally {
    (process.stderr as any).write = orig;
  }
});

test("completion scripts incluem combos/providers/models no cache dinamicamente", async () => {
  const { runCompletionCommand } = await import("../../bin/cli/commands/completion.mjs");
  const chunks: string[] = [];
  const orig = process.stdout.write.bind(process.stdout);
  (process.stdout as any).write = (c: string | Uint8Array) => {
    if (typeof c === "string") chunks.push(c);
    return true;
  };
  try {
    await runCompletionCommand("zsh");
  } finally {
    (process.stdout as any).write = orig;
  }
  const out = chunks.join("");
  assert.ok(
    out.includes("completion-cache.json") || out.includes("omniroute_get_cache"),
    "should reference cache"
  );
});

test("completion scripts expõem os alvos de execução e configuração", async () => {
  const { runCompletionCommand } = await import("../../bin/cli/commands/completion.mjs");
  const expected = ["connect", "contexts", "configure", "launch", "launch-codex", "run", "repair"];

  for (const shell of ["bash", "zsh", "fish"] as const) {
    const chunks: string[] = [];
    const originalWrite = process.stdout.write.bind(process.stdout);
    process.stdout.write = ((chunk: unknown) => {
      if (typeof chunk === "string") chunks.push(chunk);
      return true;
    }) as typeof process.stdout.write;
    try {
      assert.equal(await runCompletionCommand(shell), 0);
    } finally {
      process.stdout.write = originalWrite;
    }
    const output = chunks.join("");
    for (const command of expected) {
      assert.ok(output.includes(command), `${shell} completion should include ${command}`);
    }
  }
});