From 5316df67027c3cfe9d0f0590ea60e3b8c747db0d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=89=E5=92=B2=E6=99=BA=E5=AD=90=20Kevin=20Deng?=
 <sxzz@sxzz.moe>
Date: Fri, 13 Dec 2024 23:05:07 +0800
Subject: [PATCH] test: remove node transformer

---
 tests/remove-node.test.ts | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 tests/remove-node.test.ts

diff --git a/tests/remove-node.test.ts b/tests/remove-node.test.ts
new file mode 100644
index 0000000..1a85106
--- /dev/null
+++ b/tests/remove-node.test.ts
@@ -0,0 +1,34 @@
+import { isCallOf } from 'ast-kit'
+import { expect, test } from 'vitest'
+import { transform } from '../src/core/transform'
+import { RemoveNode } from '../src/transformers'
+import type { OptionsResolved } from '../src/core/options'
+
+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' || isCallOf(node, 'mutable'),
+      ),
+    ],
+    parserOptions: {},
+  }
+  const code = (await transform(source, 'foo.ts', options))?.code
+  expect(code).toMatchInlineSnapshot(`
+    "const comp = defineComponent({
+        render() {
+          
+        }
+      })
+      console.log()
+      "
+  `)
+})