fix unit tests

bornhack · Dec 8, 2024 · 0c2ebd2 · 0c2ebd2
1 parent bb869ac
commit 0c2ebd2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/files/api.py b/src/files/api.py
@@ -159,12 +159,13 @@ def upload(  # noqa: C901,PLR0913
         ts.save()
         logger.debug(f"ThumbnailSource {ts.uuid} created for file {uploaded_file.uuid}")
 
-    # create jobs
-    uploaded_file.create_jobs()
-
     # get file using the manager so the returned object has annotations
     uploaded_file = BaseFile.bmanager.get(uuid=uploaded_file.uuid)
 
+    # create jobs
+    uploaded_file.create_jobs()
+    uploaded_file.refresh_from_db()
+
     # all good
     return 201, {"bma_response": uploaded_file, "message": f"File {uploaded_file.uuid} uploaded OK!"}
 

diff --git a/src/files/tests.py b/src/files/tests.py
@@ -975,7 +975,6 @@ def test_file_list_view_attribution_filters(self) -> None:
         self.assert_file_list_rows(24, qs="?attribution__icontains=foto")
         self.assert_file_list_rows(0, qs="?attribution__icontains=notthere")
         self.assert_file_list_rows(1, qs="?attribution__icontains=fotofonzy")
-        self.assert_file_list_rows(1, qs="?attribution=fotoflummer")
 
     def test_file_list_view_license_filters(self) -> None:
         """Test the license filter of the file list view."""