Refactor tests

saritasa-nest · Dec 7, 2023 · 87d87f2 · 87d87f2
1 parent 46bb743
commit 87d87f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_api/test_export.py b/tests/test_api/test_export.py
@@ -17,8 +17,8 @@ def test_export_api_creates_export_job(admin_api_client: test.APIClient):
         },
     )
     assert response.status_code == status.HTTP_201_CREATED
-    assert response.data["export_status"] == "CREATED"
-    assert ExportJob.objects.first()
+    assert response.data["export_status"] == ExportJob.ExportStatus.CREATED
+    assert ExportJob.objects.filter(id=response.data["id"]).exists()
 
 
 @pytest.mark.django_db(transaction=True)