vitest-environment-mongodb 1.0.2 → 2.0.0

package/dist/mongodb-environment.js

@@ -1,24 +1,31 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const mongodb_memory_server_1 = require("mongodb-memory-server");
-exports.default = {
+import { MongoMemoryReplSet, MongoMemoryServer } from "mongodb-memory-server";
+let sharedMongoServer = null;
+let referenceCount = 0;
+export default {
     name: "mongodb",
-    transformMode: "ssr",
+    viteEnvironment: "ssr",
     async setup(_, options) {
-        // parse out options
-        const isReplicaSet = options?.replicaSet;
-        const serverOptions = options?.serverOptions ?? {};
-        const mongoUrlEnvName = options?.mongoUrlEnvName ?? "MONGO_CONNECTION_STRING";
-        // create server
-        const mongoServer = isReplicaSet
-            ? await mongodb_memory_server_1.MongoMemoryServer.create(serverOptions)
-            : await mongodb_memory_server_1.MongoMemoryReplSet.create(serverOptions);
-        // put the connection string in the environment
-        process.env[mongoUrlEnvName] = mongoServer.getUri();
+        referenceCount++;
+        if (!sharedMongoServer) {
+            // parse out options
+            const isReplicaSet = options?.replicaSet;
+            const serverOptions = options?.serverOptions ?? {};
+            const mongoUrlEnvName = options?.mongoUrlEnvName ?? "MONGO_CONNECTION_STRING";
+            // create server
+            sharedMongoServer = isReplicaSet
+                ? await MongoMemoryServer.create(serverOptions)
+                : await MongoMemoryReplSet.create(serverOptions);
+            // put the connection string in the environment
+            process.env[mongoUrlEnvName] = sharedMongoServer.getUri();
+        }
         return {
             async teardown() {
-                // stop the server
-                await mongoServer.stop();
+                referenceCount--;
+                // only stop the server if there are no more references to it
+                if (referenceCount === 0 && sharedMongoServer) {
+                    await sharedMongoServer.stop();
+                    sharedMongoServer = null;
+                }
             },
         };
     },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vitest-environment-mongodb",
-  "version": "1.0.2",
+  "version": "2.0.0",
   "description": "Sets up a MongoDB memory-server environment for testing with Vitest.",
   "main": "dist/mongodb-environment.js",
   "scripts": {
@@ -9,6 +9,7 @@
     "prepack": "npm run build",
     "semantic-release": "semantic-release"
   },
+  "repository": "https://github.com/jb-1980/vitest-environment-mongodb",
   "keywords": [],
   "author": "Joseph Gilgen <2569898+jb-1980@users.noreply.github.com>",
   "license": "MIT",
@@ -19,6 +20,6 @@
     "@types/node": "^22.13.14",
     "semantic-release": "^24.2.3",
     "typescript": "^5.8.2",
-    "vitest": "^3.0.9"
+    "vitest": "^4.0.16"
   }
 }