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")