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)