opensource/opentelemetry-python-contrib
1
0
Fork 0
mirror of https://github.com/open-telemetry/opentelemetry-python-contrib.git synced 2025-08-02 02:52:18 +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 psycopg2
import opentelemetry.instrumentation.psycopg2 import opentelemetry.instrumentation.psycopg2
from opentelemetry import trace
from opentelemetry.instrumentation.psycopg2 import Psycopg2Instrumentor from opentelemetry.instrumentation.psycopg2 import Psycopg2Instrumentor
from opentelemetry.sdk import resources from opentelemetry.sdk import resources
from opentelemetry.test.test_base import TestBase from opentelemetry.test.test_base import TestBase
@ -269,3 +270,14 @@ class TestPostgresqlIntegration(TestBase):
cursor.execute(query) cursor.execute(query)
kwargs = event_mocked.call_args[1] kwargs = event_mocked.call_args[1]
self.assertEqual(kwargs["enable_commenter"], False) 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)