john
79 lines
3.1 KiB
JavaScript
79 lines
3.1 KiB
JavaScript
import assert from 'node:assert/strict';
|
|
import fs from 'node:fs';
|
|
import test from 'node:test';
|
|
import { createMindSpaceImageGenerationService } from './mindspace-image-generation.mjs';
|
|
|
|
test('MindSpace asset schema accepts image_make as a source type', () => {
|
|
const schemaSql = fs.readFileSync(new URL('./schema.sql', import.meta.url), 'utf8');
|
|
const dbSource = fs.readFileSync(new URL('./db.mjs', import.meta.url), 'utf8');
|
|
assert.match(schemaSql, /source_type ENUM\([^\n]*'image_make'/);
|
|
assert.match(dbSource, /MODIFY source_type ENUM\([^\n]*'image_make'/);
|
|
});
|
|
|
|
test('MindSpace image generation stays inert while the admin purpose is disabled', async () => {
|
|
const service = createMindSpaceImageGenerationService({
|
|
configService: {
|
|
async resolveImageGenerationPurpose() {
|
|
return { ok: false, code: 'purpose_disabled', message: 'disabled' };
|
|
},
|
|
},
|
|
});
|
|
assert.deepEqual(await service.generate({ userId: 'user-1', purpose: 'hero', prompt: 'x' }), {
|
|
ok: false,
|
|
code: 'purpose_disabled',
|
|
message: 'disabled',
|
|
fallback: true,
|
|
});
|
|
});
|
|
|
|
test('MindSpace image generation stores the result before acknowledging temporary output', async () => {
|
|
const order = [];
|
|
const service = createMindSpaceImageGenerationService({
|
|
configService: {
|
|
async resolveImageGenerationPurpose() {
|
|
return { ok: true, presetId: 'memind_dark_hero' };
|
|
},
|
|
},
|
|
imageMakeClient: {
|
|
async generateImage(input) {
|
|
order.push(['generate', input.presetId]);
|
|
return {
|
|
jobId: 'job-1', buffer: Buffer.from('image'), mimeType: 'image/webp',
|
|
sha256: 'abc', width: 512, height: 512,
|
|
};
|
|
},
|
|
async acknowledge(jobId, assetId) { order.push(['ack', jobId, assetId]); },
|
|
},
|
|
assetService: {
|
|
async createChatAsset(userId, input) {
|
|
order.push(['store', userId, input.sourceType, input.categoryCode]);
|
|
return { id: 'asset-1', publicUrl: 'https://example.test/image.webp' };
|
|
},
|
|
},
|
|
});
|
|
const result = await service.generate({ userId: 'user-1', purpose: 'hero', prompt: 'x' });
|
|
assert.equal(result.ok, true);
|
|
assert.equal(result.asset.id, 'asset-1');
|
|
assert.deepEqual(order, [
|
|
['generate', 'memind_dark_hero'],
|
|
['store', 'user-1', 'image_make', 'public'],
|
|
['ack', 'job-1', 'asset-1'],
|
|
]);
|
|
});
|
|
|
|
test('MindSpace image generation converts provider failures into an explicit fallback result', async () => {
|
|
const service = createMindSpaceImageGenerationService({
|
|
configService: { async resolveImageGenerationPurpose() { return { ok: true, presetId: 'x' }; } },
|
|
assetService: { async createChatAsset() { throw new Error('must not run'); } },
|
|
imageMakeClient: { async generateImage() { throw Object.assign(new Error('offline'), { code: 'IMAGE_MAKE_HTTP_ERROR' }); } },
|
|
logger: { warn() {} },
|
|
});
|
|
const result = await service.generate({ userId: 'user-1', purpose: 'hero', prompt: 'x' });
|
|
assert.deepEqual(result, {
|
|
ok: false,
|
|
fallback: true,
|
|
code: 'IMAGE_MAKE_HTTP_ERROR',
|
|
message: '图片生成暂不可用，已保留原有流程',
|
|
});
|
|
});
|