Add is_expired as arg to render factory

arxiv_vanity/papers/tests/test_models.py

@@ -1,7 +1,6 @@
 import datetime
 from django.conf import settings
 from django.test import TestCase, override_settings
-from django.utils import timezone
 import os
 import shutil
 from ..models import Render, Paper, SourceFile
@@ -65,8 +64,6 @@ def test_get_webhook_url(self):
     def test_not_deleted(self):
         paper = create_paper()
         render1 = create_render(paper=paper)
-        render1.created_at = datetime.datetime(1900, 1, 1).replace(tzinfo=timezone.utc)
-        render1.save()
         render2 = create_render(paper=paper)
 
         # haven't updated deleted status yet
@@ -84,9 +81,7 @@ def test_not_deleted(self):
 
     def test_expired(self):
         paper = create_paper()
-        render1 = create_render(paper=paper)
-        render1.created_at = datetime.datetime(1900, 1, 1).replace(tzinfo=timezone.utc)
-        render1.save()
+        render1 = create_render(paper=paper, is_expired=True)
         render2 = create_render(paper=paper)
 
         self.assertTrue(render1.is_expired())

arxiv_vanity/papers/tests/utils.py

@@ -4,6 +4,7 @@
 from unittest import mock
 import uuid
 from django.conf import settings
+from django.utils import timezone
 from ..models import Paper, Render, SourceFileBulkTarball, SourceFile, PaperQuerySet
 
 
@@ -56,14 +57,21 @@ def create_paper(
     )
 
 
-def create_render(paper=None, state=None):
-    return Render.objects.create(
+def create_render(paper=None, state=None, is_expired=False):
+    render = Render.objects.create(
         paper=paper or create_paper(), state=state or Render.STATE_UNSTARTED
     )
+    if is_expired:
+        render.created_at = datetime.datetime(1900, 1, 1).replace(tzinfo=timezone.utc)
+        render.save()
+        assert render.is_expired()
+    return render
 
 
-def create_render_with_html(paper=None):
-    render = create_render(paper=paper, state=Render.STATE_SUCCESS)
+def create_render_with_html(paper=None, is_expired=False):
+    render = create_render(
+        paper=paper, state=Render.STATE_SUCCESS, is_expired=is_expired
+    )
     source_path = os.path.join(os.path.dirname(__file__), "fixtures", "render.html")
     output_dir = os.path.join(settings.MEDIA_ROOT, "render-output", str(render.id))
     os.makedirs(output_dir)