fix value type: use interface{} (#60)

* fix value type: use interface{} * bump engine-test-data: to include different types
Flagsmith · Feb 1, 2023 · 473dd0d · 473dd0d
1 parent 6880ad7
commit 473dd0d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/flagengine/engine-test-data b/flagengine/engine-test-data
diff --git a/flagengine/features/models.go b/flagengine/features/models.go
@@ -4,7 +4,6 @@ import (
 	"encoding/json"
 	"sort"
 	"strconv"
-	"strings"
 
 	"github.com/Flagsmith/flagsmith-go-client/v2/flagengine/utils"
 )
@@ -25,7 +24,7 @@ type FeatureStateModel struct {
 	DjangoID                       int                                   `json:"django_id"`
 	FeatureStateUUID               string                                `json:"featurestate_uuid"`
 	MultivariateFeatureStateValues []*MultivariateFeatureStateValueModel `json:"multivariate_feature_state_values"`
-	RawValue                       string                                `json:"feature_state_value"`
+	RawValue                       interface{}                           `json:"feature_state_value"`
 }
 
 func (fs *FeatureStateModel) UnmarshalJSON(bytes []byte) error {
@@ -36,7 +35,7 @@ func (fs *FeatureStateModel) UnmarshalJSON(bytes []byte) error {
 		DjangoID                       int                                   `json:"django_id"`
 		FeatureStateUUID               string                                `json:"featurestate_uuid"`
 		MultivariateFeatureStateValues []*MultivariateFeatureStateValueModel `json:"multivariate_feature_state_values"`
-		RawValue                       json.RawMessage                       `json:"feature_state_value"`
+		RawValue                       interface{}                           `json:"feature_state_value"`
 	}
 
 	err := json.Unmarshal(bytes, &obj)
@@ -50,13 +49,13 @@ func (fs *FeatureStateModel) UnmarshalJSON(bytes []byte) error {
 	fs.DjangoID = obj.DjangoID
 	fs.FeatureStateUUID = obj.FeatureStateUUID
 	fs.MultivariateFeatureStateValues = obj.MultivariateFeatureStateValues
-	fs.RawValue = strings.Trim(string(obj.RawValue), `"`)
+	fs.RawValue = obj.RawValue
 	return nil
 }
 
 type MultivariateFeatureOptionModel struct {
-	ID    int    `json:"id"`
-	Value string `json:"value"`
+	ID    int         `json:"id"`
+	Value interface{} `json:"value"`
 }
 
 type MultivariateFeatureStateValueModel struct {