Files
waooplus/tests/unit/guards/task-submit-compensation-guard.test.ts
2026-03-09 02:53:06 +08:00

44 lines
1.6 KiB
TypeScript

import { describe, expect, it } from 'vitest'
import { inspectTaskSubmitCompensation } from '../../../scripts/guards/task-submit-compensation-guard.mjs'
describe('task submit compensation guard', () => {
it('passes routes that create data before submitTask and define rollback handling', () => {
const content = `
async function rollbackCreatedRecord() {}
export const POST = apiHandler(async () => {
await prisma.panel.create({ data: {} })
try {
return await submitTask({})
} catch (error) {
await rollbackCreatedRecord()
throw error
}
})
`
expect(
inspectTaskSubmitCompensation('src/app/api/novel-promotion/[projectId]/panel-variant/route.ts', content),
).toEqual([])
})
it('ignores routes that do not combine create and submitTask', () => {
expect(inspectTaskSubmitCompensation('src/app/api/user/api-config/route.ts', 'await submitTask({})')).toEqual([])
expect(inspectTaskSubmitCompensation('src/app/api/projects/route.ts', 'await prisma.project.create({ data: {} })')).toEqual([])
})
it('flags routes that create data before submitTask without compensation marker', () => {
const content = `
export const POST = apiHandler(async () => {
await prisma.panel.create({ data: {} })
return await submitTask({})
})
`
expect(
inspectTaskSubmitCompensation('src/app/api/example/route.ts', content),
).toEqual([
'src/app/api/example/route.ts creates data before submitTask without explicit rollback/compensation marker',
])
})
})