@reproapp/node-sdk 0.0.27 → 0.0.28

package/dist/index.js

@@ -3997,6 +3997,7 @@ async function maybeCaptureDbValueAsync(params, sink) {
 }
 function reproMiddleware(cfg) {
     const masking = normalizeMaskingConfig(cfg.masking);
+    patchMongoDb(cfg);
     return async function (req, res, next) {
         const querySid = readRequestQueryParam(req, REPRO_QUERY_SESSION_PARAM);
         const queryAid = readRequestQueryParam(req, REPRO_QUERY_ACTION_PARAM);
@@ -7979,10 +7980,6 @@ function patchMongoDb(cfg) {
     mongoDbPatchConfigs.push(cfg);
     installMongoDbRequireHook();
     const mongoModules = new Set();
-    const resolvedMongoModule = requireOptionalHostModule('mongodb');
-    if (resolvedMongoModule) {
-        mongoModules.add(resolvedMongoModule);
-    }
     for (const loadedMongoModule of getLoadedModuleExportsByPackage('mongodb')) {
         mongoModules.add(loadedMongoModule);
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@reproapp/node-sdk",
-  "version": "0.0.27",
+  "version": "0.0.28",
   "description": "Repro Nest SDK",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/index.ts

@@ -4892,6 +4892,7 @@ export type ReproInitConfig = ReproMiddlewareConfig & {
 
 export function reproMiddleware(cfg: ReproMiddlewareConfig) {
     const masking = normalizeMaskingConfig(cfg.masking);
+    patchMongoDb(cfg);
     return async function (req: Request, res: Response, next: NextFunction) {
         const querySid = readRequestQueryParam(req, REPRO_QUERY_SESSION_PARAM);
         const queryAid = readRequestQueryParam(req, REPRO_QUERY_ACTION_PARAM);
@@ -9225,10 +9226,6 @@ export function patchMongoDb(cfg: MongoDbPatchConfig): void {
     installMongoDbRequireHook();
 
     const mongoModules = new Set<any>();
-    const resolvedMongoModule = requireOptionalHostModule('mongodb');
-    if (resolvedMongoModule) {
-        mongoModules.add(resolvedMongoModule);
-    }
     for (const loadedMongoModule of getLoadedModuleExportsByPackage('mongodb')) {
         mongoModules.add(loadedMongoModule);
     }

package/test/mongodb-integration.test.js

@@ -7,6 +7,16 @@ process.env.TRACE_LINGER_AFTER_FINISH_MS = '0';
 
 const assert = require('assert');
 const { EventEmitter } = require('events');
+const mongodb = require('mongodb');
+
+const originalNativeMongoInsertOne = mongodb.Collection.prototype.insertOne;
+
+mongodb.Collection.prototype.insertOne = async function reproFakeNativeMongoInsertOne(doc, options) {
+    assert.strictEqual(options.session.id, 'native-real-session-1');
+    doc._id = doc._id ?? 'subject-real-1';
+    return { acknowledged: true, insertedId: doc._id };
+};
+
 const { flushIngestQueue } = require('../dist/ingest/client');
 const { reproMiddleware, __reproTestHooks } = require('../dist');
 
@@ -122,6 +132,50 @@ function dbChanges(events) {
         .flatMap((event) => Array.isArray(event.payload?.db) ? event.payload.db : []);
 }
 
+async function testReproMiddlewarePatchesPreloadedNativeMongoClientInsertOne() {
+    await withCapturedFetch(async (capturedBodies) => {
+        __reproTestHooks.resetSdkBackgroundQueuesForTest();
+        const sessionId = 'S_mongo_real_client_insert';
+        const actionId = 'A_mongo_real_client_insert';
+        const cfg = makeConfig(sessionId, actionId);
+
+        const collection = new mongodb.MongoClient('mongodb://127.0.0.1:27017/clinical')
+            .db('clinical')
+            .collection('subjects');
+
+        await runCapturedRequest(cfg, sessionId, actionId, async () => {
+            const result = await collection.insertOne(
+                { subjectNumber: '10013038', status: 'screened' },
+                {
+                    session: {
+                        id: 'native-real-session-1',
+                        hasEnded: false,
+                        inTransaction: () => true,
+                    },
+                },
+            );
+            assert.deepStrictEqual(result, { acknowledged: true, insertedId: 'subject-real-1' });
+        });
+
+        await waitFor(() => dbChanges(flattenEvents(capturedBodies)).length === 1);
+        await drainSdk();
+
+        const [change] = dbChanges(flattenEvents(capturedBodies));
+        assert.strictEqual(change.type, 'mongo');
+        assert.strictEqual(change.collection, 'subjects');
+        assert.strictEqual(change.op, 'insertOne');
+        assert.strictEqual(change.pk._id, 'subject-real-1');
+        assert.strictEqual(change.query.driver, 'mongodb');
+        assert.strictEqual(change.query.db, 'clinical');
+        assert.strictEqual(change.query.doc.subjectNumber, '10013038');
+        assert.strictEqual(change.query.options.session.__kind, 'mongodb_client_session');
+        assert.strictEqual(change.query.options.session.id, 'native-real-session-1');
+        assert.strictEqual(change.after.subjectNumber, '10013038');
+        assert.strictEqual(change.resultMeta.insertedId, 'subject-real-1');
+        assert.strictEqual(change.resultMeta.acknowledged, true);
+    });
+}
+
 async function testNativeInsertOneIsCaptured() {
     await withCapturedFetch(async (capturedBodies) => {
         __reproTestHooks.resetSdkBackgroundQueuesForTest();
@@ -321,6 +375,7 @@ async function testNativeOperationFailureIsCapturedAndRethrown() {
 }
 
 (async () => {
+    await testReproMiddlewarePatchesPreloadedNativeMongoClientInsertOne();
     await testNativeInsertOneIsCaptured();
     await testNativeUpdateOneUpsertIsCaptured();
     await testNativeFindCursorToArrayIsCaptured();
@@ -329,4 +384,6 @@ async function testNativeOperationFailureIsCapturedAndRethrown() {
 })().catch((err) => {
     console.error(err);
     process.exit(1);
+}).finally(() => {
+    mongodb.Collection.prototype.insertOne = originalNativeMongoInsertOne;
 });