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'])