add test cases for logging

test_app.py

@@ -57,3 +57,57 @@ class TestGeneratePrompt:
 
         assert result == "Could not generate a prompt. No fish today."
 
+    @patch("app.logger")
+    @patch("app.OpenAI")
+    def test_logging_on_success(self, mock_openai_class, mock_logger):
+        """Test that logging works correctly on successful API call."""
+        api_total_token_count = 500
+
+        # Mock the OpenAI client and response
+        mock_client = Mock()
+        mock_openai_class.return_value = mock_client
+
+        mock_response = Mock()
+        mock_response.choices = [Mock()]
+        mock_response.choices[0].message = Mock()
+        mock_response.choices[0].message.content = TEST_OPENAI_CHAT_RESPONSE_CONTENT
+        mock_response.usage.total_tokens = api_total_token_count
+
+        mock_client.chat.completions.create.return_value = mock_response
+
+        # Mock the gradio request
+        mock_gradio_request = Mock()
+        mock_gradio_request.session_hash = TEST_GRADIO_SESSION_HASH
+
+        # Call the function
+        generate_prompt(mock_gradio_request)
+
+        mock_logger.info.assert_any_call(
+            f"Making OpenAI API request - Session: {TEST_GRADIO_SESSION_HASH}"
+        )
+        mock_logger.info.assert_any_call(
+            f"API request successful - Session: {TEST_GRADIO_SESSION_HASH}"
+            f" - tokens used: {api_total_token_count}"
+        )
+
+    @patch("app.logger")
+    @patch("app.OpenAI")
+    def test_logging_on_failure(self, mock_openai_class, mock_logger):
+        """Test that logging works correctly on failed API call."""
+        api_error_message = "OpenAI is down."
+
+        # Mock the OpenAI client and response
+        mock_client = Mock()
+        mock_openai_class.return_value = mock_client
+        mock_client.chat.completions.create.side_effect = Exception(api_error_message)
+
+        # Mock the gradio request
+        mock_gradio_request = Mock()
+        mock_gradio_request.session_hash = TEST_GRADIO_SESSION_HASH
+
+        # Call the function
+        generate_prompt(mock_gradio_request)
+
+        mock_logger.error.assert_any_call(
+            f"API request failed - Session: {TEST_GRADIO_SESSION_HASH} - Error: {api_error_message}"
+        )