抽出上传参考图状态模型
扩展 ImageCanvasUploadModel 承载生成参考图上传后的状态转换 精简 useImageCanvasUploadWorkflow 中的参考图 dialog 写回逻辑 补充上传模型单测覆盖参考图写入和失败态清理 更新 TRACKING.md 记录第四十五执行批次验证
This commit is contained in:
@@ -1,14 +1,18 @@
|
||||
import { describe, expect, it } from 'vitest';
|
||||
import { afterEach, describe, expect, it, vi } from 'vitest';
|
||||
|
||||
import type {
|
||||
CanvasLayer,
|
||||
EditorAssetFolder,
|
||||
GenerateDialogState,
|
||||
} from './ImageCanvasEditorTypes';
|
||||
import {
|
||||
applyGenerationReferenceUpload,
|
||||
createUploadCanvasLayer,
|
||||
createUploadedGenerationReference,
|
||||
createUploadingAssetPlaceholder,
|
||||
resizeUploadCanvasLayerToImage,
|
||||
resolveUploadFolderId,
|
||||
setFailedGenerationIdle,
|
||||
} from './ImageCanvasUploadModel';
|
||||
|
||||
function createFolder(
|
||||
@@ -43,7 +47,23 @@ function createLayer(overrides: Partial<CanvasLayer> = {}): CanvasLayer {
|
||||
};
|
||||
}
|
||||
|
||||
function createDialog(
|
||||
overrides: Partial<GenerateDialogState> = {},
|
||||
): GenerateDialogState {
|
||||
return {
|
||||
mode: 'character',
|
||||
prompt: '',
|
||||
status: 'failed',
|
||||
errorMessage: '旧错误',
|
||||
...overrides,
|
||||
};
|
||||
}
|
||||
|
||||
describe('ImageCanvasUploadModel', () => {
|
||||
afterEach(() => {
|
||||
vi.useRealTimers();
|
||||
});
|
||||
|
||||
it('resolves upload folder ids with project fallback', () => {
|
||||
const folders = [
|
||||
createFolder(),
|
||||
@@ -89,6 +109,115 @@ describe('ImageCanvasUploadModel', () => {
|
||||
});
|
||||
});
|
||||
|
||||
it('creates uploaded generation references with fallback labels', () => {
|
||||
vi.useFakeTimers();
|
||||
vi.setSystemTime(new Date('2026-06-17T12:00:00.000Z'));
|
||||
|
||||
expect(
|
||||
createUploadedGenerationReference({
|
||||
idPrefix: 'upload-character-reference',
|
||||
index: 2,
|
||||
fileName: '',
|
||||
fallbackLabel: '参考图3',
|
||||
imageSrc: 'data:image/png;base64,ref',
|
||||
}),
|
||||
).toEqual({
|
||||
id: 'upload-character-reference-1781697600000-2',
|
||||
label: '参考图3',
|
||||
src: 'data:image/png;base64,ref',
|
||||
});
|
||||
});
|
||||
|
||||
it('resets failed generation dialogs when reference uploads are applied', () => {
|
||||
expect(
|
||||
setFailedGenerationIdle(
|
||||
createDialog({ status: 'failed', errorMessage: '旧错误' }),
|
||||
),
|
||||
).toMatchObject({ status: 'idle', errorMessage: undefined });
|
||||
expect(
|
||||
setFailedGenerationIdle(createDialog({ status: 'generating' })),
|
||||
).toMatchObject({ status: 'generating', errorMessage: '旧错误' });
|
||||
});
|
||||
|
||||
it('applies uploaded references to matching generation dialogs', () => {
|
||||
const reference = {
|
||||
id: 'ref-a',
|
||||
label: '参考图',
|
||||
src: 'data:image/png;base64,ref',
|
||||
};
|
||||
|
||||
expect(
|
||||
applyGenerationReferenceUpload({
|
||||
dialog: createDialog({ mode: 'spec' }),
|
||||
target: 'spec-reference',
|
||||
references: [reference],
|
||||
}),
|
||||
).toMatchObject({
|
||||
mode: 'spec',
|
||||
status: 'idle',
|
||||
specReference: reference,
|
||||
errorMessage: undefined,
|
||||
});
|
||||
expect(
|
||||
applyGenerationReferenceUpload({
|
||||
dialog: createDialog({ mode: 'character' }),
|
||||
target: 'character-spec',
|
||||
references: [reference],
|
||||
}),
|
||||
).toMatchObject({
|
||||
mode: 'character',
|
||||
status: 'idle',
|
||||
characterSpecReference: reference,
|
||||
});
|
||||
expect(
|
||||
applyGenerationReferenceUpload({
|
||||
dialog: createDialog({ mode: 'icon' }),
|
||||
target: 'icon-spec',
|
||||
references: [reference],
|
||||
}),
|
||||
).toMatchObject({
|
||||
mode: 'icon',
|
||||
status: 'idle',
|
||||
iconSpecReference: reference,
|
||||
});
|
||||
});
|
||||
|
||||
it('appends character reference uploads without changing unmatched dialogs', () => {
|
||||
const previousReference = {
|
||||
id: 'ref-old',
|
||||
label: '旧参考图',
|
||||
src: 'data:image/png;base64,old',
|
||||
};
|
||||
const nextReference = {
|
||||
id: 'ref-new',
|
||||
label: '新参考图',
|
||||
src: 'data:image/png;base64,new',
|
||||
};
|
||||
|
||||
expect(
|
||||
applyGenerationReferenceUpload({
|
||||
dialog: createDialog({
|
||||
mode: 'character',
|
||||
characterReferences: [previousReference],
|
||||
}),
|
||||
target: 'character-reference',
|
||||
references: [nextReference],
|
||||
}),
|
||||
).toMatchObject({
|
||||
mode: 'character',
|
||||
status: 'idle',
|
||||
characterReferences: [previousReference, nextReference],
|
||||
});
|
||||
const iconDialog = createDialog({ mode: 'icon' });
|
||||
expect(
|
||||
applyGenerationReferenceUpload({
|
||||
dialog: iconDialog,
|
||||
target: 'character-reference',
|
||||
references: [nextReference],
|
||||
}),
|
||||
).toBe(iconDialog);
|
||||
});
|
||||
|
||||
it('creates upload canvas layers centered on the target canvas point', () => {
|
||||
const layer = createUploadCanvasLayer({
|
||||
uploadIndex: 1,
|
||||
|
||||
Reference in New Issue
Block a user