fix unittestcase

Author: Pavan-Microsoft

src/tests/api/common/config/test_config.py

@@ -52,7 +52,6 @@ def test_config_initialization(mock_env_vars):
 
         # AI Project Client
         assert config.use_ai_project_client is True
-        assert config.azure_ai_project_conn_string == "Endpoint=sb://test/"
 
         # Chat history config
         assert config.use_chat_history_enabled is True

src/tests/api/helpers/test_chat_helper.py

@@ -13,7 +13,7 @@
 
 class TestChatHelper:
     @patch("helpers.chat_helper.Config")
-    @patch("helpers.chat_helper.openai.AzureOpenAI")
+    @patch("helpers.azure_openai_helper.openai.AzureOpenAI")
     def test_process_rag_response_success(self, mock_azure_openai, mock_config):
         # Mock the Azure OpenAI client and its response
         mock_client = MagicMock()
@@ -47,7 +47,7 @@ def test_process_rag_response_success(self, mock_azure_openai, mock_config):
         assert call_args["messages"][1]["role"] == "user"
 
     @patch("helpers.chat_helper.Config")
-    @patch("helpers.chat_helper.openai.AzureOpenAI")
+    @patch("helpers.azure_openai_helper.openai.AzureOpenAI")
     def test_process_rag_response_with_code_blocks(self, mock_azure_openai, mock_config):
         # Mock the Azure OpenAI client and its response
         mock_client = MagicMock()
@@ -72,7 +72,7 @@ def test_process_rag_response_with_code_blocks(self, mock_azure_openai, mock_con
         assert result == expected
 
     @patch("helpers.chat_helper.Config")
-    @patch("helpers.chat_helper.openai.AzureOpenAI")
+    @patch("helpers.azure_openai_helper.openai.AzureOpenAI")
     def test_process_rag_response_error(self, mock_azure_openai, mock_config):
         # Mock the Azure OpenAI client
         mock_client = MagicMock()
@@ -93,7 +93,7 @@ def test_process_rag_response_error(self, mock_azure_openai, mock_config):
         assert result["error"] == "Chart could not be generated from this data. Please ask a different question."
 
     @patch("helpers.chat_helper.Config")
-    @patch("helpers.chat_helper.openai.AzureOpenAI")
+    @patch("helpers.azure_openai_helper.openai.AzureOpenAI")
     def test_process_rag_response_invalid_json(self, mock_azure_openai, mock_config):
         # Mock the Azure OpenAI client
         mock_client = MagicMock()

src/tests/api/plugins/test_chat_with_data_plugin.py

@@ -26,7 +26,7 @@ def chat_plugin(mock_config):
 
 class TestChatWithDataPlugin:
     @patch("plugins.chat_with_data_plugin.get_bearer_token_provider")
-    @patch("plugins.chat_with_data_plugin.openai.AzureOpenAI")
+    @patch("helpers.azure_openai_helper.openai.AzureOpenAI")
     @pytest.mark.asyncio
     async def test_greeting(self, mock_azure_openai, mock_token_provider, chat_plugin):
         # Setup mock token provider
@@ -92,7 +92,7 @@ async def test_greeting_with_ai_project_client(self, mock_azure_credential, mock
         assert args["messages"][1]["content"] == "Hello"
 
     @pytest.mark.asyncio
-    @patch("plugins.chat_with_data_plugin.openai.AzureOpenAI")
+    @patch("helpers.azure_openai_helper.openai.AzureOpenAI")
     async def test_greeting_exception(self, mock_azure_openai, chat_plugin):
         # Setup mock to raise exception
         mock_client = MagicMock()
@@ -108,7 +108,7 @@ async def test_greeting_exception(self, mock_azure_openai, chat_plugin):
     @pytest.mark.asyncio
     @patch("plugins.chat_with_data_plugin.get_bearer_token_provider")
     @patch("plugins.chat_with_data_plugin.execute_sql_query")
-    @patch("plugins.chat_with_data_plugin.openai.AzureOpenAI")
+    @patch("helpers.azure_openai_helper.openai.AzureOpenAI")
     async def test_get_SQL_Response(self, mock_azure_openai, mock_execute_sql, mock_token_provider, chat_plugin):
 
         # Setup mocks
@@ -166,7 +166,7 @@ async def test_get_SQL_Response_with_ai_project_client(self, mock_azure_credenti
 
     @pytest.mark.asyncio
     @patch("plugins.chat_with_data_plugin.execute_sql_query")
-    @patch("plugins.chat_with_data_plugin.openai.AzureOpenAI")
+    @patch("helpers.azure_openai_helper.openai.AzureOpenAI")
     async def test_get_SQL_Response_exception(self, mock_azure_openai, mock_execute_sql, chat_plugin):
         # Setup mock to raise exception
         mock_client = MagicMock()
@@ -182,7 +182,7 @@ async def test_get_SQL_Response_exception(self, mock_azure_openai, mock_execute_
 
     @pytest.mark.asyncio
     @patch("plugins.chat_with_data_plugin.get_bearer_token_provider")
-    @patch("plugins.chat_with_data_plugin.openai.AzureOpenAI")
+    @patch("helpers.azure_openai_helper.openai.AzureOpenAI")
     async def test_get_answers_from_calltranscripts(self, mock_azure_openai, mock_token_provider, chat_plugin):
         # Setup mock
         mock_token_provider.return_value = lambda: "fake_token"
@@ -216,7 +216,7 @@ async def test_get_answers_from_calltranscripts(self, mock_azure_openai, mock_to
         assert args["extra_body"]["data_sources"][0]["type"] == "azure_search"
 
     @pytest.mark.asyncio
-    @patch("plugins.chat_with_data_plugin.openai.AzureOpenAI")
+    @patch("helpers.azure_openai_helper.openai.AzureOpenAI")
     async def test_get_answers_from_calltranscripts_with_citations(self, mock_azure_openai, chat_plugin):
         # Setup mock with citations in context
         mock_client = MagicMock()
@@ -251,7 +251,7 @@ async def test_get_answers_from_calltranscripts_with_citations(self, mock_azure_
         assert result.message.context.citations[0]["content"].endswith("...")
 
     @pytest.mark.asyncio
-    @patch("plugins.chat_with_data_plugin.openai.AzureOpenAI")
+    @patch("helpers.azure_openai_helper.openai.AzureOpenAI")
     async def test_get_answers_from_calltranscripts_exception(self, mock_azure_openai, chat_plugin):
         # Setup mock to raise exception
         mock_client = MagicMock()

src/tests/api/services/test_chat_service.py

@@ -31,7 +31,7 @@ def patched_imports(mock_format_stream, mock_openai, mock_agent_exception, mock_
          patch("services.chat_service.AzureAIAgentThread", mock_thread), \
          patch("services.chat_service.TruncationObject", mock_truncation), \
          patch("services.chat_service.AgentException", mock_agent_exception), \
-         patch("services.chat_service.openai.AzureOpenAI", mock_openai), \
+         patch("helpers.azure_openai_helper.openai.AzureOpenAI", mock_openai), \
          patch("services.chat_service.format_stream_response", mock_format_stream):
         from services.chat_service import ChatService, ExpCache
         return ChatService, ExpCache, {
@@ -173,7 +173,7 @@ def test_init(self, mock_config_class, mock_request):
         assert service.agent == mock_request.app.state.agent
         assert ChatService.thread_cache is not None
 
-    @patch('services.chat_service.openai.AzureOpenAI')
+    @patch('helpers.azure_openai_helper.openai.AzureOpenAI')
     def test_process_rag_response_success(self, mock_openai_class, chat_service):
         """Test successful RAG response processing."""
         # Setup mock OpenAI client
@@ -191,7 +191,7 @@ def test_process_rag_response_success(self, mock_openai_class, chat_service):
         assert result["data"]["labels"] == ["A", "B"]
         mock_client.chat.completions.create.assert_called_once()
 
-    @patch('services.chat_service.openai.AzureOpenAI')
+    @patch('helpers.azure_openai_helper.openai.AzureOpenAI')
     def test_process_rag_response_invalid_json(self, mock_openai_class, chat_service):
         """Test RAG response processing with invalid JSON."""
         # Setup mock OpenAI client
@@ -207,7 +207,7 @@ def test_process_rag_response_invalid_json(self, mock_openai_class, chat_service
         assert "error" in result
         assert result["error"] == "Chart could not be generated from this data. Please ask a different question."
 
-    @patch('services.chat_service.openai.AzureOpenAI')
+    @patch('helpers.azure_openai_helper.openai.AzureOpenAI')
     def test_process_rag_response_exception(self, mock_openai_class, chat_service):
         """Test RAG response processing with exception."""
         # Setup mock to raise exception