fix: dynamicImportSettled also waits for nested imports (#2389)

* fix: dynamicImportSettled also waits for nested imports

* chore: wait only not evaluated modules

* chore: always mark evaluated

* chore: mark external modules evaluated

Author: sheremet-va

packages/vite-node/src/client.ts

@@ -247,6 +247,7 @@ export class ViteNodeRunner {
       debugNative(externalize)
       const exports = await this.interopedImport(externalize)
       mod.exports = exports
+      mod.evaluated = true
       return exports
     }
 
@@ -286,7 +287,7 @@ export class ViteNodeRunner {
       },
     })
 
-    Object.assign(mod, { code: transformed, exports })
+    Object.assign(mod, { code: transformed, exports, evaluated: false })
 
     const __filename = fileURLToPath(url)
     const moduleProxy = {
@@ -347,7 +348,12 @@ export class ViteNodeRunner {
       columnOffset: -codeDefinition.length,
     })
 
-    await fn(...Object.values(context))
+    try {
+      await fn(...Object.values(context))
+    }
+    finally {
+      mod.evaluated = true
+    }
 
     return exports
   }

packages/vite-node/src/types.ts

@@ -44,6 +44,7 @@ export type CreateHotContextFunction = (runner: ViteNodeRunner, url: string) =>
 export interface ModuleCache {
   promise?: Promise<any>
   exports?: any
+  evaluated?: boolean
   code?: string
   map?: RawSourceMap
   /**

packages/vitest/src/integrations/vi.ts

@@ -247,19 +247,24 @@ class VitestUtils {
   /**
    * Wait for all imports to load.
    * Useful, if you have a synchronous call that starts
-   * importing a module, that you cannot wait otherwise.
+   * importing a module that you cannot wait otherwise.
    */
   public async dynamicImportSettled() {
     const state = getWorkerState()
     const promises: Promise<unknown>[] = []
     for (const mod of state.moduleCache.values()) {
-      if (mod.promise)
+      if (mod.promise && !mod.evaluated)
         promises.push(mod.promise)
     }
+    if (!promises.length)
+      return
     await Promise.allSettled(promises)
-    // wait until the end of the loop, so `.then` on modules called,
+    // wait until the end of the loop, so `.then` on modules is called,
     // like in import('./example').then(...)
-    await new Promise(resolve => setTimeout(resolve, 1)).then(() => Promise.resolve())
+    // also call dynamicImportSettled again in case new imports were added
+    await new Promise(resolve => setTimeout(resolve, 1))
+      .then(() => Promise.resolve())
+      .then(() => this.dynamicImportSettled())
   }
 
   private _config: null | ResolvedConfig = null