def test_saves_existing_experiment_with_changelog(self):
experiment = NimbusExperimentFactory.create_with_status(
NimbusExperiment.Status.DRAFT,
application=NimbusExperiment.Application.FENIX,
hypothesis="Existing hypothesis",
name="Existing Name",
slug="existing-name",
public_description="Existing public description",
)
self.assertEqual(experiment.changes.count(), 1)
data = {
"application": NimbusExperiment.Application.DESKTOP,
"hypothesis": "New Hypothesis",
"name": "New Name",
"public_description": "New public description",
}
serializer = NimbusExperimentOverviewSerializer(
experiment, data=data, context={"user": self.user})
self.assertTrue(serializer.is_valid())
experiment = serializer.save()
self.assertEqual(experiment.changes.count(), 2)
self.assertEqual(experiment.application,
NimbusExperiment.Application.DESKTOP)
self.assertEqual(experiment.hypothesis, "New Hypothesis")
self.assertEqual(experiment.name, "New Name")
self.assertEqual(experiment.slug, "existing-name")
self.assertEqual(experiment.public_description,
"New public description")
def test_serializer_rejects_default_hypothesis(self):
data = {
"name": "Test 1234",
"hypothesis": NimbusExperiment.HYPOTHESIS_DEFAULT,
"application": NimbusExperiment.Application.DESKTOP.value,
"public_description": "Test description",
}
serializer = NimbusExperimentOverviewSerializer(
data=data, context={"user": self.user})
self.assertFalse(serializer.is_valid())
self.assertIn("hypothesis", serializer.errors)
def test_serializer_rejects_bad_name(self):
data = {
"name": "&^%&^%&^%&^%^&%^&",
"hypothesis": "Test hypothesis",
"application": NimbusExperiment.Application.DESKTOP.value,
"public_description": "Test description",
}
serializer = NimbusExperimentOverviewSerializer(
data=data, context={"user": self.user})
self.assertFalse(serializer.is_valid())
self.assertIn("Name needs to contain alphanumeric characters",
serializer.errors["name"])
def test_saves_new_experiment_with_changelog(self):
data = {
"application": NimbusExperiment.Application.DESKTOP,
"hypothesis": "It does the thing",
"name": "The Thing",
"public_description": "Does it do the thing?",
}
serializer = NimbusExperimentOverviewSerializer(
data=data, context={"user": self.user})
self.assertTrue(serializer.is_valid())
experiment = serializer.save()
self.assertEqual(experiment.changes.count(), 1)
self.assertEqual(experiment.application,
NimbusExperiment.Application.DESKTOP)
self.assertEqual(experiment.hypothesis, "It does the thing")
self.assertEqual(experiment.name, "The Thing")
self.assertEqual(experiment.slug, "the-thing")
def test_serializer_creates_experiment_and_sets_slug_and_owner(self):
data = {
"name": "Test 1234",
"hypothesis": "Test hypothesis",
"application": NimbusExperiment.Application.DESKTOP.value,
"public_description": "Test description",
}
serializer = NimbusExperimentOverviewSerializer(
data=data, context={"user": self.user})
self.assertTrue(serializer.is_valid())
experiment = serializer.save()
self.assertEqual(experiment.slug, slugify(data["name"]))
self.assertEqual(experiment.name, data["name"])
self.assertEqual(experiment.application, data["application"])
self.assertEqual(experiment.hypothesis, data["hypothesis"])
self.assertEqual(experiment.public_description,
data["public_description"])
# Owner should match the email of the user who created the experiment
self.assertEqual(experiment.owner, self.user)
def test_serializer_returns_error_for_non_unique_slug(self):
NimbusExperimentFactory.create_with_status(
NimbusExperiment.Status.ACCEPTED,
name="non unique slug",
slug="non-unique-slug",
)
data = {
"name": "non-unique slug",
"hypothesis": "Test hypothesis",
"application": NimbusExperiment.Application.DESKTOP.value,
"public_description": "Test description",
}
serializer = NimbusExperimentOverviewSerializer(
data=data, context={"user": self.user})
self.assertFalse(serializer.is_valid())
self.assertIn(
"Name maps to a pre-existing slug, please choose another name",
serializer.errors["name"],
)
def mutate(cls, root, info, input: UpdateExperimentInput):
exp = NimbusExperiment.objects.get(id=input.id)
serializer = NimbusExperimentOverviewSerializer(
exp, data=input, partial=True, context={"user": info.context.user})
return handle_with_serializer(cls, serializer,
input.client_mutation_id)
def mutate(cls, root, info, input: CreateExperimentInput):
serializer = NimbusExperimentOverviewSerializer(
data=input, context={"user": info.context.user})
return handle_with_serializer(cls, serializer,
input.client_mutation_id)