@tsed/testcontainers-mongo 8.0.1 → 8.0.2

package/package.json

@@ -2,13 +2,14 @@
   "name": "@tsed/testcontainers-mongo",
   "description": "A module to test Ts.ED code that use MongoDB as database using Test Containers",
   "type": "module",
-  "version": "8.0.1",
+  "version": "8.0.2",
   "source": "./src/index.ts",
   "main": "./lib/esm/index.js",
   "module": "./lib/esm/index.js",
   "typings": "./lib/types/index.d.ts",
   "exports": {
     ".": {
+      "@tsed/source": "./src/index.ts",
       "types": "./lib/types/index.d.ts",
       "import": "./lib/esm/index.js",
       "default": "./lib/esm/index.js"
@@ -29,11 +30,11 @@
   },
   "devDependencies": {
     "@testcontainers/mongodb": "^10.13.2",
-    "@tsed/barrels": "8.0.1",
-    "@tsed/core": "8.0.1",
-    "@tsed/di": "8.0.1",
-    "@tsed/platform-http": "8.0.1",
-    "@tsed/typescript": "8.0.1",
+    "@tsed/barrels": "8.0.2",
+    "@tsed/core": "8.0.2",
+    "@tsed/di": "8.0.2",
+    "@tsed/platform-http": "8.0.2",
+    "@tsed/typescript": "8.0.2",
     "eslint": "9.12.0",
     "mongodb": "6.9.0",
     "testcontainers": ">=10.11.0",
@@ -41,9 +42,9 @@
   },
   "peerDependencies": {
     "@testcontainers/mongodb": ">=10.11.0",
-    "@tsed/core": "8.0.1",
-    "@tsed/di": "8.0.1",
-    "@tsed/platform-http": "8.0.1",
+    "@tsed/core": "8.0.2",
+    "@tsed/di": "8.0.2",
+    "@tsed/platform-http": "8.0.2",
     "mongodb": ">=6",
     "testcontainers": ">=10.11.0"
   },

package/src/index.ts

@@ -0,0 +1 @@
+export * from "./services/TestContainersMongo.js";

package/src/services/ContainerUtils.ts

@@ -0,0 +1,84 @@
+import type {ConnectionOptions} from "node:tls";
+
+import {MongoDBContainer, StartedMongoDBContainer} from "@testcontainers/mongodb";
+import type {ConnectOptions} from "mongoose";
+
+declare global {
+  namespace NodeJS {
+    interface Global {
+      TEST_CONTAINER_MONGO: StartedMongoDBContainer | null;
+    }
+  }
+}
+
+const KEY = "TEST_CONTAINER_MONGO";
+
+// @ts-ignore
+global[KEY] = null;
+
+function getEnvironment<T>(key: string): T | null {
+  // @ts-ignore
+  return global[key];
+}
+
+function setEnvironment(key: string, environment: unknown) {
+  // @ts-ignore
+  global[key] = environment;
+}
+
+function createMongoContainer(image = "mongo:6.0.1") {
+  return new MongoDBContainer(image).start();
+}
+
+export async function startMongoServer(image?: string) {
+  // await stopMockServer()
+  const container = getEnvironment<StartedMongoDBContainer>(KEY) || (await createMongoContainer(image));
+
+  setEnvironment(KEY, container);
+
+  process.env.MONGO_URL = container.getConnectionString();
+
+  return {
+    container,
+    url: process.env.MONGO_URL
+  };
+}
+
+export async function stopMongoServer() {
+  const container = getEnvironment<StartedMongoDBContainer>(KEY);
+
+  process.stdout.write("Stopping mongo server\n");
+  if (container) {
+    await container.stop();
+  }
+}
+
+export function getMongoUrl(): string {
+  return process.env.MONGO_URL as string;
+}
+
+export function getMongoConnectionOptions(
+  id: string = "",
+  opts: ConnectOptions = {}
+): {
+  id: string;
+  url: string;
+  connectionOptions: ConnectOptions;
+} {
+  const seed = Math.floor(Math.random() * 10000);
+  const {dbName = "db-test", ...otherOpts} = opts;
+  const url = `${getMongoUrl()}/${dbName}-${seed}`;
+
+  return {
+    id,
+    url,
+    connectionOptions: {
+      directConnection: true,
+      ...otherOpts
+    } satisfies ConnectOptions
+  };
+}
+
+export function getMongoConnectionsOptions(opts: ConnectionOptions = {}) {
+  return [getMongoConnectionOptions("default", opts)];
+}

package/src/services/TestContainersMongo.ts

@@ -0,0 +1,55 @@
+import {PlatformTest} from "@tsed/platform-http/testing";
+import {MongoClient} from "mongodb";
+
+import {getMongoConnectionOptions, getMongoConnectionsOptions, startMongoServer, stopMongoServer} from "./ContainerUtils.js";
+
+export class TestContainersMongo {
+  static startMongoServer = startMongoServer;
+  static stopMongoServer = stopMongoServer;
+  static getMongoConnectionsOptions = getMongoConnectionsOptions;
+  static getMongoConnectionOptions = getMongoConnectionOptions;
+
+  private static configuration: ReturnType<typeof getMongoConnectionsOptions>;
+
+  static create(options: Partial<TsED.Configuration> = {}) {
+    TestContainersMongo.configuration = getMongoConnectionsOptions();
+
+    return PlatformTest.create({
+      ...options,
+      mongoose: TestContainersMongo.configuration
+    });
+  }
+
+  static bootstrap(mod: unknown, opts: Partial<TsED.Configuration> = {}) {
+    return () => {
+      TestContainersMongo.configuration = getMongoConnectionsOptions();
+      return PlatformTest.bootstrap(mod, {
+        mongoose: TestContainersMongo.configuration,
+        ...opts
+      })();
+    };
+  }
+
+  static async reset(collectionName?: string) {
+    if (typeof collectionName === "string") {
+      await TestContainersMongo.cleanCollection(collectionName);
+    }
+
+    return PlatformTest.reset();
+  }
+
+  static async cleanCollection(collectionName: string, mongoSettings = TestContainersMongo.configuration) {
+    try {
+      const client = new MongoClient(mongoSettings[0].url, {
+        directConnection: true
+      });
+
+      await client.connect();
+      const db = client.db();
+      await db.collection(collectionName).deleteMany({});
+      await client.close();
+    } catch (er) {
+      console.error(`Collection ${collectionName} not found. ${er.message} ${er.stack}`);
+    }
+  }
+}

package/src/setup/vi.setup.ts

@@ -0,0 +1,9 @@
+import {startMongoServer, stopMongoServer} from "../services/ContainerUtils.js";
+
+export async function setup() {
+  await startMongoServer();
+}
+
+export async function teardown() {
+  await stopMongoServer();
+}