58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
|
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'])
|