import { expect, test } from 'vitest'

import { transform } from '../src/core/transform.ts'

import { RemoveNode } from '../src/transformers.ts'

import { ast } from '../src/yuku.ts'

import type { OptionsResolved } from '../src/core/options.ts'

test('remove node', async () => {

const source = `const comp = defineComponent({

render() {

return []

}

})

console.log(mutable({} as const))

`

const options: Pick<OptionsResolved, 'parserOptions' | 'transformer'> = {

transformer: [

RemoveNode(

(node) =>

node.type === 'ReturnStatement' || ast.isCallOf(node, 'mutable'),

),

],

parserOptions: {},

}

const code = (await transform(source, 'foo.ts', options))?.toString()

expect(code).toMatchInlineSnapshot(`

"const comp = defineComponent({

render() {

}

})

console.log()

"

`)

})