def main():
print("Pools ********************")
created_pools = set()
for pool in pools:
name = pool["name"]
print(f"Create: {name}")
create_pool(**pool)
created_pools.add(name)
for pool in get_pools():
name = pool.pool
if name == default_pool_name or name in created_pools:
continue
if name not in created_pools:
print(f"Delete: {name}")
delete_pool(name)
return
def delete_pool(name):
"""Delete pool."""
try:
pool = pool_api.delete_pool(name=name)
except AirflowException as err:
log.error(err)
response = jsonify(error="{}".format(err))
response.status_code = err.status_code
return response
else:
return jsonify(pool.to_json())
def delete_pool(name):
"""Delete pool."""
try:
pool = pool_api.delete_pool(name=name)
except AirflowException as e:
_log.error(e)
response = jsonify(error="{}".format(e))
response.status_code = getattr(e, 'status', 500)
return response
else:
return jsonify(pool.to_json())
def test_delete_default_pool_not_allowed(self):
with pytest.raises(AirflowBadRequest,
match="^default_pool cannot be deleted$"):
pool_api.delete_pool(Pool.DEFAULT_POOL_NAME)
def test_delete_pool_bad_name(self):
for name in ('', ' '):
with pytest.raises(AirflowBadRequest,
match="^Pool name shouldn't be empty$"):
pool_api.delete_pool(name=name)
def test_delete_pool_non_existing(self):
with pytest.raises(pool_api.PoolNotFound,
match="^Pool 'test' doesn't exist$"):
pool_api.delete_pool(name='test')
def test_delete_pool(self):
pool = pool_api.delete_pool(name=self.pools[-1].pool)
assert pool.pool == self.pools[-1].pool
with create_session() as session:
assert session.query(
models.Pool).count() == self.TOTAL_POOL_COUNT - 1
def delete_pool(self, name):
the_pool = pool.delete_pool(name=name)
return the_pool.pool, the_pool.slots, the_pool.description
def test_delete_default_pool_not_allowed(self):
with self.assertRaisesRegex(AirflowBadRequest,
"^default_pool cannot be deleted$"):
pool_api.delete_pool(Pool.DEFAULT_POOL_NAME)
def test_delete_pool(self):
pool = pool_api.delete_pool(name=self.pools[-1].pool)
self.assertEqual(pool.pool, self.pools[-1].pool)
with create_session() as session:
self.assertEqual(
session.query(models.Pool).count(), self.TOTAL_POOL_COUNT - 1)
def delete_pool(self, name):
p = pool.delete_pool(name=name)
return p.pool, p.slots, p.description
def test_delete_pool(self):
pool = pool_api.delete_pool(name=self.pools[0].pool,
session=self.session)
self.assertEqual(pool.pool, self.pools[0].pool)
self.assertEqual(self.session.query(models.Pool).count(), 1)
def test_delete_pool(self):
pool = pool_api.delete_pool(name=self.pools[0].pool,
session=self.session)
self.assertEqual(pool.pool, self.pools[0].pool)
self.assertEqual(self.session.query(models.Pool).count(), 1)
def delete_pool(self, name):
p = pool.delete_pool(name=name)
return p.pool, p.slots, p.description
