From fe7e078f8b560cb78beb27121e05a6fed8d72ee0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sybren=20A=2E=20St=C3=BCvel?= <sybren@stuvel.eu>
Date: Wed, 28 Mar 2018 12:42:36 +0200
Subject: [PATCH] Added unit test

Should have been part of prev commit.
---
 tests/test_web/test_comments_edit.py | 57 ++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 tests/test_web/test_comments_edit.py

diff --git a/tests/test_web/test_comments_edit.py b/tests/test_web/test_comments_edit.py
new file mode 100644
index 00000000..2917c0a7
--- /dev/null
+++ b/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'])