79 lines
2.4 KiB
TypeScript
79 lines
2.4 KiB
TypeScript
import { beforeEach, describe, expect, it, vi } from 'vitest'
|
|
import type { ApiError } from '@/lib/api-errors'
|
|
import { buildDefaultTaskBillingInfo } from '@/lib/billing/task-policy'
|
|
import { submitTask } from '@/lib/task/submitter'
|
|
import { TASK_TYPE } from '@/lib/task/types'
|
|
import { prisma } from '../../helpers/prisma'
|
|
import { resetBillingState } from '../../helpers/db-reset'
|
|
import { createTestUser, seedBalance } from '../../helpers/billing-fixtures'
|
|
|
|
vi.mock('@/lib/task/queues', () => ({
|
|
addTaskJob: vi.fn(async () => ({ id: 'mock-job' })),
|
|
}))
|
|
|
|
vi.mock('@/lib/task/publisher', () => ({
|
|
publishTaskEvent: vi.fn(async () => ({})),
|
|
}))
|
|
|
|
describe('billing/submitter integration', () => {
|
|
beforeEach(async () => {
|
|
await resetBillingState()
|
|
process.env.BILLING_MODE = 'ENFORCE'
|
|
})
|
|
|
|
it('builds billing info server-side for billable task submission', async () => {
|
|
const user = await createTestUser()
|
|
await seedBalance(user.id, 10)
|
|
|
|
const result = await submitTask({
|
|
userId: user.id,
|
|
locale: 'en',
|
|
projectId: 'project-a',
|
|
type: TASK_TYPE.VOICE_LINE,
|
|
targetType: 'VoiceLine',
|
|
targetId: 'line-a',
|
|
payload: { maxSeconds: 5 },
|
|
})
|
|
|
|
expect(result.success).toBe(true)
|
|
const task = await prisma.task.findUnique({ where: { id: result.taskId } })
|
|
expect(task).toBeTruthy()
|
|
const billing = task?.billingInfo as { billable?: boolean; source?: string } | null
|
|
expect(billing?.billable).toBe(true)
|
|
expect(billing?.source).toBe('task')
|
|
})
|
|
|
|
it('marks task as failed when balance is insufficient', async () => {
|
|
const user = await createTestUser()
|
|
await seedBalance(user.id, 0)
|
|
|
|
const billingInfo = buildDefaultTaskBillingInfo(TASK_TYPE.VOICE_LINE, { maxSeconds: 10 })
|
|
expect(billingInfo?.billable).toBe(true)
|
|
|
|
await expect(
|
|
submitTask({
|
|
userId: user.id,
|
|
locale: 'en',
|
|
projectId: 'project-b',
|
|
type: TASK_TYPE.VOICE_LINE,
|
|
targetType: 'VoiceLine',
|
|
targetId: 'line-b',
|
|
payload: { maxSeconds: 10 },
|
|
billingInfo,
|
|
}),
|
|
).rejects.toMatchObject({ code: 'INSUFFICIENT_BALANCE' } satisfies Pick<ApiError, 'code'>)
|
|
|
|
const task = await prisma.task.findFirst({
|
|
where: {
|
|
userId: user.id,
|
|
type: TASK_TYPE.VOICE_LINE,
|
|
},
|
|
orderBy: { createdAt: 'desc' },
|
|
})
|
|
|
|
expect(task).toBeTruthy()
|
|
expect(task?.status).toBe('failed')
|
|
expect(task?.errorCode).toBe('INSUFFICIENT_BALANCE')
|
|
})
|
|
})
|