add trace to test mock

carbon3d · Nov 19, 2019 · f27b8d5 · f27b8d5
1 parent 568f46a
commit f27b8d5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/instrumentation/express.spec.js b/src/instrumentation/express.spec.js
@@ -16,7 +16,8 @@ describe('instrumentation: express', () => {
     mockSpan = {
       setTag: this.sandbox.spy(),
       log: this.sandbox.spy(),
-      finish: this.sandbox.spy()
+      finish: this.sandbox.spy(),
+      context: () => {return {traceIdStr: 'cd085dc813af7cb1'}}
     }
 
     this.sandbox.stub(cls, 'startRootSpan').callsFake(() => mockSpan)

diff --git a/src/instrumentation/expressError.spec.js b/src/instrumentation/expressError.spec.js
@@ -15,12 +15,14 @@ describe('instrumentation: expressError', () => {
   beforeEach(function () {
     tracer = new Tracer()
     mockRootSpan = {
-      setTag: this.sandbox.spy()
+      setTag: this.sandbox.spy(),
+      context: () => {return {traceIdStr: 'cd085dc813af7cb1'}}
     }
     mockChildSpan = {
       setTag: this.sandbox.spy(),
       log: this.sandbox.spy(),
-      finish: this.sandbox.spy()
+      finish: this.sandbox.spy(),
+      context: () => {return {traceIdStr: 'cd085dc813af7cb1'}}
     }
 
     this.sandbox.stub(cls, 'getRootSpan').callsFake(() => mockRootSpan)