fix test

DataDog · Jun 26, 2024 · 7a8c016 · 7a8c016
1 parent 26c7ee4
commit 7a8c016
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/internal-api/src/test/groovy/datadog/trace/api/sampling/SamplingMechanismTest.groovy b/internal-api/src/test/groovy/datadog/trace/api/sampling/SamplingMechanismTest.groovy
@@ -1,10 +1,10 @@
 package datadog.trace.api.sampling
 
-import spock.lang.Specification
+import datadog.trace.test.util.DDSpecification
 import static datadog.trace.api.sampling.PrioritySampling.*
 import static datadog.trace.api.sampling.SamplingMechanism.*
 
-class SamplingMechanismTest extends Specification {
+class SamplingMechanismTest extends DDSpecification {
 
   static userDropX = USER_DROP - 1
   static userKeepX = USER_KEEP + 1
@@ -98,7 +98,7 @@ class SamplingMechanismTest extends Specification {
 
   void 'Test canAvoidSamplingPriorityLock'(){
     setup:
-    System.setProperty("dd.experimental.appsec.standalone.enabled", "true")
+    injectSysConfig("dd.experimental.appsec.standalone.enabled", "true")
 
     expect:
     canAvoidSamplingPriorityLock(priority, mechanism) == valid