Add more _base_client attributes

pquentin · pquentin · commit 428203a1ba34 · 2025-03-25T03:02:43.000+04:00
diff --git a/test_elasticsearch/test_dsl/test_connections.py b/test_elasticsearch/test_dsl/test_connections.py
@@ -123,7 +123,7 @@ def test_connection_has_correct_user_agent() -> None:
     c.create_connection("testing", hosts=["https://es.com:9200"])
     assert (
         c.get_connection("testing")
-        ._headers["user-agent"]
+        ._base_client._headers["user-agent"]
         .startswith("elasticsearch-dsl-py/")
     )
 
diff --git a/test_elasticsearch/test_server/test_otel.py b/test_elasticsearch/test_server/test_otel.py
@@ -34,7 +34,7 @@
 
 def test_otel_end_to_end(sync_client):
     tracer, memory_exporter = setup_tracing()
-    sync_client._otel.tracer = tracer
+    sync_client._base_client._otel.tracer = tracer
 
     resp = sync_client.search(index="logs-*", query={"match_all": {}})
     assert resp.meta.status == 200

Original file line number	Diff line number	Diff line change
`@@ -123,7 +123,7 @@ def test_connection_has_correct_user_agent() -> None:`
`123`	`123`	`c.create_connection("testing", hosts=["https://es.com:9200"])`
`124`	`124`	`assert (`
`125`	`125`	`c.get_connection("testing")`
`126`		`- ._headers["user-agent"]`
	`126`	`+ ._base_client._headers["user-agent"]`
`127`	`127`	`.startswith("elasticsearch-dsl-py/")`
`128`	`128`	`)`
`129`	`129`