Files
ComfyAssets-ComfyUI_PromptM…/tests/js/workspace-system-integrations.test.js
T
Vito Sansevero 1172366f83 feat(system): configure ComfyUI and LoRA integrations natively
System › Integrations replaces its hand-off with the concept's two
surfaces. ComfyUI: only the send destination and renderer the bridge
supports can be chosen; the other options and Return to canvas are
disabled with the reason, and Host compatibility shows the renderer the
open ComfyUI windows report (each bridge destination now carries
renderer: classic or nodes2) (D106). LoRA & resources: detection, the
enabled setting and the loaded trigger words read as separate facts,
the read-only LoraManager path row follows the toggle (D106), and the
CivitAI key is never filled in: typing drafts a new key, blank keeps the
saved one and Clear saved key drafts its removal (D114). They save
through the one Save row (/lora/enable with the settings revision).

LoRA reference data → Review import shows the server's read-only review
(replacement, what saved pairs block, downloads, backup box) and Start
runs the LoRA reference import Activity job (D107), with its own
Activity descriptor and Review again into Integrations when the files or
pairs overtook it.
2026-10-25 07:16:35 -07:00

155 lines
11 KiB
JavaScript
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
// System › Integrations (#249 System S10 Task 27): the host values and saves behind the LoRA rows (D114), the D106
// renderer and detection lines, and the D107 import review built from the server's plan.
const test = require('node:test');
const assert = require('node:assert/strict');
const fs = require('node:fs');
const path = require('node:path');
const integrations = import('../../web/js/workspace/system-integrations.mjs');
const kinds = import('../../web/js/workspace/activity-kinds.mjs');
const CONCEPT = fs.readFileSync(path.join(__dirname, '../../docs/concepts/workspace/app.js'), 'utf8');
const SOURCE = fs.readFileSync(path.join(__dirname, '../../web/js/workspace/system-integrations.mjs'), 'utf8');
test('the section keeps the concept rows, labels and options in order', () => {
const start = CONCEPT.indexOf('case "integrations":');
const body = CONCEPT.slice(start, CONCEPT.indexOf('break;', start));
const rows = [...body.matchAll(/setting\("([^"]+)", "([^"]+)", "([^"]+)"/g)].map(([, id, label]) => [id, label]);
assert.deepEqual(rows.map(([id]) => id), ['send-destination', 'renderer', 'return-canvas', 'lora-enable', 'lora-triggers',
'civitai-key']);
for (const [id, label] of rows) {
assert.ok(SOURCE.includes(`id: '${id}', label: '${label}'`), `${id} keeps the concept label "${label}"`);
}
for (const option of ['Ask each time', 'Selected Prompt Manager node', 'Create a Prompt Manager (+/-) node',
'Follow host renderer', 'Classic', 'Nodes 2.0']) assert.ok(body.includes(option) && SOURCE.includes(`'${option}'`), option);
for (const surface of ['ComfyUI', 'LoRA & resources']) assert.ok(SOURCE.includes(`'h2', '${surface}'`), surface);
assert.ok(body.includes('"LoRA reference data", "Review and import resource metadata."'));
// Prototype copy is never shipped (Global constraints): the sample detection line and the mock warning.
assert.ok(!SOURCE.includes('Sample detection'));
assert.ok(!SOURCE.includes('Do not enter a real key'));
assert.ok(!SOURCE.includes('Simulate import'));
});
test('LoRA status becomes host values: drafted flags and read-only facts kept apart', async () => {
const { loraValues } = await integrations;
assert.deepEqual(loraValues({ enabled: true, trigger_words_enabled: false, path: 'custom_nodes/ComfyUI-Lora-Manager',
has_civitai_api_key: true, detected: true, detected_path: 'custom_nodes/ComfyUI-Lora-Manager', trigger_cache_loaded: true }), {
lora_enabled: true, lora_trigger_words_enabled: false, clear_civitai_api_key: false,
lora_path: 'custom_nodes/ComfyUI-Lora-Manager', lora_detected: true, lora_detected_path: 'custom_nodes/ComfyUI-Lora-Manager',
lora_has_key: true, lora_cache_loaded: true,
});
const empty = loraValues({});
assert.equal(empty.lora_enabled, false);
assert.equal(empty.lora_path, '');
assert.equal(empty.lora_has_key, false);
// The key itself is never part of what the page holds from the server.
assert.ok(!Object.keys(loraValues({ civitai_api_key: 'secret' })).includes('civitai_api_key'));
});
test('host changes split between /lora/enable and /settings', async () => {
const { splitHostChanges } = await integrations;
assert.deepEqual(splitHostChanges({ lora_enabled: true, result_timeout: 5 }),
{ lora: { lora_enabled: true }, settings: { result_timeout: 5 } });
assert.deepEqual(splitHostChanges({ result_timeout: 5 }), { lora: null, settings: { result_timeout: 5 } });
assert.deepEqual(splitHostChanges({ civitai_api_key: 'k', clear_civitai_api_key: false }),
{ lora: { civitai_api_key: 'k', clear_civitai_api_key: false }, settings: null });
assert.deepEqual(splitHostChanges(undefined), { lora: null, settings: null });
});
test('key never prefilled; blank keeps; Clear saved key clears; saved through Save preferences (D114)', async () => {
const { loraPayload, keyText, KEY_COPY } = await integrations;
const saved = { lora_enabled: true, lora_trigger_words_enabled: false, lora_path: 'custom_nodes/LM', lora_has_key: true };
// The whole LoRA state goes with every save; the saved path is kept; no key unless one was typed.
assert.deepEqual(loraPayload({ lora_trigger_words_enabled: true }, saved),
{ enabled: true, trigger_words_enabled: true, path: 'custom_nodes/LM' });
assert.deepEqual(loraPayload({ civitai_api_key: ' ' }, saved), { enabled: true, trigger_words_enabled: false, path: 'custom_nodes/LM' });
assert.deepEqual(loraPayload({ civitai_api_key: ' new-key ' }, saved),
{ enabled: true, trigger_words_enabled: false, path: 'custom_nodes/LM', civitai_api_key: 'new-key' });
assert.deepEqual(loraPayload({ clear_civitai_api_key: true, civitai_api_key: 'ignored' }, saved),
{ enabled: true, trigger_words_enabled: false, path: 'custom_nodes/LM', clear_civitai_api_key: true });
assert.deepEqual(loraPayload({ lora_enabled: true }, {}), { enabled: true, trigger_words_enabled: false, path: '' });
assert.equal(keyText({ lora_has_key: true }, {}), KEY_COPY.saved);
assert.equal(keyText({ lora_has_key: false }, {}), KEY_COPY.none);
assert.equal(keyText({ lora_has_key: true }, { clear_civitai_api_key: true }), KEY_COPY.clearing);
assert.equal(keyText({ lora_has_key: false }, { civitai_api_key: 'k' }), KEY_COPY.replacing);
assert.equal(keyText({}, {}), null);
assert.deepEqual(KEY_COPY, { saved: 'A key is saved.', none: 'No key saved.', clearing: 'The saved key is removed when you save.',
replacing: 'The new key replaces the saved one when you save.' });
});
test('LoRA rows show detected / enabled / cache-loaded separately', async () => {
const { detectionText, cacheText } = await integrations;
assert.equal(detectionText({ lora_detected: true, lora_detected_path: 'custom_nodes/ComfyUI-Lora-Manager' }),
'Detected: custom_nodes/ComfyUI-Lora-Manager.');
assert.equal(detectionText({ lora_detected: true, lora_detected_path: '' }), 'Detected.');
assert.equal(detectionText({ lora_detected: false }), 'Not detected in ComfyUI’s custom_nodes folders.');
assert.equal(detectionText({}), null);
assert.equal(cacheText({ lora_cache_loaded: true }), 'Trigger words are loaded.');
assert.equal(cacheText({ lora_cache_loaded: false }), 'Trigger words are not loaded.');
assert.equal(cacheText({}), null);
});
test('host compatibility shows the detected renderer as status (D106)', async () => {
const { rendererText, RENDERER_COPY } = await integrations;
const host = (...renderers) => ({ hostId: 'h', label: 'PromptManager — ComfyUI', targets: renderers.map((renderer, id) =>
({ id: String(id), roles: ['positive'], renderer })) });
assert.equal(rendererText([]), RENDERER_COPY.none);
assert.equal(rendererText(null), RENDERER_COPY.none);
assert.equal(rendererText([host()]), RENDERER_COPY.noNodes);
assert.equal(rendererText([host('vue')]), RENDERER_COPY.noNodes);
assert.equal(rendererText([host('classic')]), 'Detected: Classic (1 ComfyUI window).');
assert.equal(rendererText([host('nodes2', 'nodes2')]), 'Detected: Nodes 2.0 (1 ComfyUI window).');
assert.equal(rendererText([host('nodes2'), host('classic')]), 'Detected: Classic and Nodes 2.0 (2 ComfyUI windows).');
});
test('the import review shows replacement, blocked items and network effects (D107)', async () => {
const { importGrid } = await integrations;
const plan = { lora_files: 3, with_metadata: 2, current_imports: 5, blocked_by_pairs: 0, planned_downloads: 4,
scope: { count: 2, label: 'Replace 5 imported prompts with 2 LoRAs' }, protected: { count: 0, reasons: [] },
backup: { policy: 'on', destination_label: 'prompts.db.backup_<date and time>' }, network: { planned_downloads: 4, api_key: true },
media: { deletes_files: false, label: 'No files are removed; image links are replaced' } };
assert.deepEqual(importGrid(plan), [
['Scope', 'Replace 5 imported prompts with 2 LoRAs'],
['Found', '3 metadata files · 2 readable'],
['Protected', 'Nothing is protected'],
['Network', 'Downloads up to 4 example images from CivitAI with your saved key'],
['Backup', 'Create and verify before changes (prompts.db.backup_<date and time>)'],
['Media on disk', 'No files are removed; image links are replaced'],
]);
const blocked = importGrid({ ...plan, planned_downloads: 1, network: { api_key: false }, protected: { count: 2,
reasons: ['2 imported prompts used by saved pairs'] } });
assert.deepEqual(blocked[2], ['Protected', '2 imported prompts used by saved pairs']);
assert.deepEqual(blocked[3], ['Network', 'Downloads up to 1 example image from CivitAI']);
assert.deepEqual(importGrid({})[0], ['Scope', '0 LoRAs']);
assert.deepEqual(importGrid({})[3], ['Network', 'No downloads']);
assert.deepEqual(importGrid({})[5], ['Media on disk', 'No files are removed']);
});
test('a LoRA import job reads in Activity: tiles, outcomes and the kept collection on failure', async () => {
const { kindOf, finishToast, outcomeLine, viewOf } = await kinds;
const view = kindOf('lora_import');
assert.equal(view.pillVerb, 'Importing LoRAs');
assert.deepEqual(view.tiles.map(([group]) => group), ['imported', 'linked', 'skipped', 'downloaded', 'download_errors', 'errors']);
const counters = { imported: 3, linked: 1, skipped: 0, downloaded: 2, download_errors: 0, errors: 0 };
const job = { kind: 'lora_import', title: 'LoRA reference import', state: 'finished', cancellable: false, counters };
assert.equal(outcomeLine(job), '3 imported · 0 errors');
assert.equal(finishToast(job), 'LoRA import finished · 3 imported · 0 errors');
assert.equal(finishToast({ ...job, state: 'cancelled' }), 'LoRA import cancelled · 3 imported');
const failed = { ...job, state: 'failed', status_line: 'The import failed; the previous LoRA imports were kept.' };
assert.equal(finishToast(failed), 'LoRA import failed · The import failed; the previous LoRA imports were kept.');
assert.equal(finishToast({ ...failed, status_line: '' }), 'LoRA import failed · 3 imported · 0 errors');
assert.equal(viewOf(job).tiles.length, 6);
// The legacy page's held import (no counters) reads as the generic job.
assert.equal(viewOf({ kind: 'lora_import', title: 'LoRA import (admin page)', cancellable: false, counters: {} }).tiles.length, 0);
});
test('a failed import the files or saved pairs overtook is reviewed again in Integrations', async () => {
const { reviewAgainOf } = await import('../../web/js/workspace/activity-job-view.mjs');
const failed = failure => ({ kind: 'lora_import', state: 'failed', scope: { operation: 'lora_import', failure } });
for (const failure of ['plan-changed', 'storage-changed', 'plan-expired', 'blocked-by-pairs']) {
assert.deepEqual(reviewAgainOf(failed(failure)), { section: '#/system/integrations' }, failure);
}
assert.equal(reviewAgainOf(failed('import-failed')), null);
assert.equal(reviewAgainOf({ ...failed('plan-changed'), state: 'cancelled' }), null);
});