root
62 lines
2.6 KiB
TypeScript
62 lines
2.6 KiB
TypeScript
import { beforeEach, describe, expect, it, vi } from 'vitest'
|
|
|
|
vi.mock('../lib/prisma', () => ({
|
|
prisma: {
|
|
contractSettings: { findUnique: vi.fn() },
|
|
additionalDriver: { createMany: vi.fn() },
|
|
reservation: { update: vi.fn() },
|
|
$transaction: vi.fn(),
|
|
},
|
|
}))
|
|
|
|
import { prisma } from '../lib/prisma'
|
|
import { applyAdditionalDriversToReservation, calculateAdditionalDriverCharge } from './additionalDriverService'
|
|
|
|
describe('additionalDriverService', () => {
|
|
beforeEach(() => {
|
|
vi.clearAllMocks()
|
|
vi.setSystemTime(new Date('2026-06-09T12:00:00.000Z'))
|
|
})
|
|
|
|
it('calculates additional driver charges by configured charge mode', () => {
|
|
expect(calculateAdditionalDriverCharge('PER_DAY', 40, 5)).toBe(200)
|
|
expect(calculateAdditionalDriverCharge('FLAT', 150, 5)).toBe(150)
|
|
expect(calculateAdditionalDriverCharge('FREE', 150, 5)).toBe(0)
|
|
})
|
|
|
|
it('returns a zero-impact result without touching Prisma when no drivers are supplied', async () => {
|
|
await expect(applyAdditionalDriversToReservation('reservation_1', 'company_1', [], 4)).resolves.toEqual({
|
|
records: [],
|
|
additionalDriverTotal: 0,
|
|
requiresManualApproval: false,
|
|
})
|
|
expect(prisma.contractSettings.findUnique).not.toHaveBeenCalled()
|
|
})
|
|
|
|
it('applies charge settings, license risk flags, and reservation totals in one transaction', async () => {
|
|
vi.mocked(prisma.contractSettings.findUnique).mockResolvedValue({
|
|
additionalDriverCharge: 'PER_DAY',
|
|
additionalDriverDailyRate: 75,
|
|
additionalDriverFlatRate: 300,
|
|
} as never)
|
|
vi.mocked(prisma.$transaction).mockResolvedValue([] as never)
|
|
|
|
const result = await applyAdditionalDriversToReservation('reservation_1', 'company_1', [
|
|
{ firstName: 'Nora', lastName: 'Safe', driverLicense: 'A123', licenseExpiry: '2027-06-09T12:00:00.000Z' },
|
|
{ firstName: 'Imad', lastName: 'Risky', driverLicense: 'B456', licenseExpiry: '2026-06-20T12:00:00.000Z' },
|
|
], 4)
|
|
|
|
expect(result.additionalDriverTotal).toBe(600)
|
|
expect(result.requiresManualApproval).toBe(true)
|
|
expect(result.records).toEqual([
|
|
expect.objectContaining({ firstName: 'Nora', totalCharge: 300, requiresApproval: false, approvalNote: null }),
|
|
expect.objectContaining({ firstName: 'Imad', totalCharge: 300, requiresApproval: true, licenseExpiringSoon: true }),
|
|
])
|
|
expect(prisma.additionalDriver.createMany).toHaveBeenCalledWith({ data: result.records })
|
|
expect(prisma.reservation.update).toHaveBeenCalledWith({
|
|
where: { id: 'reservation_1' },
|
|
data: { additionalDriverTotal: 600, totalAmount: { increment: 600 } },
|
|
})
|
|
})
|
|
})
|