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)