fix: add parens

unplugin · Jun 22, 2022 · ddf5b28 · ddf5b28
1 parent fcbd54f
commit ddf5b28
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/src/core/transform.ts b/src/core/transform.ts
@@ -87,7 +87,7 @@ export const transform = async (
         s.overwrite(
           value.start!,
           value.end!,
-          s.slice(result.start!, result.end!)
+          `(${s.slice(result.start!, result.end!)})`
         )
         newAST = result
       }

diff --git a/tests/remove-wrapper-function.test.ts b/tests/remove-wrapper-function.test.ts
@@ -10,19 +10,23 @@ test('remove wrapper function', async () => {
     }
   })
   console.log(mutable({} as const))
+  console.log(() => mutable({} as const))
   `
   const options: Pick<OptionsResolved, 'parserOptions' | 'transformer'> = {
-    transformer: [RemoveWrapperFunction(['defineComponent', 'mutable'])],
+    transformer: [
+      RemoveWrapperFunction(['defineComponent', 'mutable', 'definePropType']),
+    ],
     parserOptions: {},
   }
   const code = (await transform(source, 'foo.ts', options))?.code
   expect(code).toMatchInlineSnapshot(`
-    "const comp = {
+    "const comp = ({
         render() {
           return []
         }
-      }
-      console.log({} as const)
+      })
+      console.log(({} as const))
+      console.log(() => ({} as const))
       "
   `)
 })