|
4 | 4 | import com.iab.openrtb.request.Audio; |
5 | 5 | import com.iab.openrtb.request.Banner; |
6 | 6 | import com.iab.openrtb.request.BidRequest; |
| 7 | +import com.iab.openrtb.request.Device; |
7 | 8 | import com.iab.openrtb.request.Imp; |
8 | 9 | import com.iab.openrtb.request.Native; |
9 | 10 | import com.iab.openrtb.request.Video; |
|
36 | 37 | import static org.prebid.server.util.HttpUtil.ACCEPT_HEADER; |
37 | 38 | import static org.prebid.server.util.HttpUtil.APPLICATION_JSON_CONTENT_TYPE; |
38 | 39 | import static org.prebid.server.util.HttpUtil.CONTENT_TYPE_HEADER; |
| 40 | +import static org.prebid.server.util.HttpUtil.USER_AGENT_HEADER; |
| 41 | +import static org.prebid.server.util.HttpUtil.X_FORWARDED_FOR_HEADER; |
39 | 42 | import static org.springframework.util.MimeTypeUtils.APPLICATION_JSON_VALUE; |
40 | 43 |
|
41 | 44 | public class AdmaticBidderTest extends VertxTest { |
@@ -95,10 +98,34 @@ public void makeHttpRequestsShouldReturnExpectedHeaders() { |
95 | 98 | // then |
96 | 99 | assertThat(result.getValue()).hasSize(1).first() |
97 | 100 | .extracting(HttpRequest::getHeaders) |
98 | | - .satisfies(headers -> assertThat(headers.get(CONTENT_TYPE_HEADER)) |
99 | | - .isEqualTo(APPLICATION_JSON_CONTENT_TYPE)) |
100 | | - .satisfies(headers -> assertThat(headers.get(ACCEPT_HEADER)) |
101 | | - .isEqualTo(APPLICATION_JSON_VALUE)); |
| 101 | + .satisfies(headers -> { |
| 102 | + assertThat(headers.get(CONTENT_TYPE_HEADER)).isEqualTo(APPLICATION_JSON_CONTENT_TYPE); |
| 103 | + assertThat(headers.get(ACCEPT_HEADER)).isEqualTo(APPLICATION_JSON_VALUE); |
| 104 | + }); |
| 105 | + assertThat(result.getErrors()).isEmpty(); |
| 106 | + } |
| 107 | + |
| 108 | + @Test |
| 109 | + public void makeHttpRequestsShouldReturnRequestsWithHeadersIfDeviceIsPresent() { |
| 110 | + // given |
| 111 | + final BidRequest bidRequest = givenBidRequest(identity()) |
| 112 | + .toBuilder() |
| 113 | + .device(Device.builder().ua("ua").ip("ip").ipv6("ipv6").build()) |
| 114 | + .build(); |
| 115 | + |
| 116 | + // when |
| 117 | + final Result<List<HttpRequest<BidRequest>>> result = target.makeHttpRequests(bidRequest); |
| 118 | + |
| 119 | + // then |
| 120 | + assertThat(result.getErrors()).isEmpty(); |
| 121 | + assertThat(result.getValue()).hasSize(1).first() |
| 122 | + .extracting(HttpRequest::getHeaders) |
| 123 | + .satisfies(headers -> { |
| 124 | + assertThat(headers.get(CONTENT_TYPE_HEADER)).isEqualTo(APPLICATION_JSON_CONTENT_TYPE); |
| 125 | + assertThat(headers.get(ACCEPT_HEADER)).isEqualTo(APPLICATION_JSON_VALUE); |
| 126 | + assertThat(headers.get(USER_AGENT_HEADER)).isEqualTo("ua"); |
| 127 | + assertThat(headers.get(X_FORWARDED_FOR_HEADER)).isEqualTo("ipv6"); |
| 128 | + }); |
102 | 129 | assertThat(result.getErrors()).isEmpty(); |
103 | 130 | } |
104 | 131 |
|
|
0 commit comments