コード例 #1
0
  public static BidRequest.Builder newBidRequest(String id, Object... adGroupMincpm) {
    BidRequest.Builder request = BidRequest.newBuilder().setId(id);

    for (int i = 0; i < adGroupMincpm.length; i += 3) {
      String adid = String.valueOf(adGroupMincpm[i + 0]);
      request.addImp(
          Impression.newBuilder()
              .setId(adid)
              .setBidfloor(((Number) adGroupMincpm[i + 2]).floatValue())
              .setBanner(Banner.newBuilder().setId(adid).setW(728).setH(90)));
    }

    return request;
  }
  @Test
  public void testRequest() throws IOException {
    // -1=NO_SLOT, 0=no size, 1, 2=multisize/1 MAD, 3=2 MAD/1 deal, 4..8=3 MAD/2 deals
    for (int size = -1; size <= 8; ++size) {
      for (int flags = 0; flags < 0b10000; ++flags) {
        boolean coppa = (flags & 0b1) != 0;
        boolean mobile = (flags & 0b1) != 0;
        boolean impVideo = (flags & 0b10) != 0;
        boolean impNativ = !impVideo && (flags & 0b100) != 0;
        boolean impBanner = !impVideo && !impNativ;
        boolean multiBid = (flags & 0b1000) != 0;
        boolean linkExt = (flags & 0b1000) != 0;
        String testDesc =
            String.format(
                "imp=%s, mobile=%s, coppa=%s, link=%s, size=%s%s",
                (impNativ ? "native" : impVideo ? "video" : "banner"),
                mobile,
                coppa,
                linkExt,
                size,
                multiBid ? "/full" : "");
        ImmutableList.Builder<ExtMapper> extMappers = ImmutableList.builder();
        if (linkExt) {
          extMappers.add(DoubleClickLinkMapper.INSTANCE);
        }
        DoubleClickOpenRtbMapper mapper =
            new DoubleClickOpenRtbMapper(
                new MetricRegistry(),
                TestUtil.getMetadata(),
                OpenRtbJsonFactory.create(),
                extMappers.build());

        NetworkBid.BidRequest.Builder dcRequest = TestData.newRequest(size, coppa, impNativ);
        if (mobile) {
          dcRequest.setMobile(TestData.newMobile(size, coppa));
        }
        if (impVideo) {
          dcRequest.setVideo(TestData.newVideo(size));
        }
        BidRequest request = mapper.toOpenRtbBidRequest(dcRequest.build()).build();

        assertWithMessage(testDesc).that(request.getRegs().hasCoppa()).isEqualTo(coppa);
        assertWithMessage(testDesc).that(request.getRegs().getCoppa()).isEqualTo(coppa);
        assertWithMessage(testDesc)
            .that(request.getUser().getCustomdata())
            .isEqualTo(coppa ? "" : "7CLmnMiwSsq7bNTaiPsztg");

        if (request.getImpCount() == 0) {
          BidResponse response = TestUtil.newBidResponse();
          NetworkBid.BidResponse dcResponse =
              mapper.toExchangeBidResponse(request, response).build();
          assertWithMessage(testDesc).that(dcResponse.getAdCount()).isEqualTo(0);
        } else {
          Imp imp = request.getImp(0);
          assertWithMessage(testDesc).that(imp.hasVideo()).isEqualTo(impVideo);
          assertWithMessage(testDesc).that(imp.hasBanner()).isEqualTo(impBanner);
          assertWithMessage(testDesc).that(imp.hasNative()).isEqualTo(impNativ);
          if (impVideo) {
            if (imp.getVideo().getCompanionadCount() != 0) {
              Banner compAd = imp.getVideo().getCompanionad(0);
              assertWithMessage(testDesc)
                  .that(
                      compAd.hasWmin() && compAd.hasWmax() && compAd.hasHmin() && compAd.hasHmax())
                  .isEqualTo(size > 1);
              assertWithMessage(testDesc)
                  .that(compAd.hasW() && compAd.hasH())
                  .isNotEqualTo(size == 0);
            }
          } else if (impBanner) {
            Banner banner = imp.getBanner();
            assertWithMessage(testDesc)
                .that(banner.hasWmin() && banner.hasWmax() && banner.hasHmin() && banner.hasHmax())
                .isEqualTo(size > 1);
            assertWithMessage(testDesc)
                .that(banner.hasW() && banner.hasH())
                .isNotEqualTo(size == 0);
            assertWithMessage(testDesc).that(banner.hasTopframe()).isEqualTo(size >= 2);
          } else if (impNativ) {
            Native nativ = imp.getNative();
            assertWithMessage(testDesc)
                .that(nativ.getRequestNative().getAssetsCount())
                .isEqualTo(size == 0 ? 5 : 10);
          }

          Bid.Builder bid = TestData.newBid(multiBid || imp.getInstl());
          if (impNativ) {
            NativeResponse nativResp = TestData.newNativeResponse(size - 1).build();
            if (size % 2 == 0) {
              bid.clearAdm().setAdmNative(nativResp);
            } else {
              bid.setAdm(
                  OpenRtbJsonFactory.create().newNativeWriter().writeNativeResponse(nativResp));
            }
          }
          BidResponse response = TestUtil.newBidResponse(bid);
          NetworkBid.BidResponse dcResponse =
              mapper.toExchangeBidResponse(request, response).build();
          if (linkExt) {
            Ad ad = dcResponse.getAd(0);
            assertWithMessage(testDesc)
                .that(ad.hasWidth())
                .isEqualTo(((size > 1 && multiBid) || imp.getInstl()) && !impNativ);
            assertWithMessage(testDesc)
                .that(ad.getAdslot(0).hasBillingId())
                .isEqualTo(size > 2 && multiBid);
          }
        }
      }
    }
  }