def test_experiment_firefox_channel_sort_does_sorting(self):
ExperimentFactory.create(firefox_channel=Experiment.CHANNEL_NIGHTLY)
ExperimentFactory.create(firefox_channel=Experiment.CHANNEL_RELEASE)
ExperimentFactory.create(firefox_channel=Experiment.CHANNEL_BETA)
ExperimentFactory.create(firefox_channel="")
sorted_experiments = (Experiment.objects.annotate(
firefox_channel_sort=Experiment.firefox_channel_sort()).order_by(
"firefox_channel_sort").values("firefox_channel_sort"))
self.assertEqual(
[r["firefox_channel_sort"] for r in sorted_experiments],
[
Experiment.CHANNEL_UNSET_ORDER,
Experiment.CHANNEL_NIGHTLY_ORDER,
Experiment.CHANNEL_BETA_ORDER,
Experiment.CHANNEL_RELEASE_ORDER,
],
)
def get_queryset(self):
qs = super().get_queryset()
qs = qs.annotate(
firefox_channel_sort=Experiment.firefox_channel_sort())
return qs