fix names

DataDog · Dec 24, 2024 · 0a9cc8d · 0a9cc8d
1 parent 3cc25af
commit 0a9cc8d
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 10 deletions.
diff --git a/packages/dd-trace/src/appsec/iast/iast-context.js b/packages/dd-trace/src/appsec/iast/iast-context.js
@@ -12,12 +12,12 @@ function getIastContext (store, topContext) {
 function getIastStackTraceId (iastContext) {
   if (!iastContext) return 0
 
-  if (!iastContext.iastStackTraceId) {
-    iastContext.iastStackTraceId = 0
+  if (!iastContext.stackTraceId) {
+    iastContext.stackTraceId = 0
   }
 
-  iastContext.iastStackTraceId += 1
-  return iastContext.iastStackTraceId
+  iastContext.stackTraceId += 1
+  return iastContext.stackTraceId
 }
 
 /* TODO Fix storage problem when the close event is called without

diff --git a/packages/dd-trace/src/appsec/iast/vulnerability-reporter.js b/packages/dd-trace/src/appsec/iast/vulnerability-reporter.js
@@ -103,7 +103,7 @@ function stopClearCacheTimer () {
 }
 
 function isDuplicatedVulnerability (vulnerability) {
-  return VULNERABILITY_HASHES.has(`${vulnerability.type}${vulnerability.hash}`)
+  return VULNERABILITY_HASHES.get(`${vulnerability.type}${vulnerability.hash}`)
 }
 
 function getVulnerabilityCallSiteList () {

diff --git a/packages/dd-trace/src/appsec/rasp/utils.js b/packages/dd-trace/src/appsec/rasp/utils.js
@@ -33,9 +33,9 @@ function handleResult (actions, req, res, abortController, config) {
 
   const { enabled, maxDepth, maxStackTraces } = config.appsec.stackTrace
 
-  const callSiteList = getCallSiteList(maxDepth)
-
   if (generateStackTraceAction && enabled) {
+    const callSiteList = getCallSiteList(maxDepth)
+
     const rootSpan = web.root(req)
     reportStackTrace(
       rootSpan,

diff --git a/packages/dd-trace/test/appsec/iast/analyzers/vulnerability-analyzer.spec.js b/packages/dd-trace/test/appsec/iast/analyzers/vulnerability-analyzer.spec.js
@@ -117,7 +117,7 @@ describe('vulnerability-analyzer', () => {
       }
     }
     vulnerabilityAnalyzer._report(VULNERABLE_VALUE, context)
-    expect(vulnerabilityReporter.addVulnerability).to.have.been.calledOnceWith(
+    expect(vulnerabilityReporter.addVulnerability).to.have.been.calledOnceWithExactly(
       context,
       {
         type: 'TEST_ANALYZER',

diff --git a/packages/dd-trace/test/appsec/rasp/utils.spec.js b/packages/dd-trace/test/appsec/rasp/utils.spec.js
@@ -12,7 +12,7 @@ describe('RASP - utils.js', () => {
 
     stackTrace = {
       reportStackTrace: sinon.stub(),
-      getCallSiteList: sinon.stub().returns({})
+      getCallSiteList: sinon.stub().returns([])
     }
 
     utils = proxyquire('../../../src/appsec/rasp/utils', {
@@ -45,7 +45,7 @@ describe('RASP - utils.js', () => {
       web.root.returns(rootSpan)
 
       utils.handleResult(result, req, undefined, undefined, config)
-      sinon.assert.calledOnceWithExactly(stackTrace.reportStackTrace, rootSpan, stackId, 42, 2, {})
+      sinon.assert.calledOnceWithExactly(stackTrace.reportStackTrace, rootSpan, stackId, 42, 2, [])
     })
 
     it('should not report stack trace when no action is present in waf result', () => {