抽出上传参考图状态模型

扩展 ImageCanvasUploadModel 承载生成参考图上传后的状态转换

精简 useImageCanvasUploadWorkflow 中的参考图 dialog 写回逻辑

补充上传模型单测覆盖参考图写入和失败态清理

更新 TRACKING.md 记录第四十五执行批次验证
This commit is contained in:
2026-06-17 20:21:07 +08:00
parent 7dec8b7a66
commit 6afc1cf920
4 changed files with 272 additions and 51 deletions

View File

@@ -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,