yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2026-01-08 13:06:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Fix datadog tracer tests (#9253)

Browse Source 
* Fix tests

* Make tests work in old versions of datadog
This commit is contained in:
Sebastian Husch Lee 2025-04-17 09:26:27 +02:00 committed by GitHub
parent 5154d1c7eb
commit c5684b64a6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
test/tracing/test_datadog.py
View File

@ -7,6 +7,7 @@ from typing import List, Dict
from ddtrace.trace import Span as ddSpan
from ddtrace.trace import Tracer as ddTracer
from ddtrace._trace.processor import SpanAggregator
import pytest
from _pytest.capture import CaptureFixture
from _pytest.monkeypatch import MonkeyPatch
@ -14,10 +15,18 @@ from _pytest.monkeypatch import MonkeyPatch
from haystack.tracing.datadog import DatadogTracer
def safe_patch(monkeypatch, target, attr_name, replacement):
if hasattr(target, attr_name):
attr = getattr(target, attr_name)
name = attr.__name__ if callable(attr) and hasattr(attr, "__name__") else attr_name
monkeypatch.setattr(target, name, replacement)
@pytest.fixture()
def datadog_tracer(monkeypatch: MonkeyPatch) -> ddTracer:
# For the purpose of the tests we want to use the log writer
monkeypatch.setattr(ddTracer, ddTracer._use_log_writer.__name__, lambda *_: True)
safe_patch(monkeypatch, SpanAggregator, "_use_log_writer", lambda *_: True)
safe_patch(monkeypatch, ddTracer, "_use_log_writer", lambda *_: True)
tracer = ddTracer()