Add functional tests for strict media type check

src/test/groovy/org/prebid/server/functional/model/config/Ortb2BlockingAttributeConfig.groovy

@@ -17,6 +17,7 @@ import static org.prebid.server.functional.model.config.Ortb2BlockingAttribute.V
 class Ortb2BlockingAttributeConfig {
 
  Boolean enforceBlocks
+ Boolean strictMediaType
 
  Object blockedAdomain
  Object blockedApp

src/test/groovy/org/prebid/server/functional/tests/module/ortb2blocking/Ortb2BlockingSpec.groovy

@@ -2,6 +2,7 @@ package org.prebid.server.functional.tests.module.ortb2blocking
 
 import org.prebid.server.functional.model.bidder.BidderName
 import org.prebid.server.functional.model.bidder.Generic
+import org.prebid.server.functional.model.config.AccountAuctionConfig
 import org.prebid.server.functional.model.config.AccountConfig
 import org.prebid.server.functional.model.config.AccountHooksConfiguration
 import org.prebid.server.functional.model.config.ExecutionPlan
@@ -16,6 +17,8 @@ import org.prebid.server.functional.model.db.Account
 import org.prebid.server.functional.model.request.auction.Asset
 import org.prebid.server.functional.model.request.auction.Audio
 import org.prebid.server.functional.model.request.auction.Banner
+import org.prebid.server.functional.model.request.auction.BidderControls
+import org.prebid.server.functional.model.request.auction.GenericPreferredBidder
 import org.prebid.server.functional.model.request.auction.Ix
 import org.prebid.server.functional.model.request.auction.BidRequest
 import org.prebid.server.functional.model.request.auction.Imp
@@ -53,12 +56,12 @@ import static org.prebid.server.functional.testcontainers.Dependencies.getNetwor
 
 class Ortb2BlockingSpec extends ModuleBaseSpec {
 
+ private static final String WILDCARD = '*'
  private static final Map IX_CONFIG = ["adapters.ix.enabled" : "true",
  "adapters.ix.endpoint": "$networkServiceContainer.rootUri/auction".toString()]
- private static final String WILDCARD = '*'
 
  private final PrebidServerService pbsServiceWithEnabledOrtb2Blocking = pbsServiceFactory.getService(ortb2BlockingSettings + IX_CONFIG +
- ["adapters.generic.ortb.multiformat-supported": "true"])
+ ['adapter-defaults.ortb.multiformat-supported': 'false'])
 
  def "PBS should send original array ortb2 attribute to bidder when enforce blocking is disabled"() {
  given: "Default bid request with proper ortb attribute"
@@ -131,6 +134,265 @@ class Ortb2BlockingSpec extends ModuleBaseSpec {
  PBSUtils.randomNumber | BTYPE
  }
 
+ def "PBS should be able to send original battr ortb2 attribute when strictMediaType doesn't enabled and original bidder doesn't contain such type impression"() {
+ given: "Account in the DB with blocking configuration"
+ def ortb2Attribute = PBSUtils.randomNumber
+ def account = getAccountWithOrtb2BlockingConfig(bidRequest.accountId, [ortb2Attribute], attributeName)
+ accountDao.save(account)
+
+ and: "Default bidder response with ortb2 attributes"
+ def bidResponse = BidResponse.getDefaultBidResponse(bidRequest).tap {
+ it.seatbid.first.bid = [getBidWithOrtb2Attribute(bidRequest.imp.first, ortb2Attribute, attributeName)]
+ }
+ bidder.setResponse(bidRequest.id, bidResponse)
+
+ when: "PBS processes the auction request"
+ def response = pbsServiceWithEnabledOrtb2Blocking.sendAuctionRequest(bidRequest)
+
+ then: "PBS request should contain proper ortb2 attributes from account config"
+ def bidderRequest = bidder.getBidderRequest(bidRequest.id)
+ assert getOrtb2Attributes(bidderRequest, attributeName) == [ortb2Attribute]*.toString()
+
+ and: "PBS request should contain multiple media types"
+ assert bidderRequest.imp.first.mediaTypes.size() == 2
+
+ and: "PBS response shouldn't contain any module errors"
+ assert !response?.ext?.prebid?.modules?.errors
+
+ and: "PBS response shouldn't contain any module warning"
+ assert !response?.ext?.prebid?.modules?.warnings
+
+ where:
+ bidRequest | attributeName
+ BidRequest.defaultVideoRequest | BANNER_BATTR
+ BidRequest.defaultAudioRequest | VIDEO_BATTR
+ BidRequest.defaultBidRequest | AUDIO_BATTR
+ }
+
+ def "PBS shouldn't be able to send original battr ortb2 attribute when strictMediaType enabled and original bidder doesn't contain such type impression"() {
+ given: "Account in the DB with blocking configuration"
+ def ortb2Attribute = PBSUtils.randomNumber
+ def account = getAccountWithOrtb2BlockingConfig(bidRequest.accountId, [ortb2Attribute], attributeName).tap {
+ config.hooks.modules.ortb2Blocking.attributes[attributeName].strictMediaType = true
+ }
+ accountDao.save(account)
+
+ and: "Default bidder response with ortb2 attributes"
+ def bidResponse = BidResponse.getDefaultBidResponse(bidRequest).tap {
+ it.seatbid.first.bid = [getBidWithOrtb2Attribute(bidRequest.imp.first, ortb2Attribute, attributeName)]
+ }
+ bidder.setResponse(bidRequest.id, bidResponse)
+
+ when: "PBS processes the auction request"
+ def response = pbsServiceWithEnabledOrtb2Blocking.sendAuctionRequest(bidRequest)
+
+ then: "PBS request shouldn't contain ortb2 attributes from account config for any media-type"
+ def bidderRequest = bidder.getBidderRequest(bidRequest.id)
+ assert !bidderRequest?.imp?.first?.banner?.battr
+ assert !bidderRequest?.imp?.first?.video?.battr
+ assert !bidderRequest?.imp?.first?.audio?.battr
+
+ and: "PBS request should contain single media type"
+ assert bidderRequest.imp.first.mediaTypes.size() == 1
+
+ and: "PBS response shouldn't contain any module errors"
+ assert !response?.ext?.prebid?.modules?.errors
+
+ and: "PBS response shouldn't contain any module warning"
+ assert !response?.ext?.prebid?.modules?.warnings
+
+ where:
+ bidRequest | attributeName
+ BidRequest.defaultVideoRequest | BANNER_BATTR
+ BidRequest.defaultAudioRequest | VIDEO_BATTR
+ BidRequest.defaultBidRequest | AUDIO_BATTR
+ }
+
+ def "PBS should be able to send original battr ortb2 attribute when strictMediaType enabled and default preferred media type specified at request level"() {
+ given: "Default bid request with multiply types"
+ def bidRequest = BidRequest.defaultBidRequest.tap {
+ imp.first.banner = Banner.defaultBanner
+ imp.first.video = Video.defaultVideo
+ imp.first.audio = Audio.defaultAudio
+ ext.prebid.bidderControls = new BidderControls(generic: new GenericPreferredBidder(preferredMediaType: preferredMediaType))
+ }
+
+ and: "Account in the DB with blocking configuration"
+ def ortb2Attribute = PBSUtils.randomNumber
+ def account = getAccountWithOrtb2BlockingConfig(bidRequest.accountId, [ortb2Attribute], attributeName).tap {
+ config.hooks.modules.ortb2Blocking.attributes[attributeName].strictMediaType = true
+ }
+ accountDao.save(account)
+
+ and: "Default bidder response with ortb2 attributes"
+ def bidResponse = BidResponse.getDefaultBidResponse(bidRequest).tap {
+ it.seatbid.first.bid = [getBidWithOrtb2Attribute(bidRequest.imp.first, ortb2Attribute, attributeName)]
+ }
+ bidder.setResponse(bidRequest.id, bidResponse)
+
+ when: "PBS processes the auction request"
+ def response = pbsServiceWithEnabledOrtb2Blocking.sendAuctionRequest(bidRequest)
+
+ then: "PBS request should contain proper ortb2 attributes from account config"
+ def bidderRequest = bidder.getBidderRequest(bidRequest.id)
+ assert getOrtb2Attributes(bidderRequest, attributeName) == [ortb2Attribute]*.toString()
+
+ and: "PBS request should contain multiple media types"
+ assert bidderRequest.imp.first.mediaTypes.size() == 2
+ assert bidderRequest.imp.first.mediaTypes.contains(preferredMediaType)
+
+ and: "PBS response shouldn't contain any module errors"
+ assert !response?.ext?.prebid?.modules?.errors
+
+ and: "PBS response shouldn't contain any module warning"
+ assert !response?.ext?.prebid?.modules?.warnings
+
+ where:
+ preferredMediaType | attributeName
+ VIDEO | BANNER_BATTR
+ AUDIO | VIDEO_BATTR
+ BANNER | AUDIO_BATTR
+ }
+
+ def "PBS shouldn't be able to send original battr ortb2 attribute when strictMediaType enabled and default preferred media type specified at request level"() {
+ given: "Default bid request with multiply types"
+ def bidRequest = BidRequest.defaultBidRequest.tap {
+ imp.first.banner = Banner.defaultBanner
+ imp.first.video = Video.defaultVideo
+ imp.first.audio = Audio.defaultAudio
+ ext.prebid.bidderControls = new BidderControls(generic: new GenericPreferredBidder(preferredMediaType: preferredMediaType))
+ }
+
+ and: "Account in the DB with blocking configuration"
+ def ortb2Attribute = PBSUtils.randomNumber
+ def account = getAccountWithOrtb2BlockingConfig(bidRequest.accountId, [ortb2Attribute], attributeName).tap {
+ config.hooks.modules.ortb2Blocking.attributes[attributeName].strictMediaType = true
+ }
+ accountDao.save(account)
+
+ and: "Default bidder response with ortb2 attributes"
+ def bidResponse = BidResponse.getDefaultBidResponse(bidRequest).tap {
+ it.seatbid.first.bid = [getBidWithOrtb2Attribute(bidRequest.imp.first, ortb2Attribute, attributeName)]
+ }
+ bidder.setResponse(bidRequest.id, bidResponse)
+
+ when: "PBS processes the auction request"
+ def response = pbsServiceWithEnabledOrtb2Blocking.sendAuctionRequest(bidRequest)
+
+ then: "PBS request shouldn't contain ortb2 attributes from account config for any media-type"
+ def bidderRequest = bidder.getBidderRequest(bidRequest.id)
+ assert !bidderRequest?.imp?.first?.banner?.battr
+ assert !bidderRequest?.imp?.first?.video?.battr
+ assert !bidderRequest?.imp?.first?.audio?.battr
+
+ and: "PBS request should contain only preferred media type"
+ assert bidderRequest.imp.first.mediaTypes == [preferredMediaType]
+
+ and: "PBS response shouldn't contain any module errors"
+ assert !response?.ext?.prebid?.modules?.errors
+
+ and: "PBS response shouldn't contain any module warning"
+ assert !response?.ext?.prebid?.modules?.warnings
+
+ where:
+ preferredMediaType | attributeName
+ VIDEO | BANNER_BATTR
+ AUDIO | VIDEO_BATTR
+ BANNER | AUDIO_BATTR
+ }
+
+ def "PBS should be able to send original battr ortb2 attribute when strictMediaType doesn't enabled and default preferred media type specified at account level"() {
+ given: "Default bid request with alias"
+ def bidRequest = BidRequest.defaultBidRequest.tap {
+ imp.first.banner = Banner.defaultBanner
+ imp.first.video = Video.defaultVideo
+ imp.first.audio = Audio.defaultAudio
+ }
+
+ and: "Account in the DB with blocking configuration"
+ def ortb2Attribute = PBSUtils.randomNumber
+ def account = getAccountWithOrtb2BlockingConfig(bidRequest.accountId, [ortb2Attribute], attributeName).tap {
+ config.auction = new AccountAuctionConfig(preferredMediaType: [(GENERIC): preferredMediaType])
+ }
+ accountDao.save(account)
+
+ and: "Default bidder response with ortb2 attributes"
+ def bidResponse = BidResponse.getDefaultBidResponse(bidRequest).tap {
+ it.seatbid.first.bid = [getBidWithOrtb2Attribute(bidRequest.imp.first, ortb2Attribute, attributeName)]
+ }
+ bidder.setResponse(bidRequest.id, bidResponse)
+
+ when: "PBS processes the auction request"
+ def response = pbsServiceWithEnabledOrtb2Blocking.sendAuctionRequest(bidRequest)
+
+ then: "PBS request should contain proper ortb2 attributes from account config"
+ def bidderRequest = bidder.getBidderRequest(bidRequest.id)
+ assert getOrtb2Attributes(bidderRequest, attributeName) == [ortb2Attribute]*.toString()
+
+ and: "PBS request should contain multiple media types"
+ assert bidderRequest.imp.first.mediaTypes.size() == 2
+ assert bidderRequest.imp.first.mediaTypes.contains(preferredMediaType)
+
+ and: "PBS response shouldn't contain any module errors"
+ assert !response?.ext?.prebid?.modules?.errors
+
+ and: "PBS response shouldn't contain any module warning"
+ assert !response?.ext?.prebid?.modules?.warnings
+
+ where:
+ preferredMediaType | attributeName
+ VIDEO | BANNER_BATTR
+ AUDIO | VIDEO_BATTR
+ BANNER | AUDIO_BATTR
+ }
+
+ def "PBS shouldn't be able to send original battr ortb2 attribute when strictMediaType enabled and default preferred media type specified at account level"() {
+ given: "Default bid request with multiply types"
+ def bidRequest = BidRequest.defaultBidRequest.tap {
+ imp.first.banner = Banner.defaultBanner
+ imp.first.video = Video.defaultVideo
+ imp.first.audio = Audio.defaultAudio
+ }
+
+ and: "Account in the DB with blocking configuration"
+ def ortb2Attribute = PBSUtils.randomNumber
+ def account = getAccountWithOrtb2BlockingConfig(bidRequest.accountId, [ortb2Attribute], attributeName).tap {
+ config.auction = new AccountAuctionConfig(preferredMediaType: [(GENERIC): preferredMediaType])
+ config.hooks.modules.ortb2Blocking.attributes[attributeName].strictMediaType = true
+ }
+ accountDao.save(account)
+
+ and: "Default bidder response with ortb2 attributes"
+ def bidResponse = BidResponse.getDefaultBidResponse(bidRequest).tap {
+ it.seatbid.first.bid = [getBidWithOrtb2Attribute(bidRequest.imp.first, ortb2Attribute, attributeName)]
+ }
+ bidder.setResponse(bidRequest.id, bidResponse)
+
+ when: "PBS processes the auction request"
+ def response = pbsServiceWithEnabledOrtb2Blocking.sendAuctionRequest(bidRequest)
+
+ then: "PBS request shouldn't contain ortb2 attributes from account config for any media-type"
+ def bidderRequest = bidder.getBidderRequest(bidRequest.id)
+ assert !bidderRequest?.imp?.first?.banner?.battr
+ assert !bidderRequest?.imp?.first?.video?.battr
+ assert !bidderRequest?.imp?.first?.audio?.battr
+
+ and: "PBS request should contain only preferred media type"
+ assert bidderRequest.imp.first.mediaTypes == [preferredMediaType]
+
+ and: "PBS response shouldn't contain any module errors"
+ assert !response?.ext?.prebid?.modules?.errors
+
+ and: "PBS response shouldn't contain any module warning"
+ assert !response?.ext?.prebid?.modules?.warnings
+
+ where:
+ preferredMediaType | attributeName
+ VIDEO | BANNER_BATTR
+ AUDIO | VIDEO_BATTR
+ BANNER | AUDIO_BATTR
+ }
+
  def "PBS shouldn't send original single ortb2 attribute to bidder when enforce blocking is disabled"() {
  given: "Default bid request with proper ortb attribute"
  def bidRequest = getBidRequestForOrtbAttribute(attributeName)