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

"""

 (c) 2017 - Copyright Red Hat Inc

 Authors:

   Pierre-Yves Chibon <pingou@pingoured.fr></pingou@pingoured.fr>

"""

from __future__ import unicode_literals

__requires__ = ['SQLAlchemy >= 0.8']

import pkg_resources

import json

import unittest

import sys

import os

import pygit2

from mock import patch, MagicMock

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

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

import pagure

import pagure.lib

import tests

def create_templates(repopath):

    """ Create a couple of templates at the specified repo.

    """

    clone_repo = pygit2.Repository(repopath)

    # Create the RFE template

    os.mkdir(os.path.join(repopath, 'templates'))

    template = os.path.join(repopath, 'templates', 'RFE.md')

    with open(template, 'w') as stream:

        stream.write('RFE\n###\n\n* Idea description')

    clone_repo.index.add(os.path.join('templates', 'RFE.md'))

    clone_repo.index.write()

    # Commit

    tree = clone_repo.index.write_tree()

    author = pygit2.Signature(

        'Alice Author', 'alice@authors.tld')

    committer = pygit2.Signature(

        'Cecil Committer', 'cecil@committers.tld')

    commit = clone_repo.create_commit(

        'refs/heads/master',  # the name of the reference to update

        author,

        committer,

        'Add a RFE template',

        # binary string representing the tree object ID

        tree,

        # list of binary strings representing parents of the new commit

        []

    )

    # Create the 2018-bid.md template

    template = os.path.join(repopath, 'templates', '2018-bid.md')

    with open(template, 'w') as stream:

        stream.write('Bid for 2018\n############\n\n* Location:')

    clone_repo.index.add(os.path.join('templates', '2018-bid.md'))

    clone_repo.index.write()

    # Commit

    tree = clone_repo.index.write_tree()

    author = pygit2.Signature(

        'Alice Author', 'alice@authors.tld')

    committer = pygit2.Signature(

        'Cecil Committer', 'cecil@committers.tld')

    commit = clone_repo.create_commit(

        'refs/heads/master',  # the name of the reference to update

        author,

        committer,

        'Add a RFE template',

        # binary string representing the tree object ID

        tree,

        # list of binary strings representing parents of the new commit

        [commit.hex]

    )

    # Create the default.md template

    template = os.path.join(repopath, 'templates', 'default.md')

    with open(template, 'w') as stream:

        stream.write('Report your issue')

    clone_repo.index.add(os.path.join('templates', 'default.md'))

    clone_repo.index.write()

    # Commit

    tree = clone_repo.index.write_tree()

    author = pygit2.Signature(

        'Alice Author', 'alice@authors.tld')

    committer = pygit2.Signature(

        'Cecil Committer', 'cecil@committers.tld')

    clone_repo.create_commit(

        'refs/heads/master',  # the name of the reference to update

        author,

        committer,

        'Add a default template',

        # binary string representing the tree object ID

        tree,

        # list of binary strings representing parents of the new commit

        [commit.hex]

    )

class PagureFlaskIssuesTemplatetests(tests.Modeltests):

    """ Tests for flask issues controller of pagure """

    @patch('pagure.lib.git.update_git', MagicMock(return_value=True))

    @patch('pagure.lib.notify.send_email', MagicMock(return_value=True))

    def setUp(self):

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

        super(PagureFlaskIssuesTemplatetests, self).setUp()

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

            self.path, 'tickets')

        tests.create_projects(self.session)

        tests.create_projects_git(

            os.path.join(self.path, 'repos'), bare=True)

        tests.create_projects_git(

            os.path.join(self.path, 'tickets'))

        # Add a couple of templates to test2

        repopath = os.path.join(self.path, 'tickets', 'test2.git')

        create_templates(repopath)

        # Add a couple of templates to somenamespace/test3

        repopath = os.path.join(

            self.path, 'tickets', 'somenamespace', 'test3.git')

        create_templates(repopath)

    def test_new_issue_no_template(self):

        """ Test the new_issue endpoint when the project has no templates.

        """

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

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

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '
New Issue
\n',

                output_text)

            self.assertNotIn(

                'Issue Templates',

                output_text)

    def test_new_issue_w_template(self):

        """ Test the new_issue endpoint when the project has templates. """

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            output = self.app.get('/test2/new_issue')

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '
New Issue
\n',

                output_text)

            self.assertIn(

                'Issue Templates',

                output_text)

            self.assertIn(

                'RFE',

                output_text)

            self.assertIn(

                '2018-bid',

                output_text)

            self.assertIn(

                'default',

                output_text)

    def test_get_ticket_template_no_csrf(self):

        """ Test the get_ticket_template endpoint when the project has no

        templates.

        """

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            output = self.app.post('/pv/test/issue/template')

            self.assertEqual(output.status_code, 400)

            data = json.loads(output.get_data(as_text=True))

            self.assertEqual(

                data,

                {"code": "ERROR", "message": "Invalid input submitted"})

    def test_get_ticket_template_no_template_specified(self):

        """ Test the get_ticket_template endpoint when not specifying which

        template to get.

        """

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            csrf = self.get_csrf()

            data = {'csrf_token': csrf}

            output = self.app.post('/pv/test/issue/template', data=data)

            self.assertEqual(output.status_code, 400)

            data = json.loads(output.get_data(as_text=True))

            self.assertEqual(

                data,

                {"code": "ERROR", "message": "No template provided"})

    def test_get_ticket_template_no_project(self):

        """ Test the get_ticket_template endpoint when the project does not

        exist.

        """

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            csrf = self.get_csrf()

            data = {'csrf_token': csrf}

            output = self.app.post(

                '/pv/foobar/issue/template', data=data)

            self.assertEqual(output.status_code, 404)

    def test_get_ticket_template_no_template(self):

        """ Test the get_ticket_template endpoint when the project has no

        templates.

        """

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            csrf = self.get_csrf()

            data = {'csrf_token': csrf}

            output = self.app.post(

                '/pv/test/issue/template?template=RFE', data=data)

            self.assertEqual(output.status_code, 404)

            data = json.loads(output.get_data(as_text=True))

            self.assertEqual(

                data,

                {"code": "ERROR", "message": "No such template found"})

    def test_get_ticket_template_issue_tracker_disabled(self):

        """ Test the get_ticket_template endpoint when the project has

        disabled its issue tracker.

        """

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

        settings = repo.settings

        settings['issue_tracker'] = False

        repo.settings = settings

        self.session.add(repo)

        self.session.commit()

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            csrf = self.get_csrf()

            data = {'csrf_token': csrf}

            output = self.app.post(

                '/pv/test/issue/template?template=RFE', data=data)

            self.assertEqual(output.status_code, 404)

            data = json.loads(output.get_data(as_text=True))

            self.assertEqual(

                data,

                {

                    u'code': u'ERROR',

                    u'message': u'No issue tracker found for this project'

                }

            )

    def test_get_ticket_template_w_template(self):

        """ Test the get_ticket_template endpoint when the project has

        templates.

        """

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            csrf = self.get_csrf()

            data = {'csrf_token': csrf}

            output = self.app.post(

                '/pv/test2/issue/template?template=RFE', data=data)

            self.assertEqual(output.status_code, 200)

            data = json.loads(output.get_data(as_text=True))

            self.assertEqual(

                data,

                {

                    "code": "OK",

                    "message": "RFE\n###\n\n* Idea description"

                }

            )

    def test_get_ticket_template_w_template_namespace(self):

        """ Test the get_ticket_template endpoint when the project has

        templates and a namespace.

        """

        user = tests.FakeUser()

        with tests.user_set(self.app.application, user):

            csrf = self.get_csrf()

            data = {'csrf_token': csrf}

            output = self.app.post(

                '/pv/somenamespace/test3/issue/template?template=RFE',

                data=data)

            self.assertEqual(output.status_code, 200)

            data = json.loads(output.get_data(as_text=True))

            self.assertEqual(

                data,

                {

                    "code": "OK",

                    "message": "RFE\n###\n\n* Idea description"

                }

            )

if __name__ == '__main__':

    unittest.main(verbosity=2)