Add tests

DataDog · Aug 8, 2024 · fb80364 · fb80364
1 parent 7e0fb93
commit fb80364
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/packages/dd-trace/test/appsec/iast/taint-tracking/resources/propagationFunctions.js b/packages/dd-trace/test/appsec/iast/taint-tracking/resources/propagationFunctions.js
@@ -6,6 +6,12 @@ function insertStr (str) {
   return `pre_${str}_suf`
 }
 
+function templateLiteralEndingWithNumberParams (str) {
+  const num1 = 1
+  const num2 = 2
+  return `${str}Literal${num1}${num2}`
+}
+
 function appendStr (str) {
   let pre = 'pre_'
   pre += str
@@ -101,6 +107,7 @@ module.exports = {
   sliceStr,
   substrStr,
   substringStr,
+  templateLiteralEndingWithNumberParams,
   toLowerCaseStr,
   toUpperCaseStr,
   trimEndStr,

diff --git a/packages/dd-trace/test/appsec/iast/taint-tracking/taint-tracking-impl.spec.js b/packages/dd-trace/test/appsec/iast/taint-tracking/taint-tracking-impl.spec.js
@@ -25,6 +25,7 @@ const propagationFns = [
   'sliceStr',
   'substrStr',
   'substringStr',
+  'templateLiteralEndingWithNumberParams',
   'toLowerCaseStr',
   'toUpperCaseStr',
   'trimEndStr',
@@ -135,7 +136,8 @@ describe('TaintTracking', () => {
       'arrayProtoJoin',
       'concatSuffix',
       'concatTaintedStr',
-      'insertStr'
+      'insertStr',
+      'templateLiteralEndingWithNumberParams'
     ]
     propagationFns.forEach((propFn) => {
       if (filtered.includes(propFn)) return