def test_serializer_outputs_expected_int_val(self):
experiment = ExperimentFactory(pref_type=Experiment.PREF_TYPE_INT)
variant = ExperimentVariantFactory.create(
experiment=experiment, value="28"
)
serializer = ExperimentRecipeVariantSerializer(variant)
self.assertEqual(type(serializer.data["value"]), int)
self.assertEqual(
serializer.data,
{"ratio": variant.ratio, "slug": variant.slug, "value": 28},
)
def test_serializer_outputs_expected_schema(self):
experiment = ExperimentFactory(pref_type=Experiment.PREF_TYPE_STR)
variant = ExperimentVariantFactory.create(experiment=experiment)
serializer = ExperimentRecipeVariantSerializer(variant)
self.assertEqual(
serializer.data,
{
"ratio": variant.ratio,
"slug": variant.slug,
"value": variant.value,
},
)
def test_serializer_outputs_expected_bool(self):
experiment = ExperimentFactory(pref_type=Experiment.PREF_TYPE_BOOL)
variant = ExperimentVariantFactory.create(experiment=experiment,
value="true")
serializer = ExperimentRecipeVariantSerializer(variant)
self.assertEqual(type(serializer.data["value"]), bool)
self.assertEqual(serializer.data, {
"ratio": variant.ratio,
"slug": variant.slug,
"value": True
})
def test_serializer_outputs_expected_schema(self):
experiment = ExperimentFactory.create_with_status(
Experiment.STATUS_SHIP
)
serializer = ExperimentRecipePrefArgumentsSerializer(experiment)
self.assertEqual(
serializer.data,
{
"preferenceBranchType": experiment.pref_branch,
"slug": experiment.normandy_slug,
"experimentDocumentUrl": experiment.experiment_url,
"preferenceName": experiment.pref_key,
"preferenceType": experiment.pref_type,
"branches": [
ExperimentRecipeVariantSerializer(variant).data
for variant in experiment.variants.all()
],
},
)
