feat:Strengthen the testing framework

tests/system/generate-image.system.test.ts

@@ -0,0 +1,142 @@
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
+import { callRoute } from '../integration/api/helpers/call-route'
+import { installAuthMocks, mockAuthenticated, resetAuthMockState } from '../helpers/auth'
+import { resetSystemState } from '../helpers/db-reset'
+import { prisma } from '../helpers/prisma'
+import { seedMinimalDomainState } from './helpers/seed'
+import { expectLifecycleEvents, listTaskEventTypes, waitForTaskTerminalState } from './helpers/tasks'
+import { startSystemWorkers, stopSystemWorkers, type SystemWorkers } from './helpers/workers'
+
+const imageState = vi.hoisted(() => ({
+  mode: 'success' as 'success' | 'fatal',
+  cosKey: 'cos/system-image-generated.png',
+  errorMessage: 'IMAGE_GENERATION_FATAL',
+}))
+
+vi.mock('@/lib/workers/handlers/image-task-handler-shared', async () => {
+  const actual = await vi.importActual<typeof import('@/lib/workers/handlers/image-task-handler-shared')>(
+    '@/lib/workers/handlers/image-task-handler-shared',
+  )
+  return {
+    ...actual,
+    generateLabeledImageToCos: vi.fn(async () => {
+      if (imageState.mode === 'fatal') {
+        throw new Error(imageState.errorMessage)
+      }
+      return imageState.cosKey
+    }),
+  }
+})
+
+vi.mock('@/lib/media/outbound-image', async () => {
+  const actual = await vi.importActual<typeof import('@/lib/media/outbound-image')>('@/lib/media/outbound-image')
+  return {
+    ...actual,
+    normalizeReferenceImagesForGeneration: vi.fn(async (refs: string[]) => refs.map((item) => `normalized:${item}`)),
+  }
+})
+
+describe('system - generate image', () => {
+  let workers: SystemWorkers = {}
+
+  beforeEach(async () => {
+    vi.resetModules()
+    vi.clearAllMocks()
+    imageState.mode = 'success'
+    imageState.cosKey = 'cos/system-image-generated.png'
+    imageState.errorMessage = 'IMAGE_GENERATION_FATAL'
+    await resetSystemState()
+    installAuthMocks()
+  })
+
+  afterEach(async () => {
+    await stopSystemWorkers(workers)
+    workers = {}
+    resetAuthMockState()
+  })
+
+  it('route -> queue -> worker -> db writes imageUrl and lifecycle events', async () => {
+    const seeded = await seedMinimalDomainState()
+    mockAuthenticated(seeded.user.id)
+    workers = await startSystemWorkers(['image'])
+
+    const mod = await import('@/app/api/novel-promotion/[projectId]/generate-image/route')
+    const response = await callRoute(
+      mod.POST,
+      'POST',
+      {
+        locale: 'zh',
+        type: 'character',
+        id: seeded.character.id,
+        appearanceId: seeded.appearance.id,
+        count: 1,
+      },
+      { params: { projectId: seeded.project.id } },
+    )
+
+    expect(response.status).toBe(200)
+    const json = await response.json() as { async: boolean; taskId: string }
+    expect(json.async).toBe(true)
+    expect(typeof json.taskId).toBe('string')
+
+    const task = await waitForTaskTerminalState(json.taskId)
+    expect(task.status).toBe('completed')
+    expect(task.type).toBe('image_character')
+    expect(task.targetId).toBe(seeded.appearance.id)
+
+    const appearance = await prisma.characterAppearance.findUnique({
+      where: { id: seeded.appearance.id },
+      select: { imageUrl: true, imageUrls: true, selectedIndex: true },
+    })
+    expect(appearance).toEqual({
+      imageUrl: imageState.cosKey,
+      imageUrls: JSON.stringify([imageState.cosKey]),
+      selectedIndex: 0,
+    })
+
+    const eventTypes = await listTaskEventTypes(json.taskId)
+    expectLifecycleEvents(eventTypes, 'completed')
+  })
+
+  it('fatal provider path -> task fails and existing appearance images stay unchanged', async () => {
+    const seeded = await seedMinimalDomainState()
+    mockAuthenticated(seeded.user.id)
+    imageState.mode = 'fatal'
+    imageState.errorMessage = 'IMAGE_GENERATION_FATAL'
+    workers = await startSystemWorkers(['image'])
+
+    const originalAppearance = await prisma.characterAppearance.findUnique({
+      where: { id: seeded.appearance.id },
+      select: { imageUrl: true, imageUrls: true, selectedIndex: true },
+    })
+
+    const mod = await import('@/app/api/novel-promotion/[projectId]/generate-image/route')
+    const response = await callRoute(
+      mod.POST,
+      'POST',
+      {
+        locale: 'zh',
+        type: 'character',
+        id: seeded.character.id,
+        appearanceId: seeded.appearance.id,
+        count: 1,
+      },
+      { params: { projectId: seeded.project.id } },
+    )
+
+    expect(response.status).toBe(200)
+    const json = await response.json() as { taskId: string }
+    const task = await waitForTaskTerminalState(json.taskId)
+    expect(task.status).toBe('failed')
+    expect(task.errorMessage).toContain('IMAGE_GENERATION_FATAL')
+
+    const appearance = await prisma.characterAppearance.findUnique({
+      where: { id: seeded.appearance.id },
+      select: { imageUrl: true, imageUrls: true, selectedIndex: true },
+    })
+    expect(appearance).toEqual(originalAppearance)
+
+    const eventTypes = await listTaskEventTypes(json.taskId)
+    expectLifecycleEvents(eventTypes, 'failed')
+  })
+})