test(ci): fix postgres fail on

Author: rrd108

package.json

@@ -53,7 +53,7 @@
     "nuxt-users": "./dist/cli.mjs"
   },
   "scripts": {
-    "prepack": "nuxt-module-build build && node scripts/post-build.js",
+    "prepack": "nuxt-module-build build --stub && node scripts/post-build.js",
     "postinstall": "node -e \"try { require('fs').chmodSync('./dist/cli.mjs', '755') } catch(e) {}\"",
     "predev": "node -v | grep -q 'v22' || (echo 'Please use Node.js v22 for development' && exit 1)",
     "dev": "yarn dev:prepare && nuxi dev playground",

src/runtime/server/utils/db.ts

@@ -1,6 +1,18 @@
 import { createDatabase } from 'db0'
+import type { Database } from 'db0'
 import type { ModuleOptions } from 'nuxt-users/utils'
 
+const dbCache = new Map<string, Database>()
+
+export const closeAllDbConnections = async () => {
+  for (const db of dbCache.values()) {
+    if (db && typeof (db as any).disconnect === 'function') {
+      await (db as any).disconnect()
+    }
+  }
+  dbCache.clear()
+}
+
 export const getConnector = async (name: string) => {
   try {
     switch (name) {
@@ -17,15 +29,21 @@ export const getConnector = async (name: string) => {
   catch (error) {
     if (error instanceof Error && error.message.includes('Cannot resolve')) {
       throw new Error(`Database connector "${name}" not found. Please install the required peer dependency:\n`
-        + '- For sqlite: yarn add better-sqlite3\n'
-        + '- For mysql: yarn add mysql2\n'
+        + '- For sqlite: yarn add better-sqlite3\n' 
+        + '- For mysql: yarn add mysql2\n' 
         + '- For postgresql: yarn add pg')
     }
     throw error
   }
 }
 
-export const useDb = async (options: ModuleOptions) => {
+export const useDb = async (options: ModuleOptions): Promise<Database> => {
+  const cacheKey = JSON.stringify(options.connector)
+
+  if (dbCache.has(cacheKey)) {
+    return dbCache.get(cacheKey)!
+  }
+
   const connectorName = options.connector!.name
   const connector = await getConnector(connectorName)
 
@@ -36,7 +54,9 @@ export const useDb = async (options: ModuleOptions) => {
   }
 
   try {
-    return createDatabase(connector(connectorOptions))
+    const db = createDatabase(connector(connectorOptions))
+    dbCache.set(cacheKey, db)
+    return db
   }
   catch (error) {
     console.warn(`[Nuxt Users] ⚠️  Failed to connect to ${connectorName} database:`, error instanceof Error ? error.message : 'Unknown error')
@@ -54,4 +74,4 @@ export const checkTableExists = async (options: ModuleOptions, tableName: string
     // Table doesn't exist or connection failed
     return false
   }
-}
+}

test/test-setup.ts

@@ -88,11 +88,6 @@ export const cleanupTestSetup = async (dbType: DatabaseType, db: Database, clean
     if (db) {
       try {
         await db.sql`DROP TABLE IF EXISTS {${tableName}}`
-        // Close the database connection
-        const dbWithDisconnect = db as { disconnect?: () => Promise<void> }
-        if (dbWithDisconnect.disconnect) {
-          await dbWithDisconnect.disconnect()
-        }
       }
       catch {
         // Ignore errors during cleanup

test/utils.user.test.ts

@@ -57,19 +57,6 @@ describe('User Utilities (src/utils/user.ts)', () => {
     await cleanupTestSetup(dbType, db, [testOptions.connector!.options.path!], testOptions.tables.users)
     await cleanupTestSetup(dbType, db, [testOptions.connector!.options.path!], testOptions.tables.passwordResetTokens)
     await cleanupTestSetup(dbType, db, [testOptions.connector!.options.path!], testOptions.tables.personalAccessTokens)
-
-    // Close the database connection
-    if (db) {
-      try {
-        const dbWithDisconnect = db as { disconnect?: () => Promise<void> }
-        if (dbWithDisconnect.disconnect) {
-          await dbWithDisconnect.disconnect()
-        }
-      }
-      catch {
-        // Ignore errors during cleanup
-      }
-    }
   })
 
   describe('createUser', () => {

test/vitest.setup.ts

@@ -0,0 +1,6 @@
+import { afterEach } from 'vitest'
+import { closeAllDbConnections } from '../src/runtime/server/utils/db'
+
+afterEach(async () => {
+  await closeAllDbConnections()
+})

vitest.config.ts

@@ -1,7 +1,14 @@
 import { defineConfig } from 'vitest/config'
+import { resolve } from 'node:path'
 
 export default defineConfig({
   test: {
     fileParallelism: false, // This will make all test files run sequentially
+    setupFiles: ['./test/vitest.setup.ts'],
+  },
+  resolve: {
+    alias: {
+      'nuxt-users/utils': resolve(__dirname, './src/utils'),
+    },
   },
 })