@@ -62,6 +62,21 @@ def test_instrumentor(self, mock_connect):
6262 spans_list = self .memory_exporter .get_finished_spans ()
6363 self .assertEqual (len (spans_list ), 1 )
6464
65+ @mock .patch ("opentelemetry.instrumentation.dbapi.wrap_connect" )
66+ @mock .patch ("mysql.connector.connect" )
67+ def test_instrumentor_sqlcomment_enabled (
68+ self ,
69+ mock_connect ,
70+ mock_wrap_connect ,
71+ ):
72+ MySQLInstrumentor ().instrument (
73+ enable_commenter = True ,
74+ commenter_options = {"foo" : True },
75+ )
76+ kwargs = mock_wrap_connect .call_args [1 ]
77+ self .assertEqual (kwargs ["enable_commenter" ], True )
78+ self .assertEqual (kwargs ["commenter_options" ], {"foo" : True })
79+
6580 @mock .patch ("mysql.connector.connect" )
6681 # pylint: disable=unused-argument
6782 def test_custom_tracer_provider (self , mock_connect ):
@@ -122,22 +137,6 @@ def test_instrument_connection_enable_commenter(
122137 self .assertEqual (kwargs ["enable_commenter" ], True )
123138 self .assertEqual (kwargs ["commenter_options" ], {"foo" : True })
124139
125- @mock .patch ("opentelemetry.instrumentation.dbapi.wrap_connect" )
126- @mock .patch ("mysql.connector.connect" )
127- # pylint: disable=unused-argument
128- def test__instrument_enable_commenter (
129- self ,
130- mock_connect ,
131- mock_wrap_connect ,
132- ):
133- MySQLInstrumentor ()._instrument (
134- enable_commenter = True ,
135- commenter_options = {"foo" : True },
136- )
137- kwargs = mock_wrap_connect .call_args [1 ]
138- self .assertEqual (kwargs ["enable_commenter" ], True )
139- self .assertEqual (kwargs ["commenter_options" ], {"foo" : True })
140-
141140 @mock .patch ("mysql.connector.connect" )
142141 # pylint: disable=unused-argument
143142 def test_uninstrument_connection (self , mock_connect ):
0 commit comments