fix tests

This commit is contained in:
Vikhyath Mondreti
2026-03-09 12:22:40 -07:00
parent 7b43091984
commit 15db69231f
2 changed files with 26 additions and 14 deletions

View File

@@ -26,6 +26,22 @@ vi.mock('@sim/db/schema', () => ({
},
}))
vi.mock('drizzle-orm', () => {
const sqlTag = () => {
const obj = { as: () => obj }
return obj
}
sqlTag.as = sqlTag
return {
sql: Object.assign(sqlTag, { raw: sqlTag }),
and: vi.fn(),
gte: vi.fn(),
lt: vi.fn(),
inArray: vi.fn(),
sum: () => ({ as: () => 'sum' }),
}
})
vi.mock('@sim/logger', () => loggerMock)
vi.mock('@/lib/billing/constants', () => ({
DAILY_REFRESH_RATE: 0.01,

View File

@@ -769,27 +769,23 @@ describe('Cost Calculation', () => {
})
describe('formatCost', () => {
it.concurrent('should format costs >= $1 with two decimal places', () => {
expect(formatCost(1.234)).toBe('$1.23')
expect(formatCost(10.567)).toBe('$10.57')
it.concurrent('should format dollar amounts as credits', () => {
expect(formatCost(1.234)).toBe('247 credits')
expect(formatCost(10.567)).toBe('2,113 credits')
})
it.concurrent('should format costs between 1¢ and $1 with three decimal places', () => {
expect(formatCost(0.0234)).toBe('$0.023')
expect(formatCost(0.1567)).toBe('$0.157')
it.concurrent('should show <1 credit for very small costs', () => {
expect(formatCost(0.0024)).toBe('<1 credit')
expect(formatCost(0.001)).toBe('<1 credit')
})
it.concurrent('should format costs between 0.1¢ and 1¢ with four decimal places', () => {
expect(formatCost(0.00234)).toBe('$0.0023')
expect(formatCost(0.00567)).toBe('$0.0057')
})
it.concurrent('should format very small costs with appropriate precision', () => {
expect(formatCost(0.000234)).toContain('$0.000234')
it.concurrent('should show credit count for small costs that round to at least 1', () => {
expect(formatCost(0.0234)).toBe('5 credits')
expect(formatCost(0.1567)).toBe('31 credits')
})
it.concurrent('should handle zero cost', () => {
expect(formatCost(0)).toBe('$0')
expect(formatCost(0)).toBe('0 credits')
})
it.concurrent('should handle undefined/null costs', () => {