wehub-resource-sync
68 lines
2.5 KiB
TypeScript
68 lines
2.5 KiB
TypeScript
import { describe, expect, it } from 'vitest';
|
|
|
|
import { IMAGE_GEN_ENV_KEYS, resolveImageGenerationMcpEnv } from '@/common/config/imageGenerationMcpEnv';
|
|
import type { IProvider } from '@/common/config/storage';
|
|
|
|
const geminiProvider: IProvider = {
|
|
id: '03c8482c',
|
|
platform: 'gemini',
|
|
name: 'Gemini',
|
|
base_url: 'https://generativelanguage.googleapis.com',
|
|
api_key: 'provider-key',
|
|
models: ['gemini-2.5-pro', 'gemini-3-pro-image-preview'],
|
|
enabled: true,
|
|
};
|
|
|
|
describe('resolveImageGenerationMcpEnv', () => {
|
|
it('resolves image generation env from provider id and selected model', () => {
|
|
const result = resolveImageGenerationMcpEnv({ id: '03c8482c', use_model: 'gemini-3-pro-image-preview' }, [
|
|
geminiProvider,
|
|
]);
|
|
|
|
expect(result.ok).toBe(true);
|
|
if (!result.ok) return;
|
|
expect(result.source).toBe('provider-id');
|
|
expect(result.env).toEqual({
|
|
[IMAGE_GEN_ENV_KEYS.providerId]: '03c8482c',
|
|
[IMAGE_GEN_ENV_KEYS.platform]: 'gemini',
|
|
[IMAGE_GEN_ENV_KEYS.baseUrl]: 'https://generativelanguage.googleapis.com',
|
|
[IMAGE_GEN_ENV_KEYS.apiKey]: 'provider-key',
|
|
[IMAGE_GEN_ENV_KEYS.model]: 'gemini-3-pro-image-preview',
|
|
});
|
|
});
|
|
|
|
it('matches legacy env by platform, base URL, and model when provider id is absent', () => {
|
|
const result = resolveImageGenerationMcpEnv(undefined, [geminiProvider], {
|
|
AIONUI_IMG_PLATFORM: 'gemini',
|
|
AIONUI_IMG_BASE_URL: 'https://generativelanguage.googleapis.com/',
|
|
AIONUI_IMG_MODEL: 'gemini-3-pro-image-preview',
|
|
AIONUI_IMG_API_KEY: 'stale-key',
|
|
});
|
|
|
|
expect(result.ok).toBe(true);
|
|
if (!result.ok) return;
|
|
expect(result.source).toBe('field-match');
|
|
expect(result.env.AIONUI_IMG_PROVIDER_ID).toBe('03c8482c');
|
|
expect(result.env.AIONUI_IMG_API_KEY).toBe('provider-key');
|
|
});
|
|
|
|
it('fails loudly when neither provider id nor legacy fields match a provider', () => {
|
|
const result = resolveImageGenerationMcpEnv(
|
|
{ platform: 'gemini', base_url: 'https://unknown.example', use_model: 'gemini-3-pro-image-preview' },
|
|
[geminiProvider]
|
|
);
|
|
|
|
expect(result.ok).toBe(false);
|
|
if (result.ok) return;
|
|
expect(result.reason).toBe('no-provider-match');
|
|
});
|
|
|
|
it('fails when the selected model is not present on the matched provider', () => {
|
|
const result = resolveImageGenerationMcpEnv({ id: '03c8482c', use_model: 'missing-image-model' }, [geminiProvider]);
|
|
|
|
expect(result.ok).toBe(false);
|
|
if (result.ok) return;
|
|
expect(result.reason).toBe('model-not-found');
|
|
});
|
|
});
|