def test_delete_batch_fail(self, mock):
mock.register_uri('DELETE',
'//livy:8998/batches/{}'.format(BATCH_ID),
json={},
status_code=400,
reason='ERROR')
hook = LivyHook()
with self.assertRaises(AirflowException):
hook.delete_batch(BATCH_ID)
def test_delete_batch_fail(self, mock):
mock.register_uri('DELETE',
f'//livy:8998/batches/{BATCH_ID}',
json={},
status_code=400,
reason='ERROR')
hook = LivyHook()
with pytest.raises(AirflowException):
hook.delete_batch(BATCH_ID)
def test_delete_batch_validation(self, mock):
mock.register_uri('DELETE',
'//livy:8998/batches/{}'.format(BATCH_ID),
json={'id': BATCH_ID},
status_code=200)
hook = LivyHook()
with self.subTest('get_batch'):
hook.delete_batch(BATCH_ID)
for val in [None, 'one', {'a': 'b'}]:
with self.subTest('get_batch {}'.format(val)):
with self.assertRaises(TypeError):
hook.delete_batch(val)
