diegosouzapw--omniroute/tests/unit/settings-schema-routing-strategies.test.ts

import test from "node:test";
import assert from "node:assert/strict";
import { SETTINGS_FALLBACK_STRATEGY_VALUES } from "@/shared/constants/routingStrategies";
import { updateSettingsSchema as settingsRouteSchema } from "@/shared/validation/settingsSchemas";
import * as sharedSchemaModule from "@/shared/validation/schemas";
import { updateSettingsSchema as sharedSettingsSchema } from "@/shared/validation/schemas";

for (const strategy of SETTINGS_FALLBACK_STRATEGY_VALUES) {
  test(`settings route schema accepts fallbackStrategy=${strategy}`, () => {
    const parsed = settingsRouteSchema.parse({ fallbackStrategy: strategy });
    assert.equal(parsed.fallbackStrategy, strategy);
  });

  test(`shared settings schema accepts fallbackStrategy=${strategy}`, () => {
    const parsed = sharedSettingsSchema.parse({ fallbackStrategy: strategy });
    assert.equal(parsed.fallbackStrategy, strategy);
  });
}

test("settings schemas reject combo-only strategies as account fallback strategies", () => {
  for (const strategy of ["auto", "lkgp", "context-optimized"]) {
    assert.equal(settingsRouteSchema.safeParse({ fallbackStrategy: strategy }).success, false);
    assert.equal(sharedSettingsSchema.safeParse({ fallbackStrategy: strategy }).success, false);
  }
});

test("shared settings schema module omits the unused fallback strategy sub-schema export", () => {
  assert.equal("settingsFallbackStrategySchema" in sharedSchemaModule, false);
});

test("settings schemas accept cooldown-aware retry knobs", () => {
  const payload = {
    requestRetry: 3,
    maxRetryIntervalSec: 30,
  };

  const routeParsed = settingsRouteSchema.parse(payload);
  const sharedParsed = sharedSettingsSchema.parse(payload);

  assert.equal(routeParsed.requestRetry, 3);
  assert.equal(routeParsed.maxRetryIntervalSec, 30);
  assert.equal(sharedParsed.requestRetry, 3);
  assert.equal(sharedParsed.maxRetryIntervalSec, 30);
});

test("settings schemas accept request body limit", () => {
  const routeParsed = settingsRouteSchema.parse({ maxBodySizeMb: 100 });
  const sharedParsed = sharedSettingsSchema.parse({ maxBodySizeMb: 100 });

  assert.equal(routeParsed.maxBodySizeMb, 100);
  assert.equal(sharedParsed.maxBodySizeMb, 100);
  assert.equal(settingsRouteSchema.safeParse({ maxBodySizeMb: 0 }).success, false);
  assert.equal(settingsRouteSchema.safeParse({ maxBodySizeMb: 501 }).success, false);
  assert.equal(sharedSettingsSchema.safeParse({ maxBodySizeMb: 0 }).success, false);
  assert.equal(sharedSettingsSchema.safeParse({ maxBodySizeMb: 501 }).success, false);
});

test("settings schemas accept wsAuth toggle", () => {
  const routeParsed = settingsRouteSchema.parse({ wsAuth: true });
  const sharedParsed = sharedSettingsSchema.parse({ wsAuth: false });

  assert.equal(routeParsed.wsAuth, true);
  assert.equal(sharedParsed.wsAuth, false);
});

test("settings schemas accept Claude Code unprefixed model routing toggle", () => {
  const routeParsed = settingsRouteSchema.parse({
    preferClaudeCodeForUnprefixedClaudeModels: true,
  });
  const sharedParsed = sharedSettingsSchema.parse({
    preferClaudeCodeForUnprefixedClaudeModels: false,
  });

  assert.equal(routeParsed.preferClaudeCodeForUnprefixedClaudeModels, true);
  assert.equal(sharedParsed.preferClaudeCodeForUnprefixedClaudeModels, false);
});

test("settings schemas accept combo configuration modes", () => {
  const routeParsed = settingsRouteSchema.parse({ comboConfigMode: "expert" });
  const sharedParsed = sharedSettingsSchema.parse({ comboConfigMode: "guided" });

  assert.equal(routeParsed.comboConfigMode, "expert");
  assert.equal(sharedParsed.comboConfigMode, "guided");
  assert.equal(settingsRouteSchema.safeParse({ comboConfigMode: "compact" }).success, false);
  assert.equal(sharedSettingsSchema.safeParse({ comboConfigMode: "compact" }).success, false);
});

test("settings schemas accept global Codex fast tier setting", () => {
  const payload = { codexServiceTier: { enabled: true } };
  const extendedPayload = {
    codexServiceTier: {
      enabled: true,
      tier: "flex",
      supportedModels: ["gpt-5.5", "gpt-5.4"],
    },
  };
  const routeParsed = settingsRouteSchema.parse(payload);
  const sharedParsed = sharedSettingsSchema.parse(payload);
  const extendedRouteParsed = settingsRouteSchema.parse(extendedPayload);
  const extendedSharedParsed = sharedSettingsSchema.parse(extendedPayload);

  assert.deepEqual(routeParsed.codexServiceTier, { enabled: true });
  assert.deepEqual(sharedParsed.codexServiceTier, { enabled: true });
  assert.deepEqual(extendedRouteParsed.codexServiceTier, extendedPayload.codexServiceTier);
  assert.deepEqual(extendedSharedParsed.codexServiceTier, extendedPayload.codexServiceTier);
  assert.equal(
    settingsRouteSchema.safeParse({ codexServiceTier: { enabled: "yes" } }).success,
    false
  );
  assert.equal(
    sharedSettingsSchema.safeParse({ codexServiceTier: { enabled: "yes" } }).success,
    false
  );
  assert.equal(
    settingsRouteSchema.safeParse({ codexServiceTier: { enabled: true, tier: "turbo" } }).success,
    false
  );
  assert.equal(
    sharedSettingsSchema.safeParse({ codexServiceTier: { enabled: true, tier: "turbo" } }).success,
    false
  );
});

test("settings schemas accept endpoint tunnel visibility toggles", () => {
  const payload = {
    hideEndpointCloudflaredTunnel: true,
    hideEndpointTailscaleFunnel: true,
    hideEndpointNgrokTunnel: true,
  };

  const routeParsed = settingsRouteSchema.parse(payload);
  const sharedParsed = sharedSettingsSchema.parse(payload);

  assert.equal(routeParsed.hideEndpointCloudflaredTunnel, true);
  assert.equal(routeParsed.hideEndpointTailscaleFunnel, true);
  assert.equal(routeParsed.hideEndpointNgrokTunnel, true);
  assert.equal(sharedParsed.hideEndpointCloudflaredTunnel, true);
  assert.equal(sharedParsed.hideEndpointTailscaleFunnel, true);
  assert.equal(sharedParsed.hideEndpointNgrokTunnel, true);
});