Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests(ui): add tests for getViewModeChunks

Browse Source
This commit is contained in:
psychedelicious 2024-08-19 08:23:07 +10:00
parent 6c404ce5f8
commit 8ca57d03d8
1 changed files with 53 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
invokeai/frontend/web/src/features/stylePresets/util/getViewModeChunks.test.ts Normal file
View File

@ -0,0 +1,53 @@
import { getViewModeChunks } from 'features/stylePresets/util/getViewModeChunks';
import { describe, expect, it } from 'vitest';
describe('getViewModeChunks', () => {
it('should return empty strings when presetPrompt is not provided', () => {
const currentPrompt = 'current prompt';
const presetPrompt = undefined;
const result = getViewModeChunks(currentPrompt, presetPrompt);
expect(result).toEqual(['', currentPrompt, '']);
});
it('should return empty strings when presetPrompt is empty', () => {
const currentPrompt = 'current prompt';
const presetPrompt = '';
const result = getViewModeChunks(currentPrompt, presetPrompt);
expect(result).toEqual(['', currentPrompt, '']);
});
it('should append presetPrompt to currentPrompt when presetPrompt does not contain PRESET_PLACEHOLDER', () => {
const currentPrompt = 'current prompt';
const presetPrompt = 'preset prompt';
const result = getViewModeChunks(currentPrompt, presetPrompt);
expect(result).toEqual(['', `${currentPrompt} `, presetPrompt]);
});
it('should split presetPrompt into 3 parts when presetPrompt contains PRESET_PLACEHOLDER', () => {
const currentPrompt = 'current prompt';
const presetPrompt = 'before {prompt} after';
const result = getViewModeChunks(currentPrompt, presetPrompt);
expect(result).toEqual(['before ', currentPrompt, ' after']);
});
it('should split presetPrompt into 3 parts when presetPrompt contains multiple PRESET_PLACEHOLDER', () => {
const currentPrompt = 'current prompt';
const presetPrompt = 'before {prompt} middle {prompt} after';
const result = getViewModeChunks(currentPrompt, presetPrompt);
expect(result).toEqual(['before ', currentPrompt, ' middle {prompt} after']);
});
it('should handle the PRESET_PLACEHOLDER being at the start of the presetPrompt', () => {
const currentPrompt = 'current prompt';
const presetPrompt = '{prompt} after';
const result = getViewModeChunks(currentPrompt, presetPrompt);
expect(result).toEqual(['', currentPrompt, ' after']);
});
it('should handle the PRESET_PLACEHOLDER being at the end of the presetPrompt', () => {
const currentPrompt = 'current prompt';
const presetPrompt = 'before {prompt}';
const result = getViewModeChunks(currentPrompt, presetPrompt);
expect(result).toEqual(['before ', currentPrompt, '']);
});
});