fix pymssql entry_point for auto-instrumentation (#3214)

Signed-off-by: emdneto <9735060+emdneto@users.noreply.github.com>
Co-authored-by: Riccardo Magliocchetti <riccardo.magliocchetti@gmail.com>
This commit is contained in:
Emídio Neto
2025-01-28 11:05:19 -03:00
committed by GitHub
parent eabab7d0ad
commit dd68241907
2 changed files with 14 additions and 1 deletions

View File

@ -37,7 +37,7 @@ instruments = [
] ]
[project.entry-points.opentelemetry_instrumentor] [project.entry-points.opentelemetry_instrumentor]
pymssql = "opentelemetry.instrumentation.pymssql:pymssqlInstrumentor" pymssql = "opentelemetry.instrumentation.pymssql:PyMSSQLInstrumentor"
[project.urls] [project.urls]
Homepage = "https://github.com/open-telemetry/opentelemetry-python-contrib/tree/main/instrumentation/opentelemetry-instrumentation-pymssql" Homepage = "https://github.com/open-telemetry/opentelemetry-python-contrib/tree/main/instrumentation/opentelemetry-instrumentation-pymssql"

View File

@ -20,6 +20,7 @@ import opentelemetry.instrumentation.pymssql
from opentelemetry.instrumentation.pymssql import PyMSSQLInstrumentor from opentelemetry.instrumentation.pymssql import PyMSSQLInstrumentor
from opentelemetry.sdk import resources from opentelemetry.sdk import resources
from opentelemetry.test.test_base import TestBase from opentelemetry.test.test_base import TestBase
from opentelemetry.util._importlib_metadata import entry_points
def mock_connect(*args, **kwargs): def mock_connect(*args, **kwargs):
@ -182,3 +183,15 @@ class TestPyMSSQLIntegration(TestBase):
spans_list = self.memory_exporter.get_finished_spans() spans_list = self.memory_exporter.get_finished_spans()
self.assertEqual(len(spans_list), 1) self.assertEqual(len(spans_list), 1)
def test_load_entry_point(self):
self.assertIs(
next(
iter(
entry_points(
group="opentelemetry_instrumentor", name="pymssql"
)
)
).load(),
PyMSSQLInstrumentor,
)