mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-08-28 08:21:27 -03:00
Will Miao
17 lines
854 B
JavaScript
17 lines
854 B
JavaScript
import { describe, expect, it } from 'vitest';
|
|
|
|
import { ModelContextMenuMixin } from '../../../static/js/components/ContextMenu/ModelContextMenuMixin.js';
|
|
|
|
describe('ModelContextMenuMixin.getModelTypePrefix', () => {
|
|
it('maps every known model type to its API route prefix', () => {
|
|
expect(ModelContextMenuMixin.getModelTypePrefix.call({ modelType: 'lora' })).toBe('loras');
|
|
expect(ModelContextMenuMixin.getModelTypePrefix.call({ modelType: 'checkpoint' })).toBe('checkpoints');
|
|
expect(ModelContextMenuMixin.getModelTypePrefix.call({ modelType: 'embedding' })).toBe('embeddings');
|
|
});
|
|
|
|
it('falls back to the loras prefix for unknown types', () => {
|
|
expect(ModelContextMenuMixin.getModelTypePrefix.call({ modelType: 'unknown' })).toBe('loras');
|
|
expect(ModelContextMenuMixin.getModelTypePrefix.call({})).toBe('loras');
|
|
});
|
|
});
|