Blob Blame Raw
# -*- coding: utf-8 -*-

"""
 (c) 2017 - Copyright Red Hat Inc

 Authors:
   Pierre-Yves Chibon <pingou@pingoured.fr>

"""

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


class PagureFlaskIssuestests(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(PagureFlaskIssuestests, 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)
            self.assertIn(
                '<div class="card-header">\n        New issue',
                output.data)
            self.assertNotIn(
                '<strong><label for="status">Type</label></strong>',
                output.data)
            self.assertNotIn(
                '<select class="form-control c-select" id="type" name="type">',
                output.data)

    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)
            self.assertIn(
                '<div class="card-header">\n        New issue',
                output.data)
            self.assertIn(
                '<strong><label for="status">Type</label></strong>',
                output.data)
            self.assertIn(
                '<select class="form-control custom-select" id="type" name="type">',
                output.data)
            self.assertIn(
                '<option selected value="RFE">RFE</option>',
                output.data)
            self.assertIn(
                '<option selected value="2018-bid">2018-bid</option>',
                output.data)

    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.data)
            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.data)
            self.assertEqual(
                data,
                {"code": "ERROR", "message": "No template provided"})

    def test_get_ticket_template_no_project(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?template=RFE', 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.data)
            self.assertEqual(
                data,
                {"code": "ERROR", "message": "No such template found"})

    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.data)
            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.data)
            self.assertEqual(
                data,
                {
                    "code": "OK",
                    "message": "RFE\n###\n\n* Idea description"
                }
            )

if __name__ == '__main__':
    unittest.main(verbosity=2)