new-test-setup-poc

Author: baileympearson

package.json

@@ -31,6 +31,7 @@
   "devDependencies": {
     "@babel/core": "7.26.10",
     "@babel/preset-env": "7.26.9",
+    "@mongodb-js/mongodb-downloader": "^0.3.9",
     "@typescript-eslint/eslint-plugin": "^8.19.1",
     "@typescript-eslint/parser": "^8.19.1",
     "acquit": "1.3.0",
@@ -58,6 +59,7 @@
     "mocha": "11.1.0",
     "moment": "2.30.1",
     "mongodb-memory-server": "10.1.4",
+    "mongodb-runner": "^5.8.2",
     "ncp": "^2.0.0",
     "nyc": "15.1.0",
     "pug": "3.0.3",
@@ -104,7 +106,7 @@
     "test-deno": "deno run --allow-env --allow-read --allow-net --allow-run --allow-sys --allow-write ./test/deno.js",
     "test-rs": "START_REPLICA_SET=1 mocha --timeout 30000 --exit ./test/*.test.js",
     "test-tsd": "node ./test/types/check-types-filename && tsd",
-    "setup-test-encryption": "bash scripts/configure-cluster-with-encryption.sh",
+    "setup-test-encryption": "node scripts/setup-encryption-tests.js",
     "test-encryption": "mocha --exit ./test/encryption/*.test.js",
     "tdd": "mocha ./test/*.test.js --inspect --watch --recursive --watch-files ./**/*.{js,ts}",
     "test-coverage": "nyc --reporter=html --reporter=text npm test",
@@ -146,4 +148,4 @@
       "target": "ES2017"
     }
   }
-}
+}

test/encryption/encryption.test.js

@@ -9,8 +9,19 @@ const fs = require('fs');
 const mongoose = require('../../lib');
 const { Map } = require('../../lib/types');
 
+const { stop, start } = require('../../scripts/setup-encryption-tests');
+
 const LOCAL_KEY = Buffer.from('Mng0NCt4ZHVUYUJCa1kxNkVyNUR1QURhZ2h2UzR2d2RrZzh0cFBwM3R6NmdWMDFBMUN3YkQ5aXRRMkhGRGdQV09wOGVNYUMxT2k3NjZKelhaQmRCZGJkTXVyZG9uSjFk', 'base64');
 
+const exists = path => {
+  try {
+    fs.statSync(path);
+    return true;
+  } catch {
+    return false;
+  }
+};
+
 /**
  * @param {object} object
  * @param {string} property
@@ -25,19 +36,29 @@ describe('encryption integration tests', () => {
   const cachedUri = process.env.MONGOOSE_TEST_URI;
   const cachedLib = process.env.CRYPT_SHARED_LIB_PATH;
 
-  before(function() {
+  before(async function() {
+    this.timeout(0);
     const cwd = process.cwd();
-    const file = fs.readFileSync(cwd + '/data/mo-expansion.yml', { encoding: 'utf-8' }).trim().split('\n');
-
-    // matches `key="value"` and extracts key and value.
-    const regex = /^(?<key>.*): "(?<value>.*)"$/;
-    const variables = Object.fromEntries(file.map((line) => regex.exec(line.trim()).groups).map(({ key, value }) => [key, value]));
-
-    process.env.CRYPT_SHARED_LIB_PATH ??= variables.CRYPT_SHARED_LIB_PATH;
-    process.env.MONGOOSE_TEST_URI ??= variables.MONGODB_URI;
+    if (!exists(cwd + '/mo-expansion.yml')) {
+      const { uri, cryptShared } = await start();
+      process.env.CRYPT_SHARED_LIB_PATH = cryptShared;
+      process.env.MONGOOSE_TEST_URI = uri;
+    } else {
+      const file = fs.readFileSync(cwd + '/mo-expansion.yml', { encoding: 'utf-8' }).trim().split('\n');
+
+      const regex = /^(?<key>.*): "(?<value>.*)"$/;
+      const variables = Object.fromEntries(file.map((line) => regex.exec(line.trim()).groups).map(({ key, value }) => [key, value]));
+
+      process.env.CRYPT_SHARED_LIB_PATH = variables.CRYPT_SHARED_LIB_PATH;
+      process.env.MONGOOSE_TEST_URI = variables.MONGODB_URI;
+    }
   });
 
-  after(function() {
+  after(async function() {
+    if (!exists(process.cwd() + '/mo-expansion.yml')) {
+      await stop();
+    }
+
     process.env.CRYPT_SHARED_LIB_PATH = cachedLib;
     process.env.MONGOOSE_TEST_URI = cachedUri;
   });