Test: Rule Engine (#4103)

Co-authored-by: Alex Maltsev <and1sscsgo@gmail.com>
Co-authored-by: osulzhenko <osulzhenko@magnite.com>

Author: 3 people

src/test/groovy/org/prebid/server/functional/model/ModuleName.groovy

@@ -8,6 +8,7 @@ enum ModuleName {
     PB_RESPONSE_CORRECTION ("pb-response-correction"),
     ORTB2_BLOCKING("ortb2-blocking"),
     PB_REQUEST_CORRECTION('pb-request-correction'),
+    PB_RULE_ENGINE('pb-rule-engine')
 
     @JsonValue
     final String code

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

@@ -11,6 +11,7 @@ enum ModuleHookImplementation {
     ORTB2_BLOCKING_BIDDER_REQUEST("ortb2-blocking-bidder-request"),
     ORTB2_BLOCKING_RAW_BIDDER_RESPONSE("ortb2-blocking-raw-bidder-response"),
     PB_REQUEST_CORRECTION_PROCESSED_AUCTION_REQUEST("pb-request-correction-processed-auction-request"),
+    PB_RULES_ENGINE_PROCESSED_AUCTION_REQUEST("pb-rule-engine-processed-auction-request")
 
     @JsonValue
     final String code

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

@@ -0,0 +1,19 @@
+package org.prebid.server.functional.model.config
+
+import java.time.ZonedDateTime
+
+class PbRulesEngine {
+
+    Boolean enabled
+    Boolean generateRulesFromBidderConfig
+    ZonedDateTime timestamp
+    List<RuleSet> ruleSets
+
+    static PbRulesEngine createRulesEngineWithRule(Boolean enabled = true) {
+        new PbRulesEngine().tap {
+            it.enabled = enabled
+            it.generateRulesFromBidderConfig = false
+            it.ruleSets = [RuleSet.createRuleSets()]
+        }
+    }
+}

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

@@ -13,4 +13,5 @@ class PbsModulesConfig {
     Ortb2BlockingConfig ortb2Blocking
     PbResponseCorrection pbResponseCorrection
     PbRequestCorrectionConfig pbRequestCorrection
+    PbRulesEngine pbRuleEngine
 }

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

@@ -0,0 +1,22 @@
+package org.prebid.server.functional.model.config
+
+import com.fasterxml.jackson.annotation.JsonValue
+
+enum ResultFunction {
+
+    INCLUDE_BIDDERS("includeBidders"),
+    EXCLUDE_BIDDER("excludeBidders"),
+    LOG_A_TAG("logAtag")
+
+    String value
+
+    ResultFunction(String value) {
+        this.value = value
+    }
+
+    @Override
+    @JsonValue
+    String toString() {
+        return value
+    }
+}

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

@@ -0,0 +1,11 @@
+package org.prebid.server.functional.model.config
+
+import org.prebid.server.functional.model.bidder.BidderName
+
+class RuleEngineArguments {
+
+    List<BidderName> bidders
+    Integer seatNonBid
+    Boolean ifSyncedId
+    String analyticsValue
+}

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

@@ -0,0 +1,44 @@
+package org.prebid.server.functional.model.config
+
+import com.fasterxml.jackson.annotation.JsonValue
+
+enum RuleEngineFunction {
+
+    DEVICE_COUNTRY("deviceCountry", null),
+    DEVICE_COUNTRY_IN("deviceCountryIn", "countries"),
+    DATA_CENTER("dataCenter", null),
+    DATA_CENTER_IN("dataCenterIn", "datacenters"),
+    CHANNEL("channel", null),
+    EID_AVAILABLE("eidAvailable", null),
+    EID_IN("eidIn", "sources"),
+    USER_FPD_AVAILABLE("userFpdAvailable", null),
+    FPD_AVAILABLE("fpdAvailable", null),
+    GPP_SID_AVAILABLE("gppSidAvailable", null),
+    GPP_SID_IN("gppSidIn", "sids"),
+    TCF_IN_SCOPE("tcfInScope", null),
+    PERCENT("percent", "pct"),
+    PREBID_KEY("prebidKey", "key"),
+    DOMAIN("domain", null),
+    DOMAIN_IN("domainIn", "domains"),
+    BUNDLE("bundle", null),
+    BUNDLE_IN("bundleIn", "bundles"),
+    MEDIA_TYPE_IN("mediaTypeIn", "types"),
+    AD_UNIT_CODE("adUnitCode", null),
+    AD_UNIT_CODE_IN("adUnitCodeIn", "codes"),
+    DEVICE_TYPE("deviceType", null),
+    DEVICE_TYPE_IN("deviceTypeIn", "types")
+
+    private String value
+    private String fieldName
+
+    RuleEngineFunction(String value, String fieldName) {
+        this.value = value
+        this.fieldName = fieldName
+    }
+
+    @JsonValue
+    @Override
+    String toString() {
+        return value
+    }
+}

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

@@ -0,0 +1,37 @@
+package org.prebid.server.functional.model.config
+
+import com.fasterxml.jackson.annotation.JsonProperty
+import org.prebid.server.functional.util.PBSUtils
+
+class RuleEngineFunctionArgs {
+
+    List<Object> countries
+    List<Object> datacenters
+    List<Object> sources
+    List<Object> sids
+    @JsonProperty("pct")
+    Object percent
+    Object key
+    List<Object> domains
+    List<Object> bundles
+    List<Object> codes
+    List<Object> types
+    String operator
+    BigDecimal value
+    Currency currency
+
+    static RuleEngineFunctionArgs getDefaultFunctionArgs() {
+        new RuleEngineFunctionArgs().tap {
+            countries = [PBSUtils.randomString]
+            datacenters = [PBSUtils.randomString]
+            sources = [PBSUtils.randomString]
+            sids = [PBSUtils.randomNumber]
+            percent = PBSUtils.getRandomNumber(1, 100)
+            key = PBSUtils.randomString
+            domains = [PBSUtils.randomString]
+            bundles = [PBSUtils.randomString]
+            codes = [PBSUtils.randomString]
+            types = [PBSUtils.randomString]
+        }
+    }
+}

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

@@ -0,0 +1,7 @@
+package org.prebid.server.functional.model.config
+
+class RuleEngineModelDefault {
+
+    ResultFunction function
+    RuleEngineModelDefaultArgs args
+}

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

@@ -0,0 +1,6 @@
+package org.prebid.server.functional.model.config
+
+class RuleEngineModelDefaultArgs {
+
+    String analyticsValue
+}