Jason HokuandClaude Fable 5.1
3eae054c11
Load Session restores Workflow Sweep arrays from meta.configs_json (workflowConfigToArray) and tells the user when an older sweep session has no saved configs. Builder Start shows 'attached'. Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
56 lines
2.7 KiB
JavaScript
56 lines
2.7 KiB
JavaScript
// workflowConfigToArray lives in an ES module that imports ComfyUI's frontend, so this test slices the
|
|
// pure helper functions out of the source and evaluates them (same trick as other tests that avoid the DOM).
|
|
const test = require('node:test');
|
|
const assert = require('node:assert');
|
|
const fs = require('fs');
|
|
const path = require('path');
|
|
|
|
const src = fs.readFileSync(path.join(__dirname, '../../web/conf_builder/conf-builder-workflow-sweep.js'), 'utf8');
|
|
|
|
function grab(name) {
|
|
const re = new RegExp('(?:export )?function ' + name + '\\(');
|
|
const m = re.exec(src);
|
|
assert.ok(m, 'function ' + name + ' not found');
|
|
let i = src.indexOf('{', m.index), depth = 0, j = i;
|
|
for (; j < src.length; j++) {
|
|
if (src[j] === '{') depth++;
|
|
else if (src[j] === '}') { depth--; if (depth === 0) break; }
|
|
}
|
|
return src.slice(m.index, j + 1).replace(/^export /, '');
|
|
}
|
|
|
|
const code = ['defaultWorkflowState', 'ensureWorkflowState', 'newWorkflowArray', 'workflowConfigToArray'].map(grab).join('\n');
|
|
const { workflowConfigToArray } = new Function(code + '\nreturn { workflowConfigToArray };')();
|
|
|
|
const cfg = {
|
|
model_type: 'workflow', name: 'My sweep', workflow: { '3': { class_type: 'KSampler', inputs: { seed: 1 } } },
|
|
workflow_source: 'wf.json', workflow_output: { node_id: '9', output_index: 0 }, workflow_run_other_outputs: true,
|
|
workflow_bindings: [{ id: 'b1', kind: 'literal', targets: [{ node_id: '3', input: 'seed' }], values: [1, 2] }],
|
|
positive: [['a cat']], negative: 'bad', _prompt_source: 'custom',
|
|
};
|
|
|
|
test('restores workflow, bindings, output and custom prompts', () => {
|
|
const a = workflowConfigToArray(cfg, 0);
|
|
assert.strictEqual(a.kind, 'workflow');
|
|
assert.strictEqual(a.name, 'My sweep');
|
|
assert.deepStrictEqual(a.workflow.api_prompt, cfg.workflow);
|
|
assert.strictEqual(a.workflow.source_name, 'wf.json');
|
|
assert.deepStrictEqual(a.workflow.output, { node_id: '9', output_index: 0 });
|
|
assert.strictEqual(a.workflow.run_other_outputs, true);
|
|
assert.deepStrictEqual(a.workflow.bindings, cfg.workflow_bindings);
|
|
assert.strictEqual(a.use_custom_prompts, true);
|
|
assert.deepStrictEqual(a.positive_prompt_groups, [['a cat']]);
|
|
assert.strictEqual(a.negative_prompt, 'bad');
|
|
});
|
|
|
|
test('global prompts are not copied into the array; odd input never throws', () => {
|
|
const a = workflowConfigToArray({ ...cfg, _prompt_source: 'global' }, 2);
|
|
assert.strictEqual(a.use_custom_prompts, false);
|
|
assert.deepStrictEqual(a.positive_prompt_groups, []);
|
|
const b = workflowConfigToArray(null, 1);
|
|
assert.strictEqual(b.kind, 'workflow');
|
|
assert.strictEqual(b.name, 'Workflow 2');
|
|
assert.strictEqual(b.workflow.api_prompt, null);
|
|
assert.deepStrictEqual(b.workflow.bindings, []);
|
|
});
|