def update(self, cluster_id, **params): c = self.get(cluster_id) if not c: raise MissingResource("No Cluster with cluster_id: '%s'" % cluster_id) for k in params.keys(): c[k] = params[k] c.save() cluster = Cluster.objects(cluster_id=cluster_id) return cluster[0]
def test_simple_create(self): cid = "cds_cluster1" c = self.cds_cluster_manager.create(cluster_id=cid) self.assertIsNotNone(c) from pulp_cds.api.models.cds_cluster import Cluster found = Cluster.objects(cluster_id=cid) self.assertEquals(found[0], c)
def get(self, cluster_id): c = Cluster.objects(cluster_id=cluster_id) if len(c) < 1: return None return c[0]
def get_all(self): return Cluster.objects()