add CRUD methods for collections

tbarbugli · tbarbugli · commit ca6674f5885e · 2018-11-30T12:27:48.000+01:00
diff --git a/stream/collections.py b/stream/collections.py
@@ -33,7 +33,7 @@ def upsert(self, collection_name, data):
 
         data_json = {collection_name: data}
 
-        response = self.client.post('meta/', service_name='api',
+        response = self.client.post('collections/', service_name='api',
                                     signature=self.token, data={'data': data_json})
         return response
 
@@ -59,12 +59,12 @@ def select(self, collection_name, ids):
             foreign_ids.append('%s:%s' % (collection_name, ids[i]))
         foreign_ids = ','.join(foreign_ids)
 
-        response = self.client.get('meta/', service_name='api', params={'foreign_ids': foreign_ids},
+        response = self.client.get('collections/', service_name='api', params={'foreign_ids': foreign_ids},
                                    signature=self.token)
 
         return response
 
-    def delete(self, collection_name, ids):
+    def delete_many(self, collection_name, ids):
         """
         Delete data from meta.
         :param collection_name: Collection Name i.e 'user'
@@ -82,7 +82,37 @@ def delete(self, collection_name, ids):
 
         params = {'collection_name': collection_name, 'ids': ids}
 
-        response = self.client.delete('meta/', service_name='api', params=params,
+        response = self.client.delete('collections/', service_name='api', params=params,
                                       signature=self.token)
 
         return response
+
+    def add(self, collection_name, data, id=None, user_id=None):
+        payload = dict(
+            id=id, data=data, user_id=user_id,
+        )
+        return self.client.post(
+            "collections/%s" % collection_name,
+            service_name="api",
+            signature=self.token,
+            data=payload,
+        )
+
+    def get(self, collection_name, id):
+        return self.client.get(
+            "collections/%s/%s" % (collection_name, id), service_name="api", signature=self.token
+        )
+
+    def update(self, collection_name, id, data=None):
+        payload = dict(data=data)
+        return self.client.put(
+            "collections/%s/%s" % (collection_name, id),
+            service_name="api",
+            signature=self.token,
+            data=payload,
+        )
+
+    def delete(self, collection_name, id):
+        return self.client.delete(
+            "collections/%s/%s" % (collection_name, id), service_name="api", signature=self.token
+        )
diff --git a/stream/tests/test_client.py b/stream/tests/test_client.py
@@ -1303,3 +1303,33 @@ def test_user_update(self):
     def test_user_delete(self):
         response = self.c.users.add(str(uuid1()))
         self.c.users.delete(response["id"])
+
+    def test_collections_add(self):
+        self.c.collections.add("items", {"data": 1}, id=str(uuid1()), user_id="tom")
+
+    def test_collections_add_twice(self):
+        id = str(uuid1())
+        r1 = self.c.collections.add("items", {"data": 1}, id=id)
+        r2 = self.c.collections.add("items", {"data": 2}, id=id)
+        entry = self.c.collections.get("items", id)
+        self.assertEqual(entry["data"], r2["data"])
+        self.assertEqual(r1["created_at"], r2["created_at"])
+        self.assertNotEqual(r1["updated_at"], r2["updated_at"])
+
+    def test_collections_get(self):
+        response = self.c.collections.add("items", {"data": 1}, id=str(uuid1()))
+        entry = self.c.collections.get("items", response["id"])
+        self.assertEqual(entry["data"], {"data": 1})
+        self.assertIn("created_at", entry)
+        self.assertIn("updated_at", entry)
+        self.assertIn("id", entry)
+
+    def test_collections_update(self):
+        response = self.c.collections.add("items", {"data": 1}, str(uuid1()))
+        self.c.collections.update("items", response["id"], data={"changed": True})
+        entry = self.c.collections.get("items", response["id"])
+        self.assertEqual(entry["data"], {"changed": True})
+
+    def test_collections_delete(self):
+        response = self.c.collections.add("items", {"data": 1}, str(uuid1()))
+        self.c.collections.delete("items", response["id"])
