insights-js/packages/next/tests/waitUntil.test.ts

jest.mock('server-only', () => ({}))

const mockVercelWaitUntil = jest.fn()

jest.mock('@vercel/functions', () => ({
    waitUntil: mockVercelWaitUntil,
}))

const mockInsightsConstructor = jest.fn()

jest.mock('@hanzo/insights-node', () => ({
    Insights: mockInsightsConstructor,
}))

describe('nodeClientCache waitUntil auto-detection', () => {
    beforeEach(() => {
        jest.clearAllMocks()
        jest.resetModules()
    })

    it('auto-detects @vercel/functions waitUntil and applies to new clients', async () => {
        const { getOrCreateNodeClient } = require('../src/server/nodeClientCache')

        await getOrCreateNodeClient('phc_test', { host: 'https://test.com' })

        expect(mockInsightsConstructor).toHaveBeenCalledWith(
            'phc_test',
            expect.objectContaining({ waitUntil: mockVercelWaitUntil })
        )
    })

    it('explicit options.waitUntil takes priority over auto-detected', async () => {
        const { getOrCreateNodeClient } = require('../src/server/nodeClientCache')

        const explicitWaitUntil = jest.fn()
        await getOrCreateNodeClient('phc_test3', { host: 'https://test3.com', waitUntil: explicitWaitUntil })

        expect(mockInsightsConstructor).toHaveBeenCalledWith(
            'phc_test3',
            expect.objectContaining({ waitUntil: explicitWaitUntil })
        )
    })
})