Added unit test

Should have been part of prev commit.

tests/test_web/test_comments_edit.py

@@ -0,0 +1,57 @@
+import json
+
+from bson import ObjectId
+
+from pillar.tests import AbstractPillarTest
+from pillar.tests import common_test_data as ctd
+
+
+class CommentEditTest(AbstractPillarTest):
+    def setUp(self, **kwargs):
+        super().setUp(**kwargs)
+
+        self.pid, self.project = self.ensure_project_exists()
+        self.uid = self.create_user(groups=[ctd.EXAMPLE_ADMIN_GROUP_ID])
+        self.create_valid_auth_token(self.uid, 'token')
+
+        self.node_id = self.create_node({
+            '_id': ObjectId('572761099837730efe8e120d'),
+            'description': 'This is an asset without file',
+            'node_type': 'asset',
+            'user': self.uid,
+            'properties': {
+                'status': 'published',
+                'content_type': 'image',
+            },
+            'name': 'Image test',
+            'project': self.pid,
+        })
+
+    def test_edit_comment(self):
+        from pillar import auth
+        from pillar.web.nodes.custom import comments
+
+        # Create the comment
+        with self.app.test_request_context(method='POST', data={
+            'content': 'My first comment',
+            'parent_id': str(self.node_id),
+        }):
+            auth.login_user('token', load_from_db=True)
+            resp, status = comments.comments_create()
+
+        self.assertEqual(201, status)
+        payload = json.loads(resp.data)
+        comment_id = payload['node_id']
+
+        # Edit the comment
+        with self.app.test_request_context(method='POST', data={
+            'content': 'Edited comment',
+        }):
+            auth.login_user('token', load_from_db=True)
+            resp = comments.comment_edit(comment_id)
+
+        self.assertEqual(200, resp.status_code)
+        payload = json.loads(resp.data)
+        self.assertEqual('success', payload['status'])
+        self.assertEqual('Edited comment', payload['data']['content'])
+        self.assertEqual('<p>Edited comment</p>\n', payload['data']['content_html'])