Updated test cases

Author: nmayorsplit

dtos/fallbacktreatment.go

@@ -6,7 +6,7 @@ const (
 
 type FallbackTreatment struct {
 	Treatment string
-	Config    string
+	Config    *string
 	Label     *string
 }
 

dtos/fallbacktreatment_test.go

@@ -8,15 +8,15 @@ import (
 
 func TestFallbackTreatmentCalculatorResolve(t *testing.T) {
 	// Initial setup with both global and flag-specific treatments
+	stringConfig := "flag1_config"
 	config := &FallbackTreatmentConfig{
 		GlobalFallbackTreatment: &FallbackTreatment{
 			Treatment: "global_treatment",
-			Config:    "global_config",
 		},
 		byFlagFallbackTreatment: map[string]FallbackTreatment{
 			"flag1": {
 				Treatment: "flag1_treatment",
-				Config:    "flag1_config",
+				Config:    &stringConfig,
 			},
 		},
 	}
@@ -26,38 +26,38 @@ func TestFallbackTreatmentCalculatorResolve(t *testing.T) {
 	label := "some_label"
 	result := calc.Resolve("flag1", &label)
 	assert.Equal(t, "flag1_treatment", result.Treatment)
-	assert.Equal(t, "flag1_config", result.Config)
+	assert.Equal(t, &stringConfig, result.Config)
 	assert.Equal(t, "fallback - some_label", *result.Label)
 
 	// Test fallback to global treatment when flag not found
 	result = calc.Resolve("flag2", &label)
 	assert.Equal(t, "global_treatment", result.Treatment)
-	assert.Equal(t, "global_config", result.Config)
+	assert.Nil(t, result.Config)
 	assert.Equal(t, "fallback - some_label", *result.Label)
 
 	// Test nil label handling
 	result = calc.Resolve("flag1", nil)
 	assert.Equal(t, "flag1_treatment", result.Treatment)
-	assert.Equal(t, "flag1_config", result.Config)
+	assert.Equal(t, &stringConfig, result.Config)
 	assert.Nil(t, result.Label)
 
 	// Test default control when no config
 	calcNoConfig := NewFallbackTreatmentCalculatorImp(nil)
 	result = calcNoConfig.Resolve("flag1", &label)
 	assert.Equal(t, "control", result.Treatment)
-	assert.Equal(t, "", result.Config)
+	assert.Nil(t, result.Config)
 	assert.Equal(t, "some_label", *result.Label)
 
 	// Test global treatment when no flag-specific treatments exist
 	configGlobalOnly := &FallbackTreatmentConfig{
 		GlobalFallbackTreatment: &FallbackTreatment{
 			Treatment: "global_treatment",
-			Config:    "global_config",
+			Config:    nil,
 		},
 	}
 	calcGlobalOnly := NewFallbackTreatmentCalculatorImp(configGlobalOnly)
 	result = calcGlobalOnly.Resolve("any_flag", &label)
 	assert.Equal(t, "global_treatment", result.Treatment)
-	assert.Equal(t, "global_config", result.Config)
+	assert.Nil(t, result.Config)
 	assert.Equal(t, "fallback - some_label", *result.Label)
-}
+}