opensource/opentelemetry-python-contrib
1
0
Fork 0
mirror of https://github.com/open-telemetry/opentelemetry-python-contrib.git synced 2025-08-01 06:33:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

973 opentelemetry instrumentation psycopg2 (#2840)

Browse Source
This commit is contained in:
kaushik95
2024-09-04 22:09:17 +05:30
committed by GitHub
parent afb7f54b50
commit f1d91097d8
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
instrumentation/opentelemetry-instrumentation-psycopg2/tests/test_psycopg2_integration.py
View File

@ -18,6 +18,7 @@ from unittest import mock
import psycopg2
import opentelemetry.instrumentation.psycopg2
from opentelemetry import trace
from opentelemetry.instrumentation.psycopg2 import Psycopg2Instrumentor
from opentelemetry.sdk import resources
from opentelemetry.test.test_base import TestBase
@ -269,3 +270,14 @@ class TestPostgresqlIntegration(TestBase):
cursor.execute(query)
kwargs = event_mocked.call_args[1]
self.assertEqual(kwargs["enable_commenter"], False)
def test_no_op_tracer_provider(self):
Psycopg2Instrumentor().instrument(
tracer_provider=trace.NoOpTracerProvider()
)
cnx = psycopg2.connect(database="test")
cursor = cnx.cursor()
query = "SELECT * FROM test"
cursor.execute(query)
spans_list = self.memory_exporter.get_finished_spans()
self.assertEqual(len(spans_list), 0)