zpan/server/services/instance-telemetry.test.ts

import { beforeEach, describe, expect, it, vi } from 'vitest'
import { getOrCreateInstanceId } from '../licensing/instance-id'
import type { Database } from '../platform/interface'
import {
  INSTANCE_TELEMETRY_CRON,
  INSTANCE_TELEMETRY_ENDPOINT,
  INSTANCE_TELEMETRY_EVENT,
  INSTANCE_TELEMETRY_POSTHOG_PROJECT_TOKEN,
  reportInstanceTelemetry,
} from './instance-telemetry'

vi.mock('../licensing/instance-id', () => ({
  getOrCreateInstanceId: vi.fn(),
}))

describe('instance telemetry', () => {
  beforeEach(() => {
    vi.mocked(getOrCreateInstanceId).mockReset()
  })

  it('does not call the telemetry endpoint when PostHog project token is disabled', async () => {
    const fetchFn = vi.fn()

    const result = await reportInstanceTelemetry({
      db: {} as Database,
      config: { posthogProjectToken: '' },
      cron: INSTANCE_TELEMETRY_CRON,
      runtime: { target: 'cloudflare-worker' },
      fetchFn,
    })

    expect(result).toEqual({ reported: false, reason: 'disabled' })
    expect(fetchFn).not.toHaveBeenCalled()
    expect(getOrCreateInstanceId).not.toHaveBeenCalled()
  })

  it('captures the expected telemetry event with built-in PostHog endpoint and project token', async () => {
    vi.mocked(getOrCreateInstanceId).mockResolvedValue('inst-1')
    const fetchFn = vi.fn().mockResolvedValue(new Response('{}', { status: 200 }))

    const result = await reportInstanceTelemetry({
      db: {} as Database,
      config: {
        configuredInstanceId: 'configured-inst',
      },
      cron: INSTANCE_TELEMETRY_CRON,
      runtime: {
        target: 'node/docker',
        osPlatform: 'linux',
        osArch: 'arm64',
        osRelease: '6.8.0',
      },
      now: new Date('2026-06-08T12:00:00.000Z'),
      fetchFn,
    })

    expect(result).toEqual({ reported: true })
    expect(getOrCreateInstanceId).toHaveBeenCalledWith({}, 'configured-inst')
    expect(fetchFn).toHaveBeenCalledTimes(1)
    expect(fetchFn).toHaveBeenCalledWith(INSTANCE_TELEMETRY_ENDPOINT, {
      method: 'POST',
      headers: {
        'content-type': 'application/json',
      },
      body: expect.any(String),
    })

    const body = JSON.parse(fetchFn.mock.calls[0][1].body)
    expect(body).toMatchObject({
      api_key: INSTANCE_TELEMETRY_POSTHOG_PROJECT_TOKEN,
      event: INSTANCE_TELEMETRY_EVENT,
      distinct_id: 'inst-1',
      timestamp: '2026-06-08T12:00:00.000Z',
      properties: {
        instance_id: 'inst-1',
        app_version: '0.0.1',
        runtime_target: 'node/docker',
        os_platform: 'linux',
        os_arch: 'arm64',
        os_release: '6.8.0',
        cron: INSTANCE_TELEMETRY_CRON,
        report_interval: '12h',
        reported_at: '2026-06-08T12:00:00.000Z',
      },
    })
  })
})