From ba6ff5a466f60397202a843360816c36c5436f91 Mon Sep 17 00:00:00 2001
From: Ran Ribenzaft <ranrib@users.noreply.github.com>
Date: Tue, 26 Jan 2021 13:50:59 +0200
Subject: [PATCH] fix(tracer.js): propagate tracer to send function (#413)

---
 src/tracer.js | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/tracer.js b/src/tracer.js
index c7253c11..500bcbbe 100644
--- a/src/tracer.js
+++ b/src/tracer.js
@@ -323,13 +323,15 @@ function addLabelsToTrace() {
  * handled
  * @param {function} traceSender: The function to use to send the trace. Gets the trace object
  *     as a parameter and sends a JSON version of it to epsagon's infrastructure
+ * @param {object} tracerObject Optional tracer object to use for sending.
  * @return {*} traceSender's result
  */
-function sendCurrentTrace(traceSender) {
-    const tracerObj = module.exports.getTrace();
+function sendCurrentTrace(traceSender, tracerObject) {
+    const tracerObj = tracerObject || module.exports.getTrace();
 
     const { sendOnlyErrors, ignoredKeys, removeIgnoredKeys } = config.getConfig();
     if (!tracerObj) {
+        utils.debugLog('Trace object not found for sending');
         return Promise.resolve();
     }
     addLabelsToTrace();
@@ -590,9 +592,9 @@ module.exports.sendTrace = function sendTrace(runnerUpdateFunc, tracerObject) {
         // Setting runner's duration.
         runnerUpdateFunc();
         if (config.getConfig().sendBatch) {
-            return sendCurrentTrace(traceObject => traceQueue.push(traceObject));
+            return sendCurrentTrace(traceObject => traceQueue.push(traceObject), tracerObj);
         }
-        return sendCurrentTrace(traceObject => module.exports.postTrace(traceObject));
+        return sendCurrentTrace(traceObject => module.exports.postTrace(traceObject), tracerObj);
     });
 };