From 822d3ac0157bc27e65ea5e6c953025e25d75931d Mon Sep 17 00:00:00 2001 From: Igor Unanua Date: Fri, 25 Oct 2024 15:37:42 +0200 Subject: [PATCH] PrioritySampler.keepTrace test --- .../dd-trace/test/priority_sampler.spec.js | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/packages/dd-trace/test/priority_sampler.spec.js b/packages/dd-trace/test/priority_sampler.spec.js index 852959551bf..88c134a5758 100644 --- a/packages/dd-trace/test/priority_sampler.spec.js +++ b/packages/dd-trace/test/priority_sampler.spec.js @@ -491,4 +491,22 @@ describe('PrioritySampler', () => { expect(context._trace.tags[DECISION_MAKER_KEY]).to.equal('-0') }) }) + + describe('keepTrace', () => { + it('should not fail if no _prioritySampler', () => { + expect(() => { + PrioritySampler.keepTrace(span, SAMPLING_MECHANISM_APPSEC) + }).to.not.throw() + }) + + it('should call setPriority with span USER_KEEP and mechanism', () => { + const setPriority = sinon.stub(prioritySampler, 'setPriority') + + span._prioritySampler = prioritySampler + + PrioritySampler.keepTrace(span, SAMPLING_MECHANISM_APPSEC) + + expect(setPriority).to.be.calledOnceWithExactly(span, USER_KEEP, SAMPLING_MECHANISM_APPSEC) + }) + }) })