Skip to content

Commit

Permalink
973 opentelemetry instrumentation psycopg2 (#2840)
Browse files Browse the repository at this point in the history
  • Loading branch information
kaushik95 authored Sep 4, 2024
1 parent afb7f54 commit f1d9109
Showing 1 changed file with 12 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
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
Expand Down Expand Up @@ -269,3 +270,14 @@ def test_sqlcommenter_disabled(self, event_mocked):
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)

0 comments on commit f1d9109

Please sign in to comment.