import { describe, expect, it } from 'vitest'
import type { DatasetCreateResponse } from '../types'
import { isDetectionImageryDataset, isMapRasterDataset } from './datasetCapabilities'

function raster(sourceName: string, status = 'ready'): DatasetCreateResponse {
  return {
    id: sourceName,
    name: sourceName,
    dataset_type: 'raster',
    source: sourceName,
    source_name: sourceName,
    project_id: 'project-id',
    status,
  }
}

describe('dataset capabilities', () => {
  it('keeps every governed map raster, including WALOUS snapshots, available to the map', () => {
    expect(isMapRasterDataset(raster('spw_walous_land_cover'))).toBe(true)
    expect(isMapRasterDataset(raster('spw_terrain'))).toBe(true)
    expect(isMapRasterDataset(raster('unclassified_uploaded_imagery'))).toBe(false)
    expect(isMapRasterDataset(raster('spw_walous_land_cover', 'processing'))).toBe(false)
  })

  it('does not offer governed thematic rasters as AI detection imagery', () => {
    expect(isDetectionImageryDataset(raster('spw_walous_land_cover'))).toBe(false)
    expect(isDetectionImageryDataset(raster('unclassified_uploaded_imagery'))).toBe(true)
  })
})