# -*- coding: utf-8 -*-

"""

 (c) 2015-2016 - Copyright Red Hat Inc

 Authors:

   Lubomír Sedlář <lsedlar@redhat.com></lsedlar@redhat.com>

"""

import mock

import os

import sys

import unittest

sys.path.insert(0, os.path.join(os.path.dirname(

    os.path.abspath(__file__)), '..'))

import pagure.lib

import pagure.lib.plugins

import pagure.lib.model

import pagure.hooks

import tests

class PagureFlaskQuickReplytest(tests.Modeltests):

    """ Tests for configuring and displaying quick replies. """

    def setUp(self):

        """ Set up the environnment, ran before every tests. """

        super(PagureFlaskQuickReplytest, self).setUp()

        pagure.APP.config['TESTING'] = True

        pagure.SESSION = self.session

        pagure.ui.app.SESSION = self.session

        pagure.ui.filters.SESSION = self.session

        pagure.ui.fork.SESSION = self.session

        pagure.ui.issues.SESSION = self.session

        pagure.ui.plugins.SESSION = self.session

        pagure.ui.repo.SESSION = self.session

        pagure.ui.SESSION = self.session

        pagure.APP.config['GIT_FOLDER'] = self.path

        pagure.APP.config['TICKETS_FOLDER'] = os.path.join(

            self.path, 'tickets')

        pagure.APP.config['DOCS_FOLDER'] = os.path.join(

            self.path, 'docs')

        self.app = pagure.APP.test_client()

        tests.create_projects(self.session)

        tests.create_projects_git(os.path.join(self.path), bare=True)

        self.admin = tests.FakeUser(username='pingou')

        self.user = tests.FakeUser(username='ralph')

        self.repo = pagure.lib.get_project(self.session, 'test')

    def disable_issues_and_pull_requests(self):

        """Disable both issues and pull requests."""

        # This can not use direct access as repo.settings is a property that

        # serializes data into JSON. Direct modification is not preserved.

        settings = self.repo.settings

        settings['issue_tracker'] = False

        settings['pull_requests'] = False

        self.repo.settings = settings

        self.session.add(self.repo)

        self.session.commit()

    def setup_quick_replies(self):

        """Create some quick replies.

        The full replies are stored as r1 and r2 attributes, with shortened

        versions in sr1 and sr2.

        """

        self.r1 = 'Ship it!'

        self.r2 = ('Nah. I would prefer if you did not submit this, as there '

                   'are problems.')

        self.sr1 = self.r1

        self.sr2 = 'Nah. I would prefer if you did not submit this, as...'

        # Set some quick replies

        self.repo.quick_replies = [self.r1, self.r2]

        self.session.add(self.repo)

    def get_csrf(self, url='/test/settings'):

        """Retrieve a CSRF token from given URL."""

        output = self.app.get(url)

        self.assertEqual(output.status_code, 200)

        return output.data.split(

            'name="csrf_token" type="hidden" value="')[1].split('">')[0]

    def assertRedirectToSettings(self, output, project='test', notice=None):

        """

        Check that user was redirected to settings page of a given project

        and that a given notice was printed.

        """

        self.assertEqual(output.status_code, 200)

        self.assertIn(

            u'<title>Settings - %s - Pagure</title>' % project, output.data)

        self.assertIn(u'
Settings for %s
' % project, output.data)

        if notice:

            self.assertIn(notice, output.data)

    def assertQuickReplies(self, quick_replies, project='test'):

        repo = pagure.lib.get_project(self.session, project)

        self.assertEqual(repo.quick_replies, quick_replies)

    def assertQuickReplyLinks(self, output):

        """Assert reply links created by setup_quick_replies are present."""

        link = 'data-qr="%s">\s*%s\s*'

        self.assertRegexpMatches(

            output.data, link % (self.r1, self.sr1))

        self.assertRegexpMatches(

            output.data, link % (self.r2, self.sr2))

    def test_new_project_has_none(self):

        self.assertQuickReplies([])

    def test_update_quick_reply_without_csrf(self):

        with tests.user_set(pagure.APP, self.admin):

            output = self.app.get('/test/settings')

            self.assertEqual(output.status_code, 200)

            data = {

                'quick_reply': 'Ship it!',

            }

            output = self.app.post(

                '/test/update/quick_replies', data=data, follow_redirects=True)

            self.assertRedirectToSettings(output)

            self.assertQuickReplies([])

    def test_update_quick_replies_single(self):

        with tests.user_set(pagure.APP, self.admin):

            data = {

                'quick_reply': 'Ship it!',

                'csrf_token': self.get_csrf(),

            }

            output = self.app.post(

                '/test/update/quick_replies', data=data, follow_redirects=True)

            self.assertRedirectToSettings(

                output, notice=u'quick replies updated')

            self.assertQuickReplies(['Ship it!'])

            self.assertIn(u'>Ship it!', output.data)

    def test_update_quick_replies_multiple(self):

        with tests.user_set(pagure.APP, self.admin):

            data = {

                'quick_reply': [u'Ship it!', u'Nah.'],

                'csrf_token': self.get_csrf(),

            }

            output = self.app.post(

                '/test/update/quick_replies', data=data, follow_redirects=True)

            self.assertRedirectToSettings(

                output, notice=u'quick replies updated')

            self.assertQuickReplies([u'Ship it!', u'Nah.'])

            # Check page has filled in textarea.

            self.assertIn(u'>Ship it!', output.data)

            self.assertIn(u'>Nah.', output.data)

    def test_update_quick_replies_empty_to_reset(self):

        # Set some quick replies

        repo = pagure.lib.get_project(self.session, 'test')

        repo.quick_replies = ['Ship it!', 'Nah.']

        self.session.add(repo)

        self.session.commit()

        with tests.user_set(pagure.APP, self.admin):

            data = {

                'quick_reply': [],

                'csrf_token': self.get_csrf(),

            }

            output = self.app.post(

                '/test/update/quick_replies', data=data, follow_redirects=True)

            self.assertRedirectToSettings(

                output, notice=u'quick replies updated')

            self.assertQuickReplies([])

    def test_update_quick_replies_unprivileged(self):

        with tests.user_set(pagure.APP, self.user):

            data = {

                'quick_reply': 'Ship it!',

                'csrf_token': 'a guess',

            }

            output = self.app.post(

                '/test/update/quick_replies', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 403)

            self.assertQuickReplies([])

    def test_no_form_with_disabled_issues_and_pull_requests(self):

        self.disable_issues_and_pull_requests()

        with tests.user_set(pagure.APP, self.admin):

            output = self.app.get('/test/settings')

            self.assertNotIn('Quick replies', output.data)

    def test_no_submit_with_disabled_issues_and_pull_requests(self):

        self.disable_issues_and_pull_requests()

        with tests.user_set(pagure.APP, self.admin):

            data = {

                'quick_reply': 'Ship it!',

                'csrf_token': 'a guess',

            }

            output = self.app.post(

                '/test/update/quick_replies', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 404)

            self.assertQuickReplies([])

    def test_submit_for_bad_project(self):

        with tests.user_set(pagure.APP, self.admin):

            data = {

                'quick_reply': 'Ship it!',

                'csrf_token': 'a guess',

            }

            output = self.app.post(

                '/boom/update/quick_replies', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 404)

    @mock.patch('pagure.lib.git.update_git')

    def test_issue_page_has_quick_replies(self, p_ugt):

        self.setup_quick_replies()

        issue = pagure.lib.new_issue(

            self.session,

            self.repo,

            'Dummy issue',

            'Just a lonely issue.',

            'pingou',

            None,

            notify=False

        )

        with tests.user_set(pagure.APP, self.user):

            output = self.app.get('/test/issue/%s' % issue.id)

            self.assertEqual(output.status_code, 200)

            self.assertQuickReplyLinks(output)

    @mock.patch('pagure.lib.git.update_git')

    @mock.patch('pagure.lib.git.diff_pull_request')

    def test_pull_request_page_has_quick_replies(self, diff, p_ugt):

        diff.return_value = ([], [])

        self.setup_quick_replies()

        pr = pagure.lib.new_pull_request(

            self.session,

            'pr',

            self.repo,

            'master',

            'Dummy PR', 'pingou',

            None,

            repo_from=self.repo,

            notify=False,

        )

        with tests.user_set(pagure.APP, self.user):

            output = self.app.get('/test/pull-request/%s' % pr.id)

            self.assertEqual(output.status_code, 200)

            self.assertQuickReplyLinks(output)

if __name__ == '__main__':

    SUITE = unittest.TestLoader().loadTestsFromTestCase(

        PagureFlaskQuickReplytest)

    unittest.TextTestRunner(verbosity=2).run(SUITE)