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