import { describe, it, expect } from 'vitest'
import {
  licenseTierBadgeClass,
  commercialBadgeClass,
  groupUseCases,
  type UseCaseRow,
} from './_helpers'

const uc = (over: Partial<UseCaseRow>): UseCaseRow => ({
  key: 'x',
  label: 'X',
  group: 'security',
  regulations: [],
  verification_methods: [],
  mapped_controls: 0,
  atom_total: 0,
  atom_relevant: 0,
  ...over,
})

describe('coverage helpers', () => {
  it('license tier badge classes', () => {
    expect(licenseTierBadgeClass(1)).toContain('green')
    expect(licenseTierBadgeClass(2)).toContain('blue')
    expect(licenseTierBadgeClass(3)).toContain('amber')
    expect(licenseTierBadgeClass(null)).toContain('gray')
  })

  it('commercial-use badge classes', () => {
    expect(commercialBadgeClass('allowed')).toContain('green')
    expect(commercialBadgeClass('restricted')).toContain('amber')
    expect(commercialBadgeClass('prohibited')).toContain('red')
    expect(commercialBadgeClass(null)).toContain('gray')
  })

  it('groups use-cases in stable order and sorts by relevant desc', () => {
    const groups = groupUseCases([
      uc({ key: 'a', group: 'security', atom_relevant: 5 }),
      uc({ key: 'b', group: 'security', atom_relevant: 15 }),
      uc({ key: 'c', group: 'document', atom_relevant: 1 }),
    ])
    expect(groups[0].group).toBe('document')
    expect(groups[1].group).toBe('security')
    expect(groups[1].rows[0].key).toBe('b')
    expect(groups[1].rows[1].key).toBe('a')
  })

  it('appends unknown groups after the known order', () => {
    const groups = groupUseCases([
      uc({ key: 'z', group: 'mystery', atom_relevant: 9 }),
      uc({ key: 'd', group: 'document', atom_relevant: 2 }),
    ])
    expect(groups.map((g) => g.group)).toEqual(['document', 'mystery'])
  })
})