Add prerequisites in split dto

Author: nmayorsplit

dtos/split.go

@@ -38,13 +38,20 @@ type SplitDTO struct {
 	Configurations        map[string]string `json:"configurations"`
 	Sets                  []string          `json:"sets"`
 	ImpressionsDisabled   bool              `json:"impressionsDisabled"`
+	Prerequisites         []Prerequisite    `json:"prerequisites"`
 }
 
 // MarshalBinary exports SplitDTO to JSON string
 func (s SplitDTO) MarshalBinary() (data []byte, err error) {
 	return json.Marshal(s)
 }
 
+// Prerequisites structure to map a Prerequisite fetched from JSON message
+type Prerequisite struct {
+	FeatureFlagName string   `json:"n"`
+	Treatments      []string `json:"ts"`
+}
+
 // ConditionDTO structure to map a Condition fetched from JSON message.
 type ConditionDTO struct {
 	ConditionType string          `json:"conditionType"`

dtos/split_test.go

@@ -50,6 +50,18 @@ func TestSplitDTO(t *testing.T) {
 			t.Error("Marshal struct mal formed [Configurations]")
 		}
 
+		if splitChangesDtoFromMarshal.FeatureFlags.Splits[0].Prerequisites == nil {
+			t.Error("Marshal struct mal formed [Prerequisites]")
+		}
+
+		if splitChangesDtoFromMarshal.FeatureFlags.Splits[0].Prerequisites[0].FeatureFlagName != "rbs_test_flag" {
+			t.Error("Marshal struct mal formed [Prerequisites]")
+		}
+
+		if splitChangesDtoFromMarshal.FeatureFlags.Splits[0].Prerequisites[0].Treatments[0] != "v1" {
+			t.Error("Marshal struct mal formed [Prerequisites]")
+		}
+
 		if splitChangesDtoFromMarshal.FeatureFlags.Splits[0].Configurations["of"] !=
 			splitChangesDtoFromMock.FeatureFlags.Splits[0].Configurations["of"] {
 			t.Error("Marshal struct mal formed [Configurations]")

testdata/split_mock.json

@@ -12,6 +12,14 @@
   "configurations": {
     "on": "{\"color\": \"blue\",\"size\": 13}"
   },
+  "prerequisites": [
+          {
+            "n": "rbs_test_flag",
+            "ts": [
+              "v1"
+            ]
+          }
+        ],
   "conditions": [
     {
       "conditionType": "WHITELIST",