From 08079eccef976deb9c5db48ea7fc4e316afca1c1 Mon Sep 17 00:00:00 2001
From: julien huang <julien.huang@outlook.fr>
Date: Sat, 19 Aug 2023 12:14:34 +0200
Subject: [PATCH] test: test transform island with dev mode

---
 packages/nuxt/test/islandTransform.test.ts | 52 +++++++++++++++++++---
 1 file changed, 45 insertions(+), 7 deletions(-)

diff --git a/packages/nuxt/test/islandTransform.test.ts b/packages/nuxt/test/islandTransform.test.ts
index 75c80d3c60..71692fffb5 100644
--- a/packages/nuxt/test/islandTransform.test.ts
+++ b/packages/nuxt/test/islandTransform.test.ts
@@ -6,7 +6,7 @@ import { islandsTransform } from '../src/components/islandsTransform'
 import { normalizeLineEndings } from './utils'
 
 const getComponents = () => [{
-  filePath: 'hello.server.vue',
+  filePath: '/root/hello.server.vue',
   mode: 'server',
   pascalName: 'HelloWorld',
   island: true,
@@ -19,22 +19,26 @@ const getComponents = () => [{
 }] as Component[]
 
 const pluginVite = islandsTransform.raw({
+  getComponents
+}, { framework: 'vite' }) as Plugin
+
+const pluginViteDev = islandsTransform.raw({
   getComponents,
-  rootDir: ''
+  rootDir: '/root',
+  isDev: true
 }, { framework: 'vite' }) as Plugin
 
 const pluginWebpack = islandsTransform.raw({
-  getComponents,
-  rootDir: ''
+  getComponents
 }, { framework: 'webpack', webpack: { compiler: {} as any } })
 
-const viteTransform = async (source: string, id: string) => {
-  const result = await (pluginVite.transform! as Function).call({ error: null, warn: null } as any, source, id)
+const viteTransform = async (source: string, id: string, dev = false) => {
+  const result = await ((dev ? pluginViteDev : pluginVite).transform! as Function)(source, id)
   return typeof result === 'string' ? result : result?.code
 }
 
 const webpackTransform = async (source: string, id: string) => {
-  const result = await ((pluginWebpack as UnpluginOptions).transform! as Function).call({ error: null, warn: null } as any, source, id)
+  const result = await ((pluginWebpack as UnpluginOptions).transform! as Function)(source, id)
   return typeof result === 'string' ? result : result?.code
 }
 
@@ -138,6 +142,40 @@ describe('islandTransform - server and island components', () => {
       `)
     })
 
+    it('test transform with vite in dev', async () => {
+      const result = await viteTransform(`<template>
+      <div>
+        <!-- should not be wrapped by TeleportIfClient -->
+        <HelloWorld />
+        <!-- should be wrapped by TeleportIfClient with a rootDir attr -->
+        <HelloWorld nuxt-client />
+      </div>
+    </template>
+    
+    <script setup lang="ts">
+    import HelloWorld from './HelloWorld.vue'
+    </script>
+    `, 'hello.server.vue', true)
+
+      expect(normalizeLineEndings(result)).toMatchInlineSnapshot(`
+        "<template>
+              <div>
+                <!-- should not be wrapped by TeleportIfClient -->
+                <HelloWorld />
+                <!-- should be wrapped by TeleportIfClient with a rootDir attr -->
+                <TeleportIfClient to=\\"HelloWorld-vxDirZrUwF\\" root-dir=\\"/root\\" :nuxt-client=\\"true\\"><HelloWorld /></TeleportIfClient>
+              </div>
+            </template>
+            
+            <script setup lang=\\"ts\\">
+        import { vforToArray as __vforToArray } from '#app/components/utils'
+        import TeleportIfClient from '#app/components/TeleportIfClient'
+            import HelloWorld from './HelloWorld.vue'
+            </script>
+            "
+      `)
+    })
+
     it('test transform with webpack', async () => {
       const spyOnWarn = vi.spyOn(console, 'warn')
       const result = await webpackTransform(`<template>