import test from "node:test";
import assert from "node:assert/strict";
import {
  ALIYUN_QWEN_CUSTOM_MODEL_VALUE,
  isAliyunQwenPresetModel,
  resolveAliyunQwenModelSelection,
  resolveAliyunQwenModelValue,
} from "../src/lib/ai-account-models";

test("阿里百炼预设模型会被识别为预设", () => {
  assert.equal(isAliyunQwenPresetModel("qwen3.5-plus"), true);
  assert.equal(isAliyunQwenPresetModel("qwen3.5-flash"), true);
  assert.equal(isAliyunQwenPresetModel(" qwen3.5-plus "), true);
});

test("阿里百炼非预设模型会落到自定义选项", () => {
  assert.equal(isAliyunQwenPresetModel("qwen-max"), false);
  assert.equal(resolveAliyunQwenModelSelection("qwen-max"), ALIYUN_QWEN_CUSTOM_MODEL_VALUE);
});

test("阿里百炼模型值会根据当前选择输出预设或自定义值", () => {
  assert.equal(resolveAliyunQwenModelValue("qwen3.5-plus", "ignored"), "qwen3.5-plus");
  assert.equal(resolveAliyunQwenModelValue(ALIYUN_QWEN_CUSTOM_MODEL_VALUE, " qwen-max "), "qwen-max");
});