update unit test

This commit is contained in:
code-october 2024-11-30 07:28:38 +00:00
parent e1ac0538b8
commit 54f6feb2d7

View File

@ -9,13 +9,23 @@ describe("isModelNotavailableInServer", () => {
expect(result).toBe(false); expect(result).toBe(false);
}); });
test("test model will return false, which means the model is not available", () => { test("test model will return true when model is not available in custom models", () => {
const customModels = "-all,gpt-4o-mini"; const customModels = "-all,gpt-4o-mini";
const modelName = "gpt-4"; const modelName = "gpt-4";
const providerNames = "OpenAI"; const providerNames = "OpenAI";
const result = isModelNotavailableInServer(customModels, modelName, providerNames); const result = isModelNotavailableInServer(customModels, modelName, providerNames);
expect(result).toBe(true); expect(result).toBe(true);
}); });
test("should respect DISABLE_GPT4 setting", () => {
process.env.DISABLE_GPT4 = "1";
const result = isModelNotavailableInServer("", "gpt-4", "OpenAI");
expect(result).toBe(true);
});
test("should handle empty provider names", () => {
const result = isModelNotavailableInServer("-all,gpt-4", "gpt-4", "");
expect(result).toBe(true);
});
test("support passing multiple providers, model unavailable on one of the providers will return true", () => { test("support passing multiple providers, model unavailable on one of the providers will return true", () => {
const customModels = "-all,gpt-4@Google"; const customModels = "-all,gpt-4@Google";