Fix otel tests

Author: pquentin

test_elasticsearch/test_otel.py

@@ -112,11 +112,11 @@ def test_forward_otel_context_to_subthreads(
 ):
     tracer, memory_exporter = setup_tracing()
     es_client = Elasticsearch("http://localhost:9200")
-    es_client._otel = OpenTelemetry(enabled=True, tracer=tracer)
+    es_client._base_client._otel = OpenTelemetry(enabled=True, tracer=tracer)
 
     _call_bulk_mock.return_value = mock.Mock()
     actions = ({"x": i} for i in range(100))
     list(helpers.parallel_bulk(es_client, actions, chunk_size=4))
     # Ensures that the OTEL context has been forwarded to all chunks
-    assert es_client._otel.helpers_span.call_count == 1
-    assert es_client._otel.use_span.call_count == 25
+    assert es_client._base_client._otel.helpers_span.call_count == 1
+    assert es_client._base_client._otel.use_span.call_count == 25

test_elasticsearch/test_server/test_otel.py

@@ -61,7 +61,7 @@ def test_otel_bulk(sync_client, elasticsearch_url, bulk_helper_name):
 
     # Create a new client with our tracer
     sync_client = sync_client.options()
-    sync_client._otel.tracer = tracer
+    sync_client._base_client._otel.tracer = tracer
     # "Disable" options to keep our custom tracer
     sync_client.options = lambda: sync_client