Skip to content

Commit 1cf2238

Browse files
Mv test
1 parent 11d4f07 commit 1cf2238

File tree

1 file changed

+15
-16
lines changed

1 file changed

+15
-16
lines changed

instrumentation/opentelemetry-instrumentation-mysql/tests/test_mysql_integration.py

Lines changed: 15 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)