[Fix]: Add user_state_collection handling in unit test upload_parse_resume

Author: Stanford997

be_repo/app.py

@@ -52,7 +52,6 @@ def upload_resume():
     user_id = request.form.get('user_id')
     if not user_id:
         return jsonify({"error": "No user ID provided."}), 400
-    # Vector the resume text
     return upload_parse_resume(request, resume_collection, user_state_collection)
 
 

be_repo/tests/test_upload.py

@@ -35,12 +35,19 @@ def test_upload_parse_resume_success(mock_parse_resume, app):
     mock_request.form = {'user_id': '123'}
 
     mock_resume_collection = MagicMock()
+    mock_user_state_collection = MagicMock()
 
     with app.app_context():  # Set up the application context
-        response = upload_parse_resume(mock_request, mock_resume_collection)
+        response = upload_parse_resume(
+            mock_request,
+            mock_resume_collection,
+            mock_user_state_collection
+        )
 
     assert response[1] == 200
     assert response[0].json['message'] == "File successfully uploaded and parsed"
+    mock_resume_collection.replace_one.assert_called_once()
+    mock_user_state_collection.replace_one.assert_called_once()
 
 
 @patch('modules.upload.parse_resume', return_value="Parsed resume text")
@@ -53,12 +60,19 @@ def test_upload_parse_resume_invalid_format(mock_parse_resume, app):
     mock_request.form = {'user_id': '123'}
 
     mock_resume_collection = MagicMock()
+    mock_user_state_collection = MagicMock()
 
     with app.app_context():
-        response = upload_parse_resume(mock_request, mock_resume_collection)
+        response = upload_parse_resume(
+            mock_request,
+            mock_resume_collection,
+            mock_user_state_collection
+        )
 
     assert response[1] == 400
     assert response[0].json['error'] == "Invalid file format, only PDF is allowed"
+    mock_resume_collection.replace_one.assert_not_called()
+    mock_user_state_collection.replace_one.assert_not_called()
 
 
 @patch('modules.upload.parse_resume', return_value="Parsed resume text")
@@ -68,9 +82,16 @@ def test_upload_parse_resume_no_file(mock_parse_resume, app):
     mock_request.form = {'user_id': '123'}
 
     mock_resume_collection = MagicMock()
+    mock_user_state_collection = MagicMock()
 
     with app.app_context():
-        response = upload_parse_resume(mock_request, mock_resume_collection)
+        response = upload_parse_resume(
+            mock_request,
+            mock_resume_collection,
+            mock_user_state_collection
+        )
 
     assert response[1] == 400
     assert response[0].json['error'] == "No file part in the request"
+    mock_resume_collection.replace_one.assert_not_called()
+    mock_user_state_collection.replace_one.assert_not_called()