import { describe, expect, it } from 'vitest'
import { buildProjectLocationGenerateImageBody } from '@/lib/query/mutations/location-image-mutations'

describe('buildProjectLocationGenerateImageBody', () => {
  it('includes artStyle when generating a project location image', () => {
    expect(buildProjectLocationGenerateImageBody({
      locationId: 'location-1',
      count: 1,
      artStyle: 'japanese-anime',
    })).toEqual({
      type: 'location',
      id: 'location-1',
      imageIndex: undefined,
      count: 1,
      artStyle: 'japanese-anime',
    })
  })
})
