Skip to content

Commit

Permalink
Issue-973 NoOpTracerProvider test case for psycopg2
Browse files Browse the repository at this point in the history
Signed-off-by: Kowshik BV <kowshik.bv@fmr.com>
  • Loading branch information
kaushik95 committed Jul 25, 2024
1 parent cc52bd2 commit 8bff533
Showing 1 changed file with 13 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 @@ -260,6 +261,18 @@ def test_sqlcommenter_enabled(self, event_mocked):
kwargs = event_mocked.call_args[1]
self.assertEqual(kwargs["enable_commenter"], True)

@mock.patch("opentelemetry.instrumentation.dbapi.wrap_connect")
def test_sqlcommenter_enabled_no_op_tracer(self, event_mocked):
tracer_provider = trace.NoOpTracerProvider()
trace.set_tracer_provider(tracer_provider)
cnx = psycopg2.connect(database="test")
Psycopg2Instrumentor().instrument(tracer_provider=tracer_provider, enable_commenter=True)
query = "SELECT * FROM test"
cursor = cnx.cursor()
cursor.execute(query)
kwargs = event_mocked.call_args[1]
self.assertEqual(kwargs["enable_commenter"], True)

@mock.patch("opentelemetry.instrumentation.dbapi.wrap_connect")
def test_sqlcommenter_disabled(self, event_mocked):
cnx = psycopg2.connect(database="test")
Expand Down

0 comments on commit 8bff533

Please sign in to comment.