def test_query_wrong_oracle(self, _ora_connect):
cursor = mock.MagicMock()
cursor.execute.side_effect = ORA_Error('test error')
with self.assertRaises(QueryError):
_ora_connect.return_value.__enter__\
.return_value.cursor.return_value.__enter__\
.return_value = cursor
oracle_access(self.query, self.domain_id)
def test_execute_oracle(self, _ora_connect):
cursor = mock.MagicMock()
_ora_connect.return_value.__enter__\
.return_value.cursor.return_value.__enter__\
.return_value = cursor
oracle_access(self.query, (self.domain_id, ))
cursor.execute.assert_called_once_with(self.query, (self.domain_id, ))
cursor.fetchall.assert_called_once_with()
def execute(cls, **kwargs):
return oracle_access(cls.query(), kwargs)
def test_connect_oracle(self, _ora_connect):
oracle_access(self.query, self.domain_id)
_ora_connect.assert_called_once_with(user=settings.ORA_USER,
password=settings.ORA_PASS,
dsn=settings.ORA_HOST)
def execute(cls, **kwargs):
oracle_data = oracle_access(cls.query(cls.query_oracle), kwargs)
impala_data = impala_execute(cls.query(cls.query_impala), kwargs)
return oracle_data + impala_data