test: add nil cases

Author: Luisfc68

internal/entities/quote/common.go

@@ -46,14 +46,32 @@ func ValidateQuoteHash(hash string) error {
 }
 
 func CalculateCallFee(amount *entities.Wei, config PegConfiguration) *entities.Wei {
+	var percentage *utils.BigFloat
+	var fixedFee *entities.Wei
 	result := new(entities.Wei)
 
-	percentageFee := calculatePercentageFee(amount, config.GetFeePercentage())
-	result.Add(percentageFee, config.GetFixedFee())
+	if config.GetFeePercentage() != nil {
+		percentage = config.GetFeePercentage()
+	} else {
+		percentage = utils.NewBigFloat64(0)
+	}
+
+	if amount == nil {
+		amount = entities.NewBigWei(big.NewInt(0))
+	}
+
+	if config.GetFixedFee() != nil {
+		fixedFee = config.GetFixedFee()
+	} else {
+		fixedFee = entities.NewBigWei(big.NewInt(0))
+	}
+
+	percentageFee := calculatePercentageFee(amount, percentage)
+	result.Add(percentageFee, fixedFee)
 
-	log.Debugf("Percentage fee: %v%% of %v = %v", config.GetFeePercentage(), amount, percentageFee)
-	log.Debugf("Fixed fee: %v", config.GetFixedFee())
-	log.Debugf("Call fee: %v + %v = %v", percentageFee, config.GetFixedFee(), result)
+	log.Debugf("Percentage fee: %v%% of %v = %v", percentage, amount, percentageFee)
+	log.Debugf("Fixed fee: %v", fixedFee)
+	log.Debugf("Call fee: %v + %v = %v", percentageFee, fixedFee, result)
 	return result
 }
 

internal/entities/quote/common_test.go

@@ -117,6 +117,24 @@ func TestCalculateCallFee(t *testing.T) {
 			fixedFee:      entities.NewWei(1110000031224),
 			result:        entities.NewWei(6043334443333364566),
 		},
+		{
+			amount:        entities.NewWei(7777777777777777789),
+			feePercentage: nil,
+			fixedFee:      entities.NewWei(1110000031224),
+			result:        entities.NewWei(1110000031224),
+		},
+		{
+			amount:        entities.NewWei(7777777777777777789),
+			feePercentage: utils.NewBigFloat64(77.7),
+			fixedFee:      nil,
+			result:        entities.NewWei(6043333333333333342),
+		},
+		{
+			amount:        nil,
+			feePercentage: utils.NewBigFloat64(77.7),
+			fixedFee:      entities.NewWei(1110000031224),
+			result:        entities.NewWei(1110000031224),
+		},
 	}
 	log.SetLevel(log.DebugLevel)
 	for _, tt := range testCases {