Add test for add_span_processor hook

Author: HakonSohoel

tests/ert/unit_tests/plugins/dummy_plugins.py

@@ -1,4 +1,7 @@
 import logging
+from io import StringIO
+
+from opentelemetry.sdk.trace.export import BatchSpanProcessor, ConsoleSpanExporter
 
 from ert import ForwardModelStepPlugin, plugin
 
@@ -76,6 +79,14 @@ def add_log_handle_to_root():
     return fh
 
 
+span_output = StringIO()
+
+
+@plugin(name="dummy")
+def add_span_processor():
+    return BatchSpanProcessor(ConsoleSpanExporter(out=span_output))
+
+
 class DummyFMStep(ForwardModelStepPlugin):
     def __init__(self):
         super().__init__(name="DummyForwardModel", command=["touch", "dummy.out"])

tests/ert/unit_tests/plugins/test_plugin_manager.py

@@ -1,7 +1,10 @@
+import json
 import logging
 import tempfile
 from unittest.mock import Mock
 
+from opentelemetry.sdk.trace import TracerProvider
+
 import ert.plugins.hook_implementations
 from ert.plugins import ErtPluginManager
 from tests.ert.unit_tests.plugins import dummy_plugins
@@ -115,6 +118,18 @@ def test_add_logging_handle(tmpdir):
         assert "I should write this to spam.log" in result
 
 
+def test_add_span_processor():
+    pm = ErtPluginManager(plugins=[dummy_plugins])
+    tracer_provider = TracerProvider()
+    tracer = tracer_provider.get_tracer("ert.tests")
+    pm.add_span_processor_to_trace_provider(tracer_provider)
+    with tracer.start_as_current_span("test_span"):
+        print("test_span")
+    tracer_provider.force_flush()
+    span_info = json.loads(dummy_plugins.span_output.getvalue())
+    assert span_info["name"] == "test_span"
+
+
 def test_that_forward_model_step_is_registered(tmpdir):
     with tmpdir.as_cwd():
         pm = ErtPluginManager(plugins=[dummy_plugins])