Updated tests of get_processed_data() function

tests/test_gradio.py

@@ -233,12 +233,14 @@ class TestGetProcessedData(unittest.TestCase):
             result = gradio.get_processed_data(
                 mock_pdf,
                 "https://github.com/user",
-                "Job posting content"
+                "Job posting content",
+                "Custom instructions"
             )
             
             self.assertEqual(result["linkedin"], mock_linkedin_result)
             self.assertEqual(result["github"], mock_github_result)
             self.assertEqual(result["job_post"], "Job posting content")
+            self.assertEqual(result["user_instructions"], "Custom instructions")
             self.assertEqual(len(result["errors"]), 0)
     
     def test_linkedin_error(self):
@@ -252,7 +254,7 @@ class TestGetProcessedData(unittest.TestCase):
                 "message": "PDF read failed"
             }
             
-            result = gradio.get_processed_data(mock_pdf, "", "")
+            result = gradio.get_processed_data(mock_pdf, "", "", "")
             
             self.assertIsNone(result["linkedin"])
             self.assertEqual(len(result["errors"]), 1)
@@ -266,7 +268,7 @@ class TestGetProcessedData(unittest.TestCase):
                 "message": "User not found"
             }
             
-            result = gradio.get_processed_data(None, "https://github.com/invalid", "")
+            result = gradio.get_processed_data(None, "https://github.com/invalid", "", "")
             
             self.assertIsNone(result["github"])
             self.assertEqual(len(result["errors"]), 1)
@@ -292,6 +294,7 @@ class TestGetProcessedData(unittest.TestCase):
             result = gradio.get_processed_data(
                 mock_pdf,
                 "https://github.com/user",
+                "",
                 ""
             )
             
@@ -304,15 +307,15 @@ class TestGetProcessedData(unittest.TestCase):
     def test_job_post_whitespace_handling(self):
         """Test job post whitespace handling."""
         # Test with leading/trailing whitespace
-        result = gradio.get_processed_data(None, "", "  Job content  ")
+        result = gradio.get_processed_data(None, "", "  Job content  ", "")
         self.assertEqual(result["job_post"], "Job content")
         
         # Test with only whitespace
-        result = gradio.get_processed_data(None, "", "   ")
+        result = gradio.get_processed_data(None, "", "   ", "")
         self.assertIsNone(result["job_post"])
         
         # Test with empty string
-        result = gradio.get_processed_data(None, "", "")
+        result = gradio.get_processed_data(None, "", "", "")
         self.assertIsNone(result["job_post"])
     
     def test_github_url_whitespace_handling(self):
@@ -321,20 +324,20 @@ class TestGetProcessedData(unittest.TestCase):
             mock_github.return_value = {"status": "success", "repositories": []}
             
             # Test with leading/trailing whitespace
-            result = gradio.get_processed_data(None, "  https://github.com/user  ", "")
+            result = gradio.get_processed_data(None, "  https://github.com/user  ", "", "")
             mock_github.assert_called_with("  https://github.com/user  ")
             
             # Test with only whitespace - should not call function
             mock_github.reset_mock()
-            result = gradio.get_processed_data(None, "   ", "")
+            result = gradio.get_processed_data(None, "   ", "", "")
             mock_github.assert_not_called()
     
     def test_data_structure_consistency(self):
         """Test that returned data structure is consistent."""
-        result = gradio.get_processed_data(None, "", "")
+        result = gradio.get_processed_data(None, "", "", "")
         
         # Check all required keys exist
-        required_keys = ["linkedin", "github", "job_post", "errors"]
+        required_keys = ["linkedin", "github", "job_post", "user_instructions", "errors"]
         for key in required_keys:
             self.assertIn(key, result)
         
@@ -352,13 +355,26 @@ class TestGetProcessedData(unittest.TestCase):
             "message": "Some warning"
         }
         
-        result = gradio.get_processed_data(mock_pdf, "", "")
+        result = gradio.get_processed_data(mock_pdf, "", "", "")
         
         # Warning status should not be treated as success
         self.assertIsNone(result["linkedin"])
         self.assertEqual(len(result["errors"]), 1)
         self.assertIn("LinkedIn: Some warning", result["errors"])
-
+    
+    def test_user_instructions_whitespace_handling(self):
+        """Test user instructions whitespace handling."""
+        # Test with leading/trailing whitespace
+        result = gradio.get_processed_data(None, "", "", "  Custom instructions  ")
+        self.assertEqual(result["user_instructions"], "Custom instructions")
+        
+        # Test with only whitespace
+        result = gradio.get_processed_data(None, "", "", "   ")
+        self.assertIsNone(result["user_instructions"])
+        
+        # Test with empty string
+        result = gradio.get_processed_data(None, "", "", "")
+        self.assertIsNone(result["user_instructions"])
 
 if __name__ == '__main__':
     unittest.main()