More fixes

ThePalaceProject · Apr 4, 2024 · 52a7119 · 52a7119
1 parent 4bfacd5
commit 52a7119
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/core/celery/tasks/test_collection_delete.py b/tests/core/celery/tasks/test_collection_delete.py
@@ -15,6 +15,7 @@ def test_collection_delete_job_collection(db: DatabaseTransactionFixture):
     assert CollectionDeleteJob.collection(db.session, 1) is None
 
     collection = db.collection(name="collection1")
+    assert collection.id is not None
     assert CollectionDeleteJob.collection(db.session, collection.id) == collection
 
 
@@ -42,6 +43,7 @@ def test_collection_delete_job_run(
 
     assert db.session.execute(query).scalar_one_or_none() == collection
 
+    assert collection.id is not None
     job = CollectionDeleteJob(mock_session_maker, collection.id)
     job.run()
     assert db.session.execute(query).scalar_one_or_none() is None