Tag all tests using spans (#1482)

Author: crusaderky

tests/conftest.py

@@ -35,7 +35,13 @@
 from benchmark_schema import TestRun
 from plugins import Durations
 
-logger = logging.getLogger("coiled-runtime")
+try:
+    from distributed.spans import span as span_ctx
+except ImportError:  # dask <2023.6.0
+    from contextlib import nullcontext as span_ctx
+
+
+logger = logging.getLogger("benchmarks")
 logger.setLevel(logging.INFO)
 
 coiled_logger = logging.getLogger("coiled")
@@ -401,13 +407,20 @@ def _get_cluster_info(cluster):
     yield _get_cluster_info
 
 
+@pytest.fixture(scope="function")
+def span(request):
+    with span_ctx(request.node.name):
+        yield
+
+
 @pytest.fixture(scope="function")
 def benchmark_all(
     benchmark_memory,
     benchmark_task_durations,
     benchmark_coiled_prometheus,
     get_cluster_info,
     benchmark_time,
+    span,
 ):
     """Benchmark all available metrics and extracts cluster information
 

tests/tpch/conftest.py

@@ -206,6 +206,7 @@ def client(
     testrun_uid,
     cluster_kwargs,
     benchmark_time,
+    span,
     restart,
     scale,
     local,

tests/tpch/test_correctness.py

@@ -70,6 +70,7 @@ def client(
     testrun_uid,
     cluster_kwargs,
     benchmark_time,
+    span,
     restart,
     local,
     query,