UOE-12687: Support for Tier4 and Tier5 in MBMF

Author: pm-saurabh-narkhede

adapters/pubmatic/pubmatic.go

@@ -22,7 +22,7 @@ import (
 )
 
 const MAX_IMPRESSIONS_PUBMATIC = 30
-const MAX_MULTIFLOORS_PUBMATIC = 3
+const MAX_MULTIFLOORS_PUBMATIC = 5
 
 const (
 	ae                  = "ae"

modules/pubmatic/openwrap/database/mysql/mbmf_test.go

@@ -48,7 +48,7 @@ func TestMySqlDBGetProfileAdUnitMultiFloors(t *testing.T) {
 			wantErr: errors.New("row error"),
 		},
 		{
-			name: "Success case",
+			name: "Success case with different no. of tiers floors",
 			fields: fields{
 				cfg: config.Database{
 					MaxDbContextTimeout: 5,
@@ -64,7 +64,7 @@ func TestMySqlDBGetProfileAdUnitMultiFloors(t *testing.T) {
 				}
 				rows := sqlmock.NewRows([]string{"adunit_name", "profile_id", "value"}).
 					AddRow("adunit1", 12344, `{"isActive":true,"tier1":1.0,"tier2":0.8,"tier3":0.6}`).
-					AddRow("adunit2", 54532, `{"isActive":true,"tier1":2.0,"tier2":1.6,"tier3":1.2}`)
+					AddRow("adunit2", 54532, `{"isActive":true,"tier1":2.0,"tier2":1.6,"tier3":1.2,"tier4":2.4}`)
 				mock.ExpectQuery("SELECT").WillReturnRows(rows)
 				return db
 			},
@@ -73,7 +73,7 @@ func TestMySqlDBGetProfileAdUnitMultiFloors(t *testing.T) {
 					"adunit1": {IsActive: true, Tier1: 1.0, Tier2: 0.8, Tier3: 0.6},
 				},
 				54532: map[string]*models.MultiFloors{
-					"adunit2": {IsActive: true, Tier1: 2.0, Tier2: 1.6, Tier3: 1.2},
+					"adunit2": {IsActive: true, Tier1: 2.0, Tier2: 1.6, Tier3: 1.2, Tier4: 2.4},
 				},
 			},
 			wantErr: nil,

modules/pubmatic/openwrap/models/openwrap.go

@@ -270,6 +270,8 @@ type MultiFloors struct {
 	Tier1    float64 `json:"tier1,omitempty"`
 	Tier2    float64 `json:"tier2,omitempty"`
 	Tier3    float64 `json:"tier3,omitempty"`
+	Tier4    float64 `json:"tier4,omitempty"`
+	Tier5    float64 `json:"tier5,omitempty"`
 }
 
 func (w WinningBids) IsWinningBid(impId, bidId string) bool {

modules/pubmatic/openwrap/models/utils.go

@@ -473,5 +473,13 @@ func GetMultiFloors(multiFloors map[string]*MultiFloors, impID string) []float64
 	if _, ok := multiFloors[impID]; !ok || multiFloors[impID] == nil {
 		return nil
 	}
-	return []float64{multiFloors[impID].Tier1, multiFloors[impID].Tier2, multiFloors[impID].Tier3}
+
+	multifloors := []float64{multiFloors[impID].Tier1, multiFloors[impID].Tier2, multiFloors[impID].Tier3}
+	if multiFloors[impID].Tier4 > 0 {
+		multifloors = append(multifloors, multiFloors[impID].Tier4)
+	}
+	if multiFloors[impID].Tier5 > 0 {
+		multifloors = append(multifloors, multiFloors[impID].Tier5)
+	}
+	return multifloors
 }

modules/pubmatic/openwrap/models/utils_test.go

@@ -1643,6 +1643,16 @@ func TestGetMultiFloors(t *testing.T) {
 			impID: "123",
 			want:  []float64{1.0, 0.8, 0.6},
 		},
+		{
+			name: "impID present with multifloors with tier4 and tier5",
+			rctx: &RequestCtx{
+				MultiFloors: map[string]*MultiFloors{
+					"123": {Tier1: 1.0, Tier2: 0.8, Tier3: 0.6, Tier4: 0.4, Tier5: 0.2},
+				},
+			},
+			impID: "123",
+			want:  []float64{1.0, 0.8, 0.6, 0.4, 0.2},
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {