def test_unwrap_connect(self, mock_dbapi):
dbapi.wrap_connect(self.tracer, mock_dbapi, "connect", "-")
connection = mock_dbapi.connect()
self.assertEqual(mock_dbapi.connect.call_count, 1)
dbapi.unwrap_connect(mock_dbapi, "connect")
connection = mock_dbapi.connect()
self.assertEqual(mock_dbapi.connect.call_count, 2)
self.assertIsInstance(connection, mock.Mock)
def _uninstrument(self, **kwargs):
""""Disable Psycopg2 instrumentation"""
dbapi.unwrap_connect(psycopg2, "connect")
def _uninstrument(self, **kwargs):
""""Disable MySQL instrumentation"""
dbapi.unwrap_connect(mysql.connector, "connect")
def _uninstrument(self, **kwargs):
""""Disable SQLite3 instrumentation"""
dbapi.unwrap_connect(sqlite3, "connect")
def _uninstrument(self, **kwargs):
""""Disable PyMySQL instrumentation"""
dbapi.unwrap_connect(pymysql, "connect")
def _uninstrument(self, **kwargs):
""" "Disable SQLite3 instrumentation"""
for module in self._TO_WRAP:
dbapi.unwrap_connect(module, "connect")
