Updating nested filter to be consistent with other filters

Financial-Times · Jun 28, 2023 · a6f79e8 · a6f79e8
1 parent 9853acd
commit a6f79e8
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/...java/com/ft/up/apipolicy/filters/RemoveJsonNestedPropertiesUnlessPolicyPresentFilter.java b/...java/com/ft/up/apipolicy/filters/RemoveJsonNestedPropertiesUnlessPolicyPresentFilter.java
@@ -4,22 +4,23 @@
 import com.ft.up.apipolicy.configuration.Policy;
 import com.ft.up.apipolicy.pipeline.MutableRequest;
 import java.util.Map;
+import java.util.Set;
 
 public class RemoveJsonNestedPropertiesUnlessPolicyPresentFilter
     extends SuppressJsonPropertiesFilter {
 
-  private final Policy policy;
+  private final Set<Policy> policies;
 
   public RemoveJsonNestedPropertiesUnlessPolicyPresentFilter(
-      final JsonConverter jsonConverter, final Policy policy, final String... jsonProperties) {
+      final JsonConverter jsonConverter, Set<Policy> policies, final String... jsonProperties) {
     super(jsonConverter, jsonProperties);
-    this.policy = policy;
+    this.policies = policies;
   }
 
   @Override
   protected boolean shouldPropertyFilteredOut(
       final String jsonProperty, final MutableRequest request, Map content) {
-    return !request.policyIs(policy);
+    return policies.stream().noneMatch(request::policyIs);
   }
 
   @Override