import { describe, expect, it } from 'vitest'
import {
    clampRowLimit,
    DEFAULT_ROW_LIMIT,
    formatDbError,
    MAX_ROW_LIMIT,
    normalizeCell,
    truncateRows
} from '../../src/format.js'

describe('clampRowLimit', () => {
    it('defaults when undefined', () => {
        expect(clampRowLimit(undefined)).toBe(DEFAULT_ROW_LIMIT)
    })

    it('clamps to max and min and truncates fractions', () => {
        expect(clampRowLimit(99999)).toBe(MAX_ROW_LIMIT)
        expect(clampRowLimit(0)).toBe(1)
        expect(clampRowLimit(-5)).toBe(1)
        expect(clampRowLimit(10.9)).toBe(10)
    })
})

describe('truncateRows', () => {
    it('passes short arrays through', () => {
        expect(truncateRows([1, 2], 5)).toEqual({ rows: [1, 2], truncated: false })
    })

    it('slices long arrays and flags truncation', () => {
        const { rows, truncated } = truncateRows([1, 2, 3], 2)
        expect(rows).toEqual([1, 2])
        expect(truncated).toBe(true)
    })
})

describe('normalizeCell', () => {
    it('converts bigint to string', () => {
        expect(normalizeCell(123n)).toBe('123')
    })

    it('converts Date to ISO string', () => {
        expect(normalizeCell(new Date('2026-01-02T03:04:05Z'))).toBe('2026-01-02T03:04:05.000Z')
    })

    it('converts Buffer to hex string', () => {
        expect(normalizeCell(Buffer.from([0xde, 0xad]))).toBe('\\xdead')
    })

    it('passes primitives and null through', () => {
        expect(normalizeCell('x')).toBe('x')
        expect(normalizeCell(5)).toBe(5)
        expect(normalizeCell(null)).toBeNull()
    })
})

describe('formatDbError', () => {
    it('formats code, detail and hint', () => {
        const error = Object.assign(new Error('relation "x" does not exist'), {
            code: '42P01',
            hint: 'check the table name'
        })
        expect(formatDbError('postgres', error)).toBe(
            '[postgres 42P01] relation "x" does not exist (check the table name)'
        )
    })

    it('prefers sqlMessage for mysql errors', () => {
        const error = Object.assign(new Error('outer'), {
            code: 'ER_NO_SUCH_TABLE',
            sqlMessage: "Table 'a.b' doesn't exist"
        })
        expect(formatDbError('mysql', error)).toBe(
            "[mysql ER_NO_SUCH_TABLE] Table 'a.b' doesn't exist"
        )
    })

    it('joins detail and hint', () => {
        const error = Object.assign(new Error('boom'), {
            code: '23505',
            detail: 'Key (id)=(1) already exists.',
            hint: 'use upsert'
        })
        expect(formatDbError('postgres', error)).toBe(
            '[postgres 23505] boom (Key (id)=(1) already exists. | use upsert)'
        )
    })

    it('handles non-Error values', () => {
        expect(formatDbError('mysql', 'boom')).toBe('[mysql] boom')
    })
})