import { expect, test } from 'vitest'

import { resolveTree } from '../../src/utils/registry'

test('resolve tree', async () => {
  const index = [
    {
      name: 'button',
      dependencies: ['@radix-ui/react-slot'],
      type: 'components:ui',
      files: ['button.tsx'],
    },
    {
      name: 'dialog',
      dependencies: ['@radix-ui/react-dialog'],
      registryDependencies: ['button'],
      type: 'components:ui',
      files: ['dialog.tsx'],
    },
    {
      name: 'input',
      registryDependencies: ['button'],
      type: 'components:ui',
      files: ['input.tsx'],
    },
    {
      name: 'alert-dialog',
      dependencies: ['@radix-ui/react-alert-dialog'],
      registryDependencies: ['button', 'dialog'],
      type: 'components:ui',
      files: ['alert-dialog.tsx'],
    },
    {
      name: 'example-card',
      type: 'components:component',
      files: ['example-card.tsx'],
      registryDependencies: ['button', 'dialog', 'input'],
    },
  ]

  expect(
    (await resolveTree(index, ['button'])).map(entry => entry.name).sort(),
  ).toEqual(['button'])

  expect(
    (await resolveTree(index, ['dialog'])).map(entry => entry.name).sort(),
  ).toEqual(['button', 'dialog'])

  expect(
    (await resolveTree(index, ['alert-dialog', 'dialog']))
      .map(entry => entry.name)
      .sort(),
  ).toEqual(['alert-dialog', 'button', 'dialog'])

  expect(
    (await resolveTree(index, ['example-card']))
      .map(entry => entry.name)
      .sort(),
  ).toEqual(['button', 'dialog', 'example-card', 'input'])

  expect(
    (await resolveTree(index, ['foo'])).map(entry => entry.name).sort(),
  ).toEqual([])

  expect(
    (await resolveTree(index, ['button', 'foo']))
      .map(entry => entry.name)
      .sort(),
  ).toEqual(['button'])
})