Fix test

Author: manuel-alvarez-alvarez

dd-java-agent/appsec/src/main/java/com/datadog/appsec/gateway/GatewayBridge.java

@@ -131,9 +131,24 @@ public GatewayBridge(
       EventProducerService producerService,
       @Nonnull Supplier<ApiSecuritySampler> requestSamplerSupplier,
       List<TraceSegmentPostProcessor> traceSegmentPostProcessors) {
+    this(
+        subscriptionService,
+        producerService,
+        requestSamplerSupplier,
+        null,
+        traceSegmentPostProcessors);
+  }
+
+  GatewayBridge(
+      SubscriptionService subscriptionService,
+      EventProducerService producerService,
+      @Nonnull Supplier<ApiSecuritySampler> requestSamplerSupplier,
+      ApiSecurityDownstreamSampler downstreamSampler,
+      List<TraceSegmentPostProcessor> traceSegmentPostProcessors) {
     this.subscriptionService = subscriptionService;
     this.producerService = producerService;
     this.requestSamplerSupplier = requestSamplerSupplier;
+    this.downstreamSampler = downstreamSampler;
     this.traceSegmentPostProcessors = traceSegmentPostProcessors;
   }
 
@@ -1207,7 +1222,6 @@ private Flow<Void> maybePublishResponseData(AppSecRequestContext ctx) {
 
   private ApiSecurityDownstreamSampler downstreamSampler() {
     if (downstreamSampler == null) {
-      // we don't care about concurrency too much
       downstreamSampler = new ApiSecurityDownstreamSamplerImpl();
     }
     return downstreamSampler;

dd-java-agent/appsec/src/test/groovy/com/datadog/appsec/gateway/GatewayBridgeIGRegistrationSpecification.groovy

@@ -10,7 +10,7 @@ class GatewayBridgeIGRegistrationSpecification extends DDSpecification {
   SubscriptionService ig = Mock()
   EventDispatcher eventDispatcher = Mock()
 
-  GatewayBridge bridge = new GatewayBridge(ig, eventDispatcher, null, [])
+  GatewayBridge bridge = new GatewayBridge(ig, eventDispatcher, null, null, [])
 
   void 'request_body_start and request_body_done are registered'() {
     given:

dd-java-agent/appsec/src/test/groovy/com/datadog/appsec/gateway/GatewayBridgeSpecification.groovy

@@ -92,9 +92,7 @@ class GatewayBridgeSpecification extends DDSpecification {
   TraceSegmentPostProcessor pp = Mock()
   ApiSecuritySamplerImpl requestSampler = Mock(ApiSecuritySamplerImpl)
   ApiSecurityDownstreamSampler downstreamSampler = Mock(ApiSecurityDownstreamSampler)
-  GatewayBridge bridge = new GatewayBridge(ig, eventDispatcher, () -> requestSampler, [pp]).tap {
-    it.downstreamSampler = downstreamSampler
-  }
+  GatewayBridge bridge = new GatewayBridge(ig, eventDispatcher, () -> requestSampler, downstreamSampler, [pp])
 
   Supplier<Flow<AppSecRequestContext>> requestStartedCB
   BiFunction<RequestContext, AgentSpan, Flow<Void>> requestEndedCB