Add tests for /downloads/latest/prerelease redirect

Author: hugovk

downloads/tests/test_models.py

@@ -74,6 +74,28 @@ def test_latest_python3(self):
         latest_3_99 = Release.objects.latest_python3(minor_version=99)
         self.assertIsNone(latest_3_99)
 
+    def test_latest_prerelease(self):
+        latest_prerelease = Release.objects.latest_prerelease()
+        self.assertEqual(latest_prerelease, self.pre_release)
+
+        # Create a newer prerelease with a future date
+        newer_prerelease = Release.objects.create(
+            version=Release.PYTHON3,
+            name="Python 3.9.99",
+            is_published=True,
+            pre_release=True,
+            release_date=self.pre_release.release_date + dt.timedelta(days=1),
+        )
+        latest_prerelease = Release.objects.latest_prerelease()
+        self.assertEqual(latest_prerelease, newer_prerelease)
+        self.assertNotEqual(latest_prerelease, self.pre_release)
+
+    def test_latest_prerelease_when_no_prerelease(self):
+        # Delete the prerelease
+        self.pre_release.delete()
+        latest_prerelease = Release.objects.latest_prerelease()
+        self.assertIsNone(latest_prerelease)
+
     def test_get_version(self):
         self.assertEqual(self.release_275.name, 'Python 2.7.5')
         self.assertEqual(self.release_275.get_version(), '2.7.5')

downloads/tests/test_views.py

@@ -95,6 +95,18 @@ def test_latest_python3x_redirects(self):
         response = self.client.get(url)
         self.assertRedirects(response, reverse("download:download"))
 
+    def test_latest_prerelease_redirect(self):
+        url = reverse("download:download_latest_prerelease")
+        response = self.client.get(url)
+        self.assertRedirects(response, self.pre_release.get_absolute_url())
+
+    def test_latest_prerelease_redirect_when_no_prerelease(self):
+        # Delete the prerelease to test fallback
+        self.pre_release.delete()
+        url = reverse("download:download_latest_prerelease")
+        response = self.client.get(url)
+        self.assertRedirects(response, reverse("download:download"))
+
     def test_redirect_page_object_to_release_detail_page(self):
         self.release_275.release_page = None
         self.release_275.save()