def _query_jdbc(self): """ Queries jdbc and returns a cursor to the results. """ jdbc = JdbcHook(jdbc_conn_id=self.jdbc_conn_id) conn = jdbc.get_conn() cursor = conn.cursor() self.log.info('Querying SQL: %s', self.sql) cursor.execute(self.sql) return cursor
def test_jdbc_conn_connection(self, jdbc_mock): jdbc_hook = JdbcHook() jdbc_conn = jdbc_hook.get_conn() self.assertTrue(jdbc_mock.called) self.assertIsInstance(jdbc_conn, Mock) self.assertEqual(jdbc_conn.name, jdbc_mock.return_value.name)
def test_jdbc_conn_get_autocommit(self, _): jdbc_hook = JdbcHook() jdbc_conn = jdbc_hook.get_conn() jdbc_hook.get_autocommit(jdbc_conn) jdbc_conn.jconn.getAutoCommit.assert_called_once_with()