53 lines
1.6 KiB
TypeScript
53 lines
1.6 KiB
TypeScript
import { describe, expect, it } from 'bun:test';
|
|
import {
|
|
exclusionsForOpenOAuthModels,
|
|
oauthExcludedRulesFromPayload,
|
|
oauthModelsFromPayload,
|
|
openOAuthModelNames,
|
|
} from '../src/services/oauthModels';
|
|
|
|
const models = [
|
|
{ id: 'gpt-5.4', displayName: 'GPT 5.4' },
|
|
{ id: 'gpt-image-1.5', displayName: 'GPT Image 1.5' },
|
|
{ id: 'gpt-image-2', displayName: 'GPT Image 2' },
|
|
];
|
|
|
|
describe('OAuth 开放模型', () => {
|
|
it('解析模型定义和提供商排除规则', () => {
|
|
expect(oauthModelsFromPayload({ models: [
|
|
{ id: 'gpt-5.4', display_name: 'GPT 5.4' },
|
|
{ id: 'GPT-5.4' },
|
|
] })).toEqual([{ id: 'gpt-5.4', displayName: 'GPT 5.4' }]);
|
|
expect(oauthExcludedRulesFromPayload({
|
|
'oauth-excluded-models': { codex: [' GPT-IMAGE-* ', 'gpt-image-*'] },
|
|
}, 'codex')).toEqual(['gpt-image-*']);
|
|
});
|
|
|
|
it('排除规则会让图像模型默认不勾选', () => {
|
|
expect(Array.from(openOAuthModelNames(models, ['gpt-image-*']))).toEqual(['gpt-5.4']);
|
|
});
|
|
|
|
it('OAuth 没有已保存排除规则时默认全选全部模型', () => {
|
|
expect(Array.from(openOAuthModelNames(models, []))).toEqual([
|
|
'gpt-5.4',
|
|
'gpt-image-1.5',
|
|
'gpt-image-2',
|
|
]);
|
|
});
|
|
|
|
it('保存时把未勾选模型转换成 OAuth 排除规则', () => {
|
|
expect(exclusionsForOpenOAuthModels([], models, ['gpt-5.4'])).toEqual([
|
|
'gpt-image-1.5',
|
|
'gpt-image-2',
|
|
]);
|
|
});
|
|
|
|
it('显式开放模型时移除与其冲突的旧通配符', () => {
|
|
expect(exclusionsForOpenOAuthModels(
|
|
['gpt-image-*', 'future-*'],
|
|
models,
|
|
['gpt-5.4', 'gpt-image-2'],
|
|
)).toEqual(['future-*', 'gpt-image-1.5']);
|
|
});
|
|
});
|