Fix tests

Author: Amnah199

test/components/embedders/test_azure_document_embedder.py

@@ -13,6 +13,7 @@
 from haystack.components.embedders import AzureOpenAIDocumentEmbedder
 from haystack.utils.azure import default_azure_ad_token_provider
 from unittest.mock import Mock, patch
+from haystack.utils.http_client import init_http_client
 
 
 class TestAzureOpenAIDocumentEmbedder:
@@ -220,14 +221,14 @@ def test_init_http_client(self, monkeypatch):
         monkeypatch.setenv("AZURE_OPENAI_ENDPOINT", "https://test.openai.azure.com")
 
         embedder = AzureOpenAIDocumentEmbedder()
-        client = embedder._init_http_client()
+        client = init_http_client(embedder.http_client_kwargs, async_client=False)
         assert client is None
 
         embedder.http_client_kwargs = {"proxy": "http://example.com:3128"}
-        client = embedder._init_http_client(async_client=False)
+        client = init_http_client(embedder.http_client_kwargs, async_client=False)
         assert isinstance(client, httpx.Client)
 
-        client = embedder._init_http_client(async_client=True)
+        client = init_http_client(embedder.http_client_kwargs, async_client=True)
         assert isinstance(client, httpx.AsyncClient)
 
     def test_http_client_kwargs_type_validation(self, monkeypatch):

test/components/embedders/test_azure_text_embedder.py

@@ -8,6 +8,7 @@
 
 from haystack.components.embedders import AzureOpenAITextEmbedder
 from haystack.utils.azure import default_azure_ad_token_provider
+from haystack.utils.http_client import init_http_client
 
 
 class TestAzureOpenAITextEmbedder:
@@ -174,14 +175,14 @@ def test_init_http_client(self, monkeypatch):
         monkeypatch.setenv("AZURE_OPENAI_ENDPOINT", "https://test.openai.azure.com")
 
         embedder = AzureOpenAITextEmbedder()
-        client = embedder._init_http_client()
+        client = init_http_client(embedder.http_client_kwargs, async_client=False)
         assert client is None
 
         embedder.http_client_kwargs = {"proxy": "http://example.com:3128"}
-        client = embedder._init_http_client(async_client=False)
+        client = init_http_client(embedder.http_client_kwargs, async_client=False)
         assert isinstance(client, httpx.Client)
 
-        client = embedder._init_http_client(async_client=True)
+        client = init_http_client(embedder.http_client_kwargs, async_client=True)
         assert isinstance(client, httpx.AsyncClient)
 
     def test_http_client_kwargs_type_validation(self, monkeypatch):

test/components/generators/chat/test_azure.py

@@ -110,6 +110,7 @@ def test_to_dict_default(self, monkeypatch):
                 "tools": None,
                 "tools_strict": False,
                 "azure_ad_token_provider": None,
+                "http_client_kwargs": None,
             },
         }
 
@@ -124,6 +125,7 @@ def test_to_dict_with_parameters(self, monkeypatch):
             max_retries=10,
             generation_kwargs={"max_tokens": 10, "some_test_param": "test-params"},
             azure_ad_token_provider=default_azure_ad_token_provider,
+            http_client_kwargs={"proxy": "http://localhost:8080"},
         )
         data = component.to_dict()
         assert data == {
@@ -143,6 +145,7 @@ def test_to_dict_with_parameters(self, monkeypatch):
                 "tools_strict": False,
                 "default_headers": {},
                 "azure_ad_token_provider": "haystack.utils.azure.default_azure_ad_token_provider",
+                "http_client_kwargs": {"proxy": "http://localhost:8080"},
             },
         }