@@ -63,7 +63,7 @@ public void experimentWithoutTypeFieldHasNullType() {
6363 public void featureRolloutExperimentGetsEveryoneElseVariationInjected () {
6464 Experiment experiment = projectConfig .getExperimentKeyMapping ().get ("feature_rollout_experiment" );
6565 assertNotNull ("Experiment should exist" , experiment );
66- assertEquals ("feature_rollout" , experiment .getType ());
66+ assertEquals (Experiment . TYPE_FR , experiment .getType ());
6767
6868 // Should have 2 variations: original + everyone else
6969 assertEquals ("Should have 2 variations after injection" , 2 , experiment .getVariations ().size ());
@@ -107,7 +107,7 @@ public void variationMapsContainInjectedVariation() {
107107 public void abTestExperimentNotModified () {
108108 Experiment experiment = projectConfig .getExperimentKeyMapping ().get ("ab_test_experiment" );
109109 assertNotNull ("Experiment should exist" , experiment );
110- assertEquals ("a/b" , experiment .getType ());
110+ assertEquals (Experiment . TYPE_AB , experiment .getType ());
111111
112112 // Should still have exactly 2 original variations
113113 assertEquals ("A/B test should keep original 2 variations" , 2 , experiment .getVariations ().size ());
@@ -127,7 +127,7 @@ public void abTestExperimentNotModified() {
127127 public void featureRolloutWithEmptyRolloutIdDoesNotCrash () {
128128 Experiment experiment = projectConfig .getExperimentKeyMapping ().get ("rollout_no_rollout_id_experiment" );
129129 assertNotNull ("Experiment should exist" , experiment );
130- assertEquals ("feature_rollout" , experiment .getType ());
130+ assertEquals (Experiment . TYPE_FR , experiment .getType ());
131131
132132 // Should keep only original variation since rollout cannot be resolved
133133 assertEquals ("Should keep only original variation" , 1 , experiment .getVariations ().size ());
@@ -142,11 +142,11 @@ public void featureRolloutWithEmptyRolloutIdDoesNotCrash() {
142142 public void typeFieldCorrectlyParsed () {
143143 Experiment rolloutExp = projectConfig .getExperimentKeyMapping ().get ("feature_rollout_experiment" );
144144 assertNotNull (rolloutExp );
145- assertEquals ("feature_rollout" , rolloutExp .getType ());
145+ assertEquals (Experiment . TYPE_FR , rolloutExp .getType ());
146146
147147 Experiment abExp = projectConfig .getExperimentKeyMapping ().get ("ab_test_experiment" );
148148 assertNotNull (abExp );
149- assertEquals ("a/b" , abExp .getType ());
149+ assertEquals (Experiment . TYPE_AB , abExp .getType ());
150150
151151 Experiment noTypeExp = projectConfig .getExperimentKeyMapping ().get ("no_type_experiment" );
152152 assertNotNull (noTypeExp );
0 commit comments