Fix test

DataDog · Jun 4, 2024 · 9180a62 · 9180a62
1 parent d1ea37d
commit 9180a62
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/dd-java-agent/agent-iast/src/test/groovy/com/datadog/iast/ReporterTest.groovy b/dd-java-agent/agent-iast/src/test/groovy/com/datadog/iast/ReporterTest.groovy
@@ -80,6 +80,7 @@ class ReporterTest extends DDSpecification {
       ]
     }''', batch.toString(), true)
     1 * traceSegment.setTagTop('manual.keep', true)
+    1 * traceSegment.setTagTop('p.appsec', true)
     0 * _
   }
 
@@ -146,6 +147,7 @@ class ReporterTest extends DDSpecification {
       ]
     }''', batch.toString(), true)
     1 * traceSegment.setTagTop('manual.keep', true)
+    1 * traceSegment.setTagTop('p.appsec', true)
     0 * _
   }
 
@@ -274,6 +276,7 @@ class ReporterTest extends DDSpecification {
     1 * traceSegment.getDataTop('iast') >> null
     1 * traceSegment.setDataTop('iast', _ as VulnerabilityBatch)
     1 * traceSegment.setTagTop('manual.keep', true)
+    1 * traceSegment.setTagTop('p.appsec', true)
     1 * traceSegment.setTagTop('_dd.iast.enabled', 1)
     0 * _
   }