Files
aigen/tests/prompt-sequences.test.mjs
T
Towsty 1bf97568bf Add Prompt Builder page for multi-scene prompt sequences.
Name, save, and load ordered scenes on the server; assemble copyable text with a Batch-compatible double blank line between scenes.
2026-10-02 07:15:00 -05:00

100 lines
2.9 KiB
JavaScript

import test from 'node:test'
import assert from 'node:assert/strict'
import {
assembleSceneText,
assembleSequenceText,
formatCharacterLine,
emptyScene,
cloneSceneForNext
} from '../shared/prompt-sequences.mjs'
test('formatCharacterLine covers name/description cases', () => {
assert.equal(formatCharacterLine({ name: 'Ada', description: 'tall' }), 'Ada: tall')
assert.equal(formatCharacterLine({ name: 'Ada', description: '' }), 'Ada')
assert.equal(formatCharacterLine({ name: '', description: 'tall' }), 'tall')
assert.equal(formatCharacterLine({ name: '', description: '' }), '')
})
test('assembleSceneText joins non-empty boxes with a single blank line', () => {
const text = assembleSceneText({
style: 'oil paint',
settings: 'night',
characters: [
{ name: 'Ada', description: 'tall' },
{ name: '', description: '' },
{ name: 'Bo', description: 'hat' }
],
prompt: 'walks'
})
assert.equal(text, 'oil paint\n\nnight\n\nAda: tall\nBo: hat\n\nwalks')
})
test('assembleSequenceText uses double blank line between scenes and skips empties', () => {
const text = assembleSequenceText({
name: 'ignored',
scenes: [
{ style: 's1', settings: '', characters: [], prompt: 'p1' },
{ style: '', settings: '', characters: [{ name: '', description: '' }], prompt: '' },
{ style: 's2', settings: 'set2', characters: [{ name: 'C', description: 'd' }], prompt: 'p2' }
]
})
assert.equal(text, 's1\n\np1\n\n\ns2\n\nset2\n\nC: d\n\np2')
assert.ok(!text.includes('ignored'))
})
test('example shape is two jobs for desktop Batch', () => {
const text = assembleSequenceText({
scenes: [
{
style: 'style scene 1',
settings: 'settings scene 1',
characters: [
{ name: 'Character one', description: 'desc' },
{ name: 'Character two', description: 'desc' }
],
prompt: 'prompt scene 1'
},
{
style: 'style scene 2',
settings: 'settings scene 2',
characters: [{ name: 'Character one', description: 'desc' }],
prompt: 'prompt scene 2'
}
]
})
assert.equal(
text,
[
'style scene 1',
'',
'settings scene 1',
'',
'Character one: desc',
'Character two: desc',
'',
'prompt scene 1',
'',
'',
'style scene 2',
'',
'settings scene 2',
'',
'Character one: desc',
'',
'prompt scene 2'
].join('\n')
)
})
test('cloneSceneForNext copies fields with new ids', () => {
const first = emptyScene(1)
first.style = 'a'
first.characters[0].name = 'Ada'
const next = cloneSceneForNext(first, 2)
assert.equal(next.title, 'Scene 2')
assert.equal(next.style, 'a')
assert.equal(next.characters[0].name, 'Ada')
assert.notEqual(next.id, first.id)
assert.notEqual(next.characters[0].id, first.characters[0].id)
})