@reproapp/node-sdk 0.0.30 → 0.0.32

package/dist/index.d.ts

@@ -262,6 +262,31 @@ export type ReproTracingInitOptions = TracerInitOpts & {
 export declare function initReproTracing(opts?: ReproTracingInitOptions): TracerApi | null;
 /** Optional helper if users want to check it. */
 export declare function isReproTracingEnabled(): boolean;
+type Ctx = {
+    sid?: string;
+    aid?: string;
+    rid?: string;
+    clockSkewMs?: number;
+    excludedSpanIds?: Set<string>;
+    maskReq?: MaskRequestContext;
+    privacyTasks?: Set<Promise<void>>;
+    captureScope?: RequestCaptureScopeState;
+    nativeMongoSuppressions?: NativeMongoSuppression[];
+};
+type NativeMongoSuppression = {
+    collection?: string | null;
+    ops?: Set<string>;
+    startedAt: number;
+    ownerStore?: Ctx;
+};
+type RequestCaptureScopeState = {
+    requestRid: string;
+    transport: 'http' | 'kafka' | 'amqp' | 'background';
+    backendRequestCount: number;
+    traceBatchCount: number;
+    dbChangeCount: number;
+    emailEventCount: number;
+};
 declare function sanitizeTraceValueForPrivacy(value: any): any;
 declare function sanitizeMaterializedTraceValue(value: any): any;
 export declare function __materializePendingTraceEventsForWorker(payload: TraceMaterializationWorkerPayload): Promise<PendingTraceEventRecord[]>;
@@ -382,6 +407,8 @@ export declare const __reproTestHooks: {
     }): void;
     drainSdkBackgroundQueueForTest(): Promise<void>;
     resetSdkBackgroundQueuesForTest(): void;
+    runWithRequestContextForTest<T>(ctx: Partial<Ctx>, fn: () => T): T;
+    getNativeMongoSuppressionCountForTest(): number;
     getRuntimePrivacyPolicyForTest(cfg: ReproMiddlewareConfig): NormalizedRuntimePrivacyPolicy | null;
     setRuntimePrivacyPolicyForTest(cfg: ReproMiddlewareConfig, policy: NormalizedRuntimePrivacyPolicy | null): void;
     recordKafkaTraceEventAsyncForTest: typeof recordKafkaTraceEventAsync;

package/dist/index.js

@@ -6,6 +6,7 @@ const async_hooks_1 = require("async_hooks");
 const crypto_1 = require("crypto");
 const module_1 = require("module");
 const path_1 = require("path");
+const url_1 = require("url");
 const client_1 = require("./ingest/client");
 const privacy_1 = require("./privacy");
 /**
@@ -834,6 +835,7 @@ function pushNativeMongoSuppression(collection, ops) {
         collection: collection == null ? null : String(collection),
         ops: normalizedOps.length ? new Set(normalizedOps) : undefined,
         startedAt: Date.now(),
+        ownerStore: store,
     };
     if (!Array.isArray(store.nativeMongoSuppressions)) {
         store.nativeMongoSuppressions = [];
@@ -844,13 +846,27 @@ function pushNativeMongoSuppression(collection, ops) {
 function popNativeMongoSuppression(token) {
     if (!token)
         return;
-    const store = als.getStore();
+    const store = token.ownerStore ?? als.getStore();
     const suppressions = store?.nativeMongoSuppressions;
-    if (!Array.isArray(suppressions) || suppressions.length === 0)
-        return;
-    const index = suppressions.indexOf(token);
-    if (index >= 0) {
-        suppressions.splice(index, 1);
+    if (Array.isArray(suppressions) && suppressions.length > 0) {
+        const index = suppressions.indexOf(token);
+        if (index >= 0) {
+            suppressions.splice(index, 1);
+        }
+    }
+    token.ownerStore = undefined;
+}
+function releaseNativeMongoSuppression(target) {
+    const token = target?.__repro_native_mongo_suppression;
+    popNativeMongoSuppression(token);
+    try {
+        delete target.__repro_native_mongo_suppression;
+    }
+    catch {
+        try {
+            target.__repro_native_mongo_suppression = undefined;
+        }
+        catch { }
     }
 }
 function isNativeMongoSuppressed(collection, op) {
@@ -860,9 +876,12 @@ function isNativeMongoSuppressed(collection, op) {
     const normalizedCollection = collection == null ? null : String(collection);
     const normalizedOp = normalizeNativeMongoSuppressionOp(op);
     const now = Date.now();
-    for (const token of suppressions) {
-        if (!token || now - token.startedAt > 60000)
+    for (let i = suppressions.length - 1; i >= 0; i -= 1) {
+        const token = suppressions[i];
+        if (!token || now - token.startedAt > 60000) {
+            suppressions.splice(i, 1);
             continue;
+        }
         if (token.collection && normalizedCollection && token.collection !== normalizedCollection)
             continue;
         if (token.ops && token.ops.size > 0 && !token.ops.has(normalizedOp))
@@ -4633,40 +4652,44 @@ function reproMongoosePlugin(cfg) {
             next();
         });
         schema.post('save', { document: true }, function () {
-            const { sid, aid } = getCtx();
-            if (!sid || !aid)
-                return;
-            if (this.$isSubdocument)
-                return;
-            const meta = this.__repro_meta || {};
-            const before = null;
-            const after = this;
-            const collection = meta.collection || resolveCollectionOrWarn(this, 'doc');
-            const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
-            if (!shouldCaptureDbSpan(spanContext))
-                return;
-            const modifiedPaths = Array.isArray(meta.modifiedPaths) ? meta.modifiedPaths : [];
-            const query = meta.wasNew
-                ? { op: 'insertOne', doc: after }
-                : {
-                    filter: { _id: this._id },
-                    update: buildModifiedPathUpdate(this, modifiedPaths),
-                    options: { upsert: false },
-                };
-            emitDbQuery(cfg, sid, aid ?? undefined, {
-                collection,
-                pk: { _id: this._id },
-                before,
-                after,
-                op: meta.wasNew ? 'insert' : 'update',
-                query,
-                t: alignTimestamp(meta.t0 ?? Date.now()),
-                spanContext,
-            });
-            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            try {
+                const { sid, aid } = getCtx();
+                if (!sid || !aid)
+                    return;
+                if (this.$isSubdocument)
+                    return;
+                const meta = this.__repro_meta || {};
+                const before = null;
+                const after = this;
+                const collection = meta.collection || resolveCollectionOrWarn(this, 'doc');
+                const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
+                if (!shouldCaptureDbSpan(spanContext))
+                    return;
+                const modifiedPaths = Array.isArray(meta.modifiedPaths) ? meta.modifiedPaths : [];
+                const query = meta.wasNew
+                    ? { op: 'insertOne', doc: after }
+                    : {
+                        filter: { _id: this._id },
+                        update: buildModifiedPathUpdate(this, modifiedPaths),
+                        options: { upsert: false },
+                    };
+                emitDbQuery(cfg, sid, aid ?? undefined, {
+                    collection,
+                    pk: { _id: this._id },
+                    before,
+                    after,
+                    op: meta.wasNew ? 'insert' : 'update',
+                    query,
+                    t: alignTimestamp(meta.t0 ?? Date.now()),
+                    spanContext,
+                });
+            }
+            finally {
+                releaseNativeMongoSuppression(this);
+            }
         });
         schema.post('save', { document: true }, function (error, _doc, next) {
-            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            releaseNativeMongoSuppression(this);
             next(error);
         });
         // -------- findOneAndUpdate capture (unchanged) --------
@@ -4688,34 +4711,38 @@ function reproMongoosePlugin(cfg) {
             next();
         });
         schema.post('findOneAndUpdate', function (res) {
-            const { sid, aid } = getCtx();
-            if (!sid || !aid)
-                return;
-            const before = null;
-            const after = res ?? null;
-            const collection = this.__repro_collection || resolveCollectionOrWarn(this, 'query');
-            const spanContext = this.__repro_spanContext || captureSpanContextFromTracer(this);
-            if (!shouldCaptureDbSpan(spanContext))
-                return;
-            const filter = this.__repro_filter;
-            const pk = after?._id ?? (filter && typeof filter === 'object' ? filter._id : undefined);
-            emitDbQuery(cfg, sid, aid ?? undefined, {
-                collection,
-                pk: { _id: pk },
-                before,
-                after,
-                op: 'update',
-                queryRaw: {
-                    filter,
-                    update: this.__repro_update,
-                },
-                t: alignTimestamp(this.__repro_t0 ?? Date.now()),
-                spanContext,
-            });
-            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            try {
+                const { sid, aid } = getCtx();
+                if (!sid || !aid)
+                    return;
+                const before = null;
+                const after = res ?? null;
+                const collection = this.__repro_collection || resolveCollectionOrWarn(this, 'query');
+                const spanContext = this.__repro_spanContext || captureSpanContextFromTracer(this);
+                if (!shouldCaptureDbSpan(spanContext))
+                    return;
+                const filter = this.__repro_filter;
+                const pk = after?._id ?? (filter && typeof filter === 'object' ? filter._id : undefined);
+                emitDbQuery(cfg, sid, aid ?? undefined, {
+                    collection,
+                    pk: { _id: pk },
+                    before,
+                    after,
+                    op: 'update',
+                    queryRaw: {
+                        filter,
+                        update: this.__repro_update,
+                    },
+                    t: alignTimestamp(this.__repro_t0 ?? Date.now()),
+                    spanContext,
+                });
+            }
+            finally {
+                releaseNativeMongoSuppression(this);
+            }
         });
         schema.post('findOneAndUpdate', function (error, _res, next) {
-            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            releaseNativeMongoSuppression(this);
             next(error);
         });
         // -------- deleteOne capture (unchanged) --------
@@ -4735,29 +4762,33 @@ function reproMongoosePlugin(cfg) {
             next();
         });
         schema.post('deleteOne', { document: false, query: true }, function () {
-            const { sid, aid } = getCtx();
-            if (!sid || !aid)
-                return;
-            const before = null;
-            const collection = this.__repro_collection || resolveCollectionOrWarn(this, 'query');
-            const filter = this.__repro_filter ?? undefined;
-            const spanContext = this.__repro_spanContext || captureSpanContextFromTracer(this);
-            if (!shouldCaptureDbSpan(spanContext))
-                return;
-            emitDbQuery(cfg, sid, aid ?? undefined, {
-                collection,
-                pk: filter && typeof filter === 'object' && '_id' in filter ? { _id: filter._id } : undefined,
-                before,
-                after: null,
-                op: 'delete',
-                queryRaw: { filter },
-                t: alignTimestamp(this.__repro_t0 ?? Date.now()),
-                spanContext,
-            });
-            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            try {
+                const { sid, aid } = getCtx();
+                if (!sid || !aid)
+                    return;
+                const before = null;
+                const collection = this.__repro_collection || resolveCollectionOrWarn(this, 'query');
+                const filter = this.__repro_filter ?? undefined;
+                const spanContext = this.__repro_spanContext || captureSpanContextFromTracer(this);
+                if (!shouldCaptureDbSpan(spanContext))
+                    return;
+                emitDbQuery(cfg, sid, aid ?? undefined, {
+                    collection,
+                    pk: filter && typeof filter === 'object' && '_id' in filter ? { _id: filter._id } : undefined,
+                    before,
+                    after: null,
+                    op: 'delete',
+                    queryRaw: { filter },
+                    t: alignTimestamp(this.__repro_t0 ?? Date.now()),
+                    spanContext,
+                });
+            }
+            finally {
+                releaseNativeMongoSuppression(this);
+            }
         });
         schema.post('deleteOne', { document: false, query: true }, function (error, _res, next) {
-            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            releaseNativeMongoSuppression(this);
             next(error);
         });
         // -------- NON-intrusive generic query telemetry via schema hooks -------
@@ -4813,29 +4844,33 @@ function reproMongoosePlugin(cfg) {
                 next();
             });
             schema.post(op, function (res) {
-                const { sid, aid } = getCtx();
-                if (!sid)
-                    return;
-                const meta = this.__repro_qmeta || { t0: Date.now(), collection: 'unknown', op };
-                const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
-                emitDbQuery(cfg, sid, aid, {
-                    collection: meta.collection,
-                    op,
-                    queryRaw: {
-                        filter: meta.rawFilter,
-                        update: meta.rawUpdate,
-                        projection: meta.rawProjection,
-                        options: meta.rawOptions,
-                    },
-                    resultMetaRaw: res,
-                    durMs: Date.now() - meta.t0,
-                    t: alignedNow(),
-                    spanContext,
-                });
-                popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+                try {
+                    const { sid, aid } = getCtx();
+                    if (!sid)
+                        return;
+                    const meta = this.__repro_qmeta || { t0: Date.now(), collection: 'unknown', op };
+                    const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
+                    emitDbQuery(cfg, sid, aid, {
+                        collection: meta.collection,
+                        op,
+                        queryRaw: {
+                            filter: meta.rawFilter,
+                            update: meta.rawUpdate,
+                            projection: meta.rawProjection,
+                            options: meta.rawOptions,
+                        },
+                        resultMetaRaw: res,
+                        durMs: Date.now() - meta.t0,
+                        t: alignedNow(),
+                        spanContext,
+                    });
+                }
+                finally {
+                    releaseNativeMongoSuppression(this);
+                }
             });
             schema.post(op, function (error, _res, next) {
-                popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+                releaseNativeMongoSuppression(this);
                 next(error);
             });
         }
@@ -4864,24 +4899,28 @@ function reproMongoosePlugin(cfg) {
             next();
         });
         schema.post('insertMany', { document: false, query: false }, function (docs) {
-            const { sid, aid } = getCtx();
-            if (!sid)
-                return;
-            const meta = this.__repro_insert_meta || { t0: Date.now(), collection: 'unknown' };
-            const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
-            emitDbQuery(cfg, sid, aid ?? undefined, {
-                collection: meta.collection,
-                op: 'insertMany',
-                queryRaw: { docs: meta.rawDocs ?? undefined },
-                resultMetaRaw: docs,
-                durMs: Date.now() - meta.t0,
-                t: alignedNow(),
-                spanContext,
-            });
-            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            try {
+                const { sid, aid } = getCtx();
+                if (!sid)
+                    return;
+                const meta = this.__repro_insert_meta || { t0: Date.now(), collection: 'unknown' };
+                const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
+                emitDbQuery(cfg, sid, aid ?? undefined, {
+                    collection: meta.collection,
+                    op: 'insertMany',
+                    queryRaw: { docs: meta.rawDocs ?? undefined },
+                    resultMetaRaw: docs,
+                    durMs: Date.now() - meta.t0,
+                    t: alignedNow(),
+                    spanContext,
+                });
+            }
+            finally {
+                releaseNativeMongoSuppression(this);
+            }
         });
         schema.post('insertMany', { document: false, query: false }, function (error, _res, next) {
-            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            releaseNativeMongoSuppression(this);
             next(error);
         });
         schema.pre('bulkWrite', { document: false, query: false }, function (next, ops) {
@@ -4906,24 +4945,28 @@ function reproMongoosePlugin(cfg) {
             next();
         });
         schema.post('bulkWrite', { document: false, query: false }, function (res) {
-            const { sid, aid } = getCtx();
-            if (!sid)
-                return;
-            const meta = this.__repro_bulk_meta || { t0: Date.now(), collection: 'unknown' };
-            const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
-            emitDbQuery(cfg, sid, aid, {
-                collection: meta.collection,
-                op: 'bulkWrite',
-                queryRaw: { ops: meta.rawOps ?? undefined },
-                resultMetaRaw: res,
-                durMs: Date.now() - meta.t0,
-                t: alignedNow(),
-                spanContext,
-            });
-            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            try {
+                const { sid, aid } = getCtx();
+                if (!sid)
+                    return;
+                const meta = this.__repro_bulk_meta || { t0: Date.now(), collection: 'unknown' };
+                const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
+                emitDbQuery(cfg, sid, aid, {
+                    collection: meta.collection,
+                    op: 'bulkWrite',
+                    queryRaw: { ops: meta.rawOps ?? undefined },
+                    resultMetaRaw: res,
+                    durMs: Date.now() - meta.t0,
+                    t: alignedNow(),
+                    spanContext,
+                });
+            }
+            finally {
+                releaseNativeMongoSuppression(this);
+            }
         });
         schema.post('bulkWrite', { document: false, query: false }, function (error, _res, next) {
-            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            releaseNativeMongoSuppression(this);
             next(error);
         });
         // Aggregate middleware (non-intrusive)
@@ -4953,24 +4996,28 @@ function reproMongoosePlugin(cfg) {
             next();
         });
         schema.post('aggregate', function (res) {
-            const { sid, aid } = getCtx();
-            if (!sid)
-                return;
-            const meta = this.__repro_aggmeta || { t0: Date.now(), collection: 'unknown' };
-            const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
-            emitDbQuery(cfg, sid, aid, {
-                collection: meta.collection,
-                op: 'aggregate',
-                queryRaw: { pipeline: meta.rawPipeline },
-                resultMetaRaw: res,
-                durMs: Date.now() - meta.t0,
-                t: alignedNow(),
-                spanContext,
-            });
-            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            try {
+                const { sid, aid } = getCtx();
+                if (!sid)
+                    return;
+                const meta = this.__repro_aggmeta || { t0: Date.now(), collection: 'unknown' };
+                const spanContext = meta.spanContext || captureSpanContextFromTracer(this);
+                emitDbQuery(cfg, sid, aid, {
+                    collection: meta.collection,
+                    op: 'aggregate',
+                    queryRaw: { pipeline: meta.rawPipeline },
+                    resultMetaRaw: res,
+                    durMs: Date.now() - meta.t0,
+                    t: alignedNow(),
+                    spanContext,
+                });
+            }
+            finally {
+                releaseNativeMongoSuppression(this);
+            }
         });
         schema.post('aggregate', function (error, _res, next) {
-            popNativeMongoSuppression(this.__repro_native_mongo_suppression);
+            releaseNativeMongoSuppression(this);
             next(error);
         });
     };
@@ -7599,6 +7646,8 @@ function patchRedis(cfg) {
 exports.patchRedis = patchRedis;
 const mongoDbPatchConfigs = [];
 let mongoDbRequireHookInstalled = false;
+let mongoDbImportHookInstalled = false;
+let mongoDbImportHookReady = null;
 const MONGODB_PROMISE_OPS = [
     'insertOne',
     'insertMany',
@@ -7618,6 +7667,11 @@ const MONGODB_PROMISE_OPS = [
 ];
 const MONGODB_CURSOR_OPS = ['find', 'aggregate'];
 const MONGODB_CURSOR_TERMINALS = ['toArray', 'next', 'forEach'];
+function registerMongoDbPatchConfig(cfg) {
+    if (!mongoDbPatchConfigs.includes(cfg)) {
+        mongoDbPatchConfigs.push(cfg);
+    }
+}
 function resolveMongoDbRuntimeContext(cfg) {
     const ctx = getCtx();
     const fallback = cfg.resolveContext?.() ?? {};
@@ -7627,6 +7681,18 @@ function resolveMongoDbRuntimeContext(cfg) {
         aid: ctx.aid ?? fallback.aid ?? null,
     };
 }
+function resolveMongoDbCaptureContext() {
+    for (let i = mongoDbPatchConfigs.length - 1; i >= 0; i -= 1) {
+        const cfg = mongoDbPatchConfigs[i];
+        if (!cfg)
+            continue;
+        const { sid, aid } = resolveMongoDbRuntimeContext(cfg);
+        if (sid) {
+            return { cfg, sid, aid };
+        }
+    }
+    return null;
+}
 function resolveMongoCollectionName(collection) {
     const candidates = [
         collection?.collectionName,
@@ -7911,10 +7977,11 @@ function patchMongoCollectionMethod(proto, op, cursorOp, cfg) {
         if (isNativeMongoSuppressed(collection, op)) {
             return original.apply(this, args);
         }
-        const { sid, aid } = resolveMongoDbRuntimeContext(cfg);
-        if (!sid) {
+        const captureContext = resolveMongoDbCaptureContext();
+        if (!captureContext) {
             return original.apply(this, args);
         }
+        const { cfg: activeCfg, sid, aid } = captureContext;
         const spanContext = captureSpanContextFromTracer(this);
         const t0 = Date.now();
         const dbName = resolveMongoDbName(this);
@@ -7925,7 +7992,7 @@ function patchMongoCollectionMethod(proto, op, cursorOp, cfg) {
             if (emitted)
                 return;
             emitted = true;
-            emitDbQuery(cfg, sid, aid ?? undefined, {
+            emitDbQuery(activeCfg, sid, aid ?? undefined, {
                 type: 'mongo',
                 collection,
                 op,
@@ -7943,7 +8010,7 @@ function patchMongoCollectionMethod(proto, op, cursorOp, cfg) {
             if (emitted)
                 return;
             emitted = true;
-            emitDbQuery(cfg, sid, aid ?? undefined, {
+            emitDbQuery(activeCfg, sid, aid ?? undefined, {
                 type: 'mongo',
                 collection,
                 op,
@@ -7968,7 +8035,7 @@ function patchMongoCollectionMethod(proto, op, cursorOp, cfg) {
             const result = original.apply(this, actualArgs);
             if (cursorOp) {
                 return patchMongoCursorInstance(result, {
-                    cfg,
+                    cfg: activeCfg,
                     sid,
                     aid,
                     collection,
@@ -8053,6 +8120,7 @@ function patchMongoClientPrototype(proto, cfg) {
     };
 }
 function patchMongoDbFromModuleExports(mongoModule, cfg) {
+    registerMongoDbPatchConfig(cfg);
     if (!mongoModule)
         return;
     const constructors = new Set();
@@ -8108,9 +8176,46 @@ function installMongoDbRequireHook() {
     }
     catch { }
 }
+function installMongoDbImportHook() {
+    if (mongoDbImportHookInstalled)
+        return mongoDbImportHookReady ?? Promise.resolve();
+    mongoDbImportHookInstalled = true;
+    mongoDbImportHookReady = (async () => {
+        try {
+            const nodeModule = requireOptionalHostModule('node:module') ?? requireOptionalHostModule('module');
+            if (typeof nodeModule?.register !== 'function')
+                return;
+            const iitm = requireOptionalHostModule('import-in-the-middle');
+            const Hook = iitm?.Hook ?? iitm;
+            if (typeof Hook !== 'function')
+                return;
+            let waitForAllMessagesAcknowledged;
+            let registerOptions = { data: { include: ['mongodb'] } };
+            if (typeof iitm?.createAddHookMessageChannel === 'function') {
+                const channel = iitm.createAddHookMessageChannel();
+                registerOptions = channel.registerOptions;
+                waitForAllMessagesAcknowledged = channel.waitForAllMessagesAcknowledged;
+            }
+            nodeModule.register('import-in-the-middle/hook.mjs', (0, url_1.pathToFileURL)(__filename), registerOptions);
+            // eslint-disable-next-line no-new
+            new Hook(['mongodb'], (exports) => {
+                for (const cfg of mongoDbPatchConfigs) {
+                    patchMongoDbFromModuleExports(exports, cfg);
+                }
+                return exports;
+            });
+            if (waitForAllMessagesAcknowledged) {
+                await waitForAllMessagesAcknowledged();
+            }
+        }
+        catch { }
+    })();
+    return mongoDbImportHookReady;
+}
 function patchMongoDb(cfg) {
-    mongoDbPatchConfigs.push(cfg);
+    registerMongoDbPatchConfig(cfg);
     installMongoDbRequireHook();
+    void installMongoDbImportHook();
     const mongoModules = new Set();
     for (const loadedMongoModule of getLoadedModuleExportsByPackage('mongodb')) {
         mongoModules.add(loadedMongoModule);
@@ -8628,6 +8733,7 @@ async function initRepro(cfg) {
     patchAmqplib(integrationCfg);
     patchRedis(integrationCfg);
     patchMongoDb(integrationCfg);
+    await installMongoDbImportHook();
     patchSendgridMail(integrationCfg);
     patchMongooseIntegration(reproAutoMiddlewareConfig);
     await initializeRuntimePrivacyPolicyAtStartup(reproAutoMiddlewareConfig);
@@ -8661,6 +8767,16 @@ exports.__reproTestHooks = {
         lastClientActivityAt = 0;
         sdkBackgroundQuietUntil = 0;
     },
+    runWithRequestContextForTest(ctx, fn) {
+        return als.run({
+            ...ctx,
+            excludedSpanIds: ctx.excludedSpanIds ?? new Set(),
+        }, fn);
+    },
+    getNativeMongoSuppressionCountForTest() {
+        const suppressions = getCtx().nativeMongoSuppressions;
+        return Array.isArray(suppressions) ? suppressions.length : 0;
+    },
     getRuntimePrivacyPolicyForTest(cfg) {
         return getRuntimePrivacyState(cfg).policy;
     },