feat: add test framework tags (#30)

pytest_mergify/resources/pytest.py

@@ -0,0 +1,14 @@
+import pytest
+from opentelemetry.sdk.resources import Resource, ResourceDetector
+
+
+class PytestResourceDetector(ResourceDetector):
+    """Detects OpenTelemetry Resource attributes for Pytest."""
+
+    def detect(self) -> Resource:
+        return Resource(
+            {
+                "test.framework": "pytest",
+                "test.framework.version": pytest.__version__,
+            }
+        )

pytest_mergify/tracer.py

@@ -15,6 +15,7 @@
 import pytest_opentelemetry.instrumentation
 import pytest_mergify.resources.ci as resources_ci
 import pytest_mergify.resources.github_actions as resources_gha
+import pytest_mergify.resources.pytest as resources_pytest
 
 
 class InterceptingSpanProcessor(SpanProcessor):
@@ -100,6 +101,7 @@ def __post_init__(self) -> None:
             [
                 resources_ci.CIResourceDetector(),
                 resources_gha.GitHubActionsResourceDetector(),
+                resources_pytest.PytestResourceDetector(),
             ]
         )
 

tests/test_resources.py

@@ -5,7 +5,7 @@
 from tests import conftest
 
 
-def test_span_resources_attributes(
+def test_span_resources_attributes_ci(
     pytester: _pytest.pytester.Pytester,
     reconfigure_mergify_tracer: conftest.ReconfigureT,
 ) -> None:
@@ -28,6 +28,30 @@ def test_span(pytestconfig):
     result.assert_outcomes(passed=1)
 
 
+def test_span_resources_attributes_pytest(
+    pytester: _pytest.pytester.Pytester,
+    reconfigure_mergify_tracer: conftest.ReconfigureT,
+) -> None:
+    reconfigure_mergify_tracer({"_PYTEST_MERGIFY_TEST": "true"})
+    pytester.makepyfile(
+        """
+        import re
+
+        import pytest
+
+        def test_span(pytestconfig):
+            plugin = pytestconfig.pluginmanager.get_plugin("PytestMergify")
+            assert plugin is not None
+            assert plugin.mergify_tracer.exporter is not None
+            spans = plugin.mergify_tracer.exporter.get_finished_spans()
+            assert spans[0].resource.attributes["test.framework"] == "pytest"
+            assert re.match(r"\d\.", spans[0].resource.attributes["test.framework.version"])
+        """
+    )
+    result = pytester.runpytest_subprocess()
+    result.assert_outcomes(passed=1)
+
+
 def test_span_github_actions(
     pytester: _pytest.pytester.Pytester,
     reconfigure_mergify_tracer: conftest.ReconfigureT,