def delete(cls, instance):
node_ids = [
_id for (_id, ) in db().query(models.Node.id).filter_by(
cluster_id=instance.id).order_by(models.Node.id)
]
fire_callback_on_node_collection_delete(node_ids)
fire_callback_on_cluster_delete(instance)
super(Cluster, cls).delete(instance)
def delete(cls, instance):
node_ids = [
_id for (_id,) in
db().query(models.Node.id).
filter_by(cluster_id=instance.id).
order_by(models.Node.id)]
fire_callback_on_node_collection_delete(node_ids)
fire_callback_on_cluster_delete(instance)
super(Cluster, cls).delete(instance)
def test_fire_callback_on_cluster_deletion(self, get_m):
cluster = mock.MagicMock()
fire_callback_on_cluster_delete(cluster)
for ext in get_m.return_value:
ext.on_cluster_delete.assert_called_once_with(cluster)
