Refactor 2 assertions in one in ExchangeServiceTest

prebid · Oct 25, 2021 · 8641766 · 8641766
1 parent 862b1e8
commit 8641766
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/test/java/org/prebid/server/auction/ExchangeServiceTest.java b/src/test/java/org/prebid/server/auction/ExchangeServiceTest.java
@@ -666,8 +666,8 @@ public void shouldPassRequestWithExtPrebidToDefinedBidder() {
         final ExtRequestPrebid prebid1 = capturedBidRequest1.getBidRequest().getExt().getPrebid();
         assertThat(prebid1).isNotNull();
         final JsonNode bidders1 = prebid1.getBidders();
-        assertThat(bidders1).isNotNull();
-        assertThat(bidders1.fields()).toIterable()
+        assertThat(bidders1).isNotNull()
+                .extracting(JsonNode::fields).asList()
                 .containsOnly(entry("bidder", mapper.createObjectNode().put("test1", "test1")));
 
         final ArgumentCaptor<BidderRequest> bidRequest2Captor = ArgumentCaptor.forClass(BidderRequest.class);
@@ -676,8 +676,8 @@ public void shouldPassRequestWithExtPrebidToDefinedBidder() {
         final ExtRequestPrebid prebid2 = capturedBidRequest2.getExt().getPrebid();
         assertThat(prebid2).isNotNull();
         final JsonNode bidders2 = prebid2.getBidders();
-        assertThat(bidders2).isNotNull();
-        assertThat(bidders2.fields()).toIterable()
+        assertThat(bidders2).isNotNull()
+                .extracting(JsonNode::fields).asList()
                 .containsOnly(entry("bidder", mapper.createObjectNode().put("test2", "test2")));
     }