Fix video info propagation.

prebid · Jul 11, 2024 · 0378dc9 · 0378dc9
1 parent 24ff2b7
commit 0378dc9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/src/main/java/org/prebid/server/bidder/yeahmobi/YeahmobiBidder.java b/src/main/java/org/prebid/server/bidder/yeahmobi/YeahmobiBidder.java
@@ -32,6 +32,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.Objects;
+import java.util.Optional;
 import java.util.function.Function;
 import java.util.stream.Collectors;
 
@@ -182,6 +183,10 @@ private ExtBidPrebid parseBidExt(ObjectNode bidExt) {
     }
 
     private ExtBidPrebidVideo videoInfo(ExtBidPrebid extBidPrebid) {
-        return extBidPrebid != null ? extBidPrebid.getVideo() : null;
+        return Optional.ofNullable(extBidPrebid)
+                .map(ExtBidPrebid::getVideo)
+                .map(ExtBidPrebidVideo::getDuration)
+                .map(duration -> ExtBidPrebidVideo.of(duration, null))
+                .orElse(null);
     }
 }
diff --git a/src/test/java/org/prebid/server/bidder/yeahmobi/YeahmobiBidderTest.java b/src/test/java/org/prebid/server/bidder/yeahmobi/YeahmobiBidderTest.java
@@ -264,7 +264,9 @@ public void makeBidsShouldReturnVideoInfo() throws JsonProcessingException {
                 mapper.writeValueAsString(
                         givenBidResponse(bidBuilder -> bidBuilder
                                 .impid("123")
-                                .ext(mapper.valueToTree(Map.of("video", Map.of("duration", 1)))))));
+                                .ext(mapper.valueToTree(Map.of("video", Map.of(
+                                        "duration", 1,
+                                        "primary_category", "cat")))))));
 
         // when
         final Result<List<BidderBid>> result = target.makeBids(httpCall, null);
@@ -275,7 +277,9 @@ public void makeBidsShouldReturnVideoInfo() throws JsonProcessingException {
                 .containsExactly(BidderBid.builder()
                         .bid(Bid.builder()
                                 .impid("123")
-                                .ext(mapper.valueToTree(Map.of("video", Map.of("duration", 1))))
+                                .ext(mapper.valueToTree(Map.of("video", Map.of(
+                                        "duration", 1,
+                                        "primary_category", "cat"))))
                                 .build())
                         .type(xNative)
                         .bidCurrency("EUR")