Fix tests

Author: stephenmcconkey

client/helpers_test.go

@@ -53,7 +53,13 @@ func MakeStringFeatureConfigs(name, defaultVariation, offVariation, state string
 
 	// If there are any PreReqs then we need to store them as flags as well.
 	for _, x := range preReqs {
-		featureConfig = append(featureConfig, MakeBoolFeatureConfig(x.Feature, "true", "false", x.Variations[0], nil))
+		var state string
+		if x.Variations[0] == "true" {
+			state = "on"
+		} else {
+			state = "off"
+		}
+		featureConfig = append(featureConfig, MakeBoolFeatureConfig(x.Feature, "true", "false", state, nil))
 	}
 
 	return featureConfig

tests/evaluator_test.go

@@ -84,10 +84,10 @@ func TestEvaluator(t *testing.T) {
 			t.Error(err)
 		}
 		for _, flag := range fixture.Flags {
-			repo.SetFlag(flag)
+			repo.SetFlag(flag, false)
 		}
 		for _, segment := range fixture.Segments {
-			repo.SetSegment(segment)
+			repo.SetSegment(segment, false)
 		}
 
 		for _, testCase := range fixture.Tests {