one more try

AditiR-42 · Dec 12, 2024 · f73e9e8 · f73e9e8
1 parent 77bc23e
commit f73e9e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/models/tests/integration/test_api_endpoints.py b/src/models/tests/integration/test_api_endpoints.py
@@ -488,7 +488,7 @@ async def test_process_pdf_success(mock_process_pdf_privacy_issues):
     mock_process_pdf_privacy_issues.return_value = ["Issue 1", "Issue 2"]
 
     pdf_file = UploadFile(filename="test.pdf", file=BytesIO(b"fake pdf content"))
-    result = process_pdf(pdf_file)
+    result = await process_pdf(pdf_file)
 
     assert result == {
         "message": "Processing completed successfully.",
@@ -508,7 +508,7 @@ async def test_get_grade_success(mock_privacy_grader):
     # Simulate that issues have been processed
     parsed_issues_storage = {"issues": ["Issue 1", "Issue 2"]}
 
-    result = get_grade()
+    result = await get_grade()
 
     assert result == {
         "overall_grade": "A",