supabase/apps/docs/internals/markdown-schema/AiPrompt.test.ts

import { aiPrompts } from '~/data/ai-prompts.data'
import { fromMarkdown } from 'mdast-util-from-markdown'
import { describe, expect, it } from 'vitest'

import { AiPrompt } from './AiPrompt'

describe('AiPrompt markdown schema', () => {
  it('omits prompts by default', () => {
    expect(AiPrompt({ props: { id: 'nextjs' } })).toBe('')
  })

  it('serializes an opted-in prompt from the shared registry', () => {
    const markdown = AiPrompt({
      props: { id: 'nextjs', includeInMarkdown: true },
    })

    expect(markdown).toContain('**Agent Prompt**')
    expect(markdown).toContain('Help me add Supabase to my Next.js project.')
    expect(markdown).toContain('```text')
  })

  it.each(Object.keys(aiPrompts).filter((id) => id.startsWith('monitoring-')))(
    'exports the complete shared %s prompt when opted in',
    (id) => {
      const markdown = AiPrompt({ props: { id, includeInMarkdown: true } })
      const codeBlocks = fromMarkdown(markdown).children.filter((node) => node.type === 'code')
      expect(codeBlocks).toHaveLength(1)
      expect(codeBlocks[0]).toMatchObject({
        lang: 'text',
        value: aiPrompts[id as keyof typeof aiPrompts],
      })
    }
  )

  it('fails clearly for an unknown opted-in prompt', () => {
    expect(() => AiPrompt({ props: { id: 'missing-prompt', includeInMarkdown: true } })).toThrow(
      'Unknown AiPrompt id: missing-prompt'
    )
  })
})