Core: Fix PAA imp.ext.igs.ae population

src/main/java/org/prebid/server/auction/ImpAdjuster.java

@@ -1,13 +1,11 @@
 package org.prebid.server.auction;
 
 import com.fasterxml.jackson.databind.JsonNode;
-import com.fasterxml.jackson.databind.node.ArrayNode;
 import com.fasterxml.jackson.databind.node.IntNode;
 import com.fasterxml.jackson.databind.node.ObjectNode;
 import com.iab.openrtb.request.Imp;
 import org.prebid.server.json.JacksonMapper;
 import org.prebid.server.json.JsonMerger;
-import org.prebid.server.util.StreamUtil;
 import org.prebid.server.validation.ImpValidator;
 
 import java.util.Iterator;
@@ -81,16 +79,14 @@ private void setAeParams(ObjectNode ext) {
 
         final boolean extIgsAePresent = Optional.ofNullable(ext)
                 .map(extNode -> extNode.get(EXT_IGS))
-                .filter(JsonNode::isArray)
-                .stream()
-                .flatMap(extNode -> StreamUtil.asStream(extNode.spliterator()))
-                .filter(Objects::nonNull)
-                .anyMatch(igsElementNode -> igsElementNode.has(EXT_AE));
+                .map(igsNode -> igsNode.get(EXT_AE))
+                .isPresent();
 
         if (!extIgsAePresent && (extAe == 0 || extAe == 1)) {
-            final ArrayNode extIgs = jacksonMapper.mapper().createArrayNode();
-            extIgs.add(jacksonMapper.mapper().createObjectNode().set(EXT_AE, IntNode.valueOf(extAe)));
-            ext.set(EXT_IGS, extIgs);
+            final ObjectNode igsNode = jacksonMapper.mapper().createObjectNode()
+                    .set(EXT_AE, IntNode.valueOf(extAe));
+
+            ext.set(EXT_IGS, igsNode);
         }
     }
 

src/test/java/org/prebid/server/auction/ImpAdjusterTest.java

@@ -92,7 +92,7 @@ public void adjustShouldSetImpExtIgsAeWhenImpExtAeIsZero() {
         final Imp result = target.adjust(givenImp, "someBidder", bidderAliases, debugMessages);
 
         // then
-        assertThat(result.getExt().get("igs").get(0).get("ae")).isEqualTo(IntNode.valueOf(0));
+        assertThat(result.getExt().get("igs").get("ae")).isEqualTo(IntNode.valueOf(0));
         assertThat(debugMessages).isEmpty();
     }
 
@@ -110,7 +110,7 @@ public void adjustShouldSetImpExtIgsAeWhenImpExtAeIsOne() {
         final Imp result = target.adjust(givenImp, "someBidder", bidderAliases, debugMessages);
 
         // then
-        assertThat(result.getExt().get("igs").get(0).get("ae")).isEqualTo(IntNode.valueOf(1));
+        assertThat(result.getExt().get("igs").get("ae")).isEqualTo(IntNode.valueOf(1));
         assertThat(debugMessages).isEmpty();
     }
 
@@ -137,7 +137,7 @@ public void adjustShouldNotModifyImpExtIgsAeWhenImpExtIgsAePresent() {
         // given
         final ObjectNode ext = mapper.createObjectNode();
         ext.set("ae", IntNode.valueOf(0));
-        ext.set("igs", mapper.createArrayNode().add(mapper.createObjectNode().set("ae", IntNode.valueOf(123))));
+        ext.set("igs", mapper.createObjectNode().set("ae", IntNode.valueOf(123)));
 
         final Imp givenImp = Imp.builder().ext(ext).build();
 
@@ -147,7 +147,7 @@ public void adjustShouldNotModifyImpExtIgsAeWhenImpExtIgsAePresent() {
         final Imp result = target.adjust(givenImp, "someBidder", bidderAliases, debugMessages);
 
         // then
-        assertThat(result.getExt().get("igs").get(0).get("ae")).isEqualTo(IntNode.valueOf(123));
+        assertThat(result.getExt().get("igs").get("ae")).isEqualTo(IntNode.valueOf(123));
         assertThat(debugMessages).isEmpty();
     }
 

src/test/resources/org/prebid/server/it/openrtb2/ix/test-ix-bid-request.json

@@ -16,11 +16,9 @@
       },
       "ext": {
         "ae": 1,
-        "igs": [
-          {
-            "ae": 1
-          }
-        ],
+        "igs": {
+          "ae": 1
+        },
         "tid": "${json-unit.any-string}",
         "bidder": {
           "siteId": "10002"