def create(self, **params):
"""
params - dictionary
"""
c = Cluster(**params)
c.save(force_insert=True)
return c
def test_info_cluster(self):
cluster_id = "unit_test_cluster"
cluster1 = Cluster(cluster_id=cluster_id)
cluster1.save(force_insert=True)
resp = self.app.get("/cluster/%s/" % (cluster_id))
self.assertEquals(resp.status_code, 200)
cluster = json.loads(resp.data)
self.assertEquals(cluster["cluster_id"], cluster_id)
def test_create_already_exists(self):
cluster_id = "unit_test_cluster"
cluster1 = Cluster(cluster_id=cluster_id)
cluster1.save(force_insert=True)
data = {"cluster_id": cluster_id}
json_data = json.dumps(data)
resp = self.app.post('/cluster/', data=json_data, content_type='application/json')
self.assertEquals(resp.status_code, 409)
def test_update(self):
cluster_id = "unit_test_cluster"
cluster1 = Cluster(cluster_id=cluster_id, cdses=[])
cluster1.save(force_insert=True)
cluster1.reload()
self.assertEquals(len(cluster1["cdses"]), 0)
data = {"cluster_id": cluster_id, "cdses": ["unit_test_cds_1"]}
json_data = json.dumps(data)
resp = self.app.put("/cluster/%s/" % (cluster_id), data=json_data, content_type='application/json')
self.assertEquals(resp.status_code, 200)
# Using json_util.loads so it can parse the {"$date":value} formats for created_at/updated_at
cluster_updated = json_util.loads(resp.data)
self.assertEquals(cluster_updated["cdses"], ['unit_test_cds_1'])
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 test_list_clusters(self):
resp = self.app.get('/cluster/')
items = json.loads(resp.data)
self.assertEquals(len(items), 0)
cluster_id_1 = "unit_test_cluster_1"
cluster_id_2 = "unit_test_cluster_2"
cluster1 = Cluster(cluster_id=cluster_id_1)
cluster2 = Cluster(cluster_id=cluster_id_2)
cluster1.save(force_insert=True)
cluster2.save(force_insert=True)
resp = self.app.get('/cluster/')
self.assertEquals(resp.status_code, 200)
clusters = json.loads(resp.data)
self.assertEquals(len(clusters), 2)
for c in clusters:
self.assertIn(c["cluster_id"], [cluster_id_1, cluster_id_2])
def drop_collections(self):
# Remember to drop the collections instead of the entire DB
# Dropping the entire DB can mess up indexes
# Removing the unique index leading to odd behavior in unit tests
CDS.drop_collection()
Cluster.drop_collection()
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()
