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

"""

 (c) 2015 - Copyright Red Hat Inc

 Authors:

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

"""

__requires__ = ['SQLAlchemy >= 0.8']

import pkg_resources

import json

import unittest

import shutil

import sys

import os

import pygit2

from mock import patch

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

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

import progit.lib

import tests

class ProgitFlaskIssuestests(tests.Modeltests):

    """ Tests for flask issues controller of progit """

    def setUp(self):

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

        super(ProgitFlaskIssuestests, self).setUp()

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

        progit.SESSION = self.session

        progit.ui.SESSION = self.session

        progit.ui.app.SESSION = self.session

        progit.ui.issues.SESSION = self.session

        progit.ui.repo.SESSION = self.session

        progit.APP.config['GIT_FOLDER'] = tests.HERE

        progit.APP.config['FORK_FOLDER'] = os.path.join(

            tests.HERE, 'forks')

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

            tests.HERE, 'tickets')

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

            tests.HERE, 'docs')

        self.app = progit.APP.test_client()

    @patch('progit.lib.git.update_git_ticket')

    @patch('progit.lib.notify.send_email')

    def test_new_issue(self, p_send_email, p_ugt):

        """ Test the new_issue endpoint. """

        p_send_email.return_value = True

        p_ugt.return_value = True

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

        self.assertEqual(output.status_code, 302)

        user = tests.FakeUser()

        with tests.user_set(progit.APP, user):

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

            self.assertEqual(output.status_code, 404)

            tests.create_projects(self.session)

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

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
New issue
' in output.data)

            csrf_token = output.data.split(

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

            data = {

            }

            # Insufficient input

            output = self.app.post('/test/new_issue', data=data)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
New issue
' in output.data)

            self.assertEqual(output.data.count(

                'This field is required.'), 2)

            data['title'] = 'Test issue'

            output = self.app.post('/test/new_issue', data=data)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
New issue
' in output.data)

            self.assertEqual(output.data.count(

                'This field is required.'), 1)

            data['issue_content'] = 'We really should improve on this issue'

            data['status'] = 'Open'

            output = self.app.post('/test/new_issue', data=data)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
New issue
' in output.data)

            self.assertEqual(output.data.count(

                'This field is required.'), 0)

            # Invalid user

            data['csrf_token'] = csrf_token

            output = self.app.post('/test/new_issue', data=data)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
New issue
' in output.data)

            self.assertEqual(output.data.count(

                'This field is required.'), 0)

            self.assertTrue(

                '
No user "username" found
'

                in output.data)

        user.username = 'pingou'

        with tests.user_set(progit.APP, user):

            output = self.app.post(

                '/test/new_issue', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue(

                '
Issue created
' in output.data)

            self.assertTrue('
test project #1
' in output.data)

            self.assertTrue('
\n    Issues (1)\n  
' in output.data)

    @patch('progit.lib.git.update_git_ticket')

    @patch('progit.lib.notify.send_email')

    def test_view_issues(self, p_send_email, p_ugt):

        """ Test the view_issues endpoint. """

        p_send_email.return_value = True

        p_ugt.return_value = True

        output = self.app.get('/foo/issues')

        self.assertEqual(output.status_code, 404)

        tests.create_projects(self.session)

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

        self.assertEqual(output.status_code, 200)

        self.assertTrue('
test project #1
' in output.data)

        self.assertTrue('
\n    Issues (0)\n  
' in output.data)

        # Create issues to play with

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

        msg = progit.lib.new_issue(

            session=self.session,

            repo=repo,

            title='Test issue',

            content='We should work on this',

            user='pingou',

            ticketfolder=None

        )

        self.session.commit()

        self.assertEqual(msg, 'Issue created')

        # Whole list

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

        self.assertEqual(output.status_code, 200)

        self.assertTrue('
test project #1
' in output.data)

        self.assertTrue('
\n    Issues (1)\n  
' in output.data)

        # Status = closed

        output = self.app.get('/test/issues?status=cloSED')

        self.assertEqual(output.status_code, 200)

        self.assertTrue('
test project #1
' in output.data)

        self.assertTrue(

            '
\n    Closed\n    Issues (0)\n  
' in output.data)

        # Status = fixed

        output = self.app.get('/test/issues?status=fixed')

        self.assertEqual(output.status_code, 200)

        self.assertTrue('
test project #1
' in output.data)

        self.assertTrue(

            '
\n    Closed\n    Issues (0)\n  
' in output.data)

        # Project w/o issue tracker

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

        repo.issue_tracker = False

        self.session.add(repo)

        self.session.commit()

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

        self.assertEqual(output.status_code, 404)

    @patch('progit.lib.git.update_git_ticket')

    @patch('progit.lib.notify.send_email')

    def test_view_issue(self, p_send_email, p_ugt):

        """ Test the view_issue endpoint. """

        p_send_email.return_value = True

        p_ugt.return_value = True

        output = self.app.get('/foo/issue/1')

        self.assertEqual(output.status_code, 404)

        tests.create_projects(self.session)

        output = self.app.get('/test/issue/1')

        self.assertEqual(output.status_code, 404)

        # Create issues to play with

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

        msg = progit.lib.new_issue(

            session=self.session,

            repo=repo,

            title='Test issue',

            content='We should work on this',

            user='pingou',

            ticketfolder=None

        )

        self.session.commit()

        self.assertEqual(msg, 'Issue created')

        output = self.app.get('/test/issue/1')

        self.assertEqual(output.status_code, 200)

        self.assertTrue('
test project #1
' in output.data)

        self.assertTrue(

            '
Login to comment on this ticket.
'

            in output.data)

        user = tests.FakeUser()

        with tests.user_set(progit.APP, user):

            output = self.app.get('/test/issue/1')

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
test project #1
' in output.data)

            self.assertFalse(

                '
Login to comment on this ticket.
'

                in output.data)

        user.username = 'pingou'

        with tests.user_set(progit.APP, user):

            output = self.app.get('/test/issue/1')

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
test project #1
' in output.data)

            csrf_token = output.data.split(

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

        # Project w/o issue tracker

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

        repo.issue_tracker = False

        self.session.add(repo)

        self.session.commit()

        output = self.app.get('/test/issue/1')

        self.assertEqual(output.status_code, 404)

    @patch('progit.lib.git.update_git_ticket')

    @patch('progit.lib.notify.send_email')

    def test_update_issue(self, p_send_email, p_ugt):

        """ Test the update_issue endpoint. """

        p_send_email.return_value = True

        p_ugt.return_value = True

        output = self.app.get('/foo/issue/1/update')

        self.assertEqual(output.status_code, 404)

        tests.create_projects(self.session)

        output = self.app.get('/test/issue/1/update')

        self.assertEqual(output.status_code, 404)

        # Create issues to play with

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

        msg = progit.lib.new_issue(

            session=self.session,

            repo=repo,

            title='Test issue',

            content='We should work on this',

            user='pingou',

            ticketfolder=None

        )

        self.session.commit()

        self.assertEqual(msg, 'Issue created')

        user = tests.FakeUser()

        user.username = 'pingou'

        with tests.user_set(progit.APP, user):

            output = self.app.get('/test/issue/1')

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
test project #1
' in output.data)

            csrf_token = output.data.split(

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

            data = {

                'status': 'fixed'

            }

            output = self.app.post(

                '/test/issue/1/update', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
test project #1
' in output.data)

            self.assertFalse(

                '<option selected="" value="Fixed">Fixed</option>'

                in output.data)

            data['status'] = 'Fixed'

            output = self.app.post(

                '/test/issue/1/update', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
test project #1
' in output.data)

            self.assertFalse(

                '<option selected="" value="Fixed">Fixed</option>'

                in output.data)

            data['csrf_token'] = csrf_token

            output = self.app.post(

                '/test/issue/1/update', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
test project #1
' in output.data)

            self.assertTrue(

                '
Edited successfully issue #1
'

                in output.data)

            self.assertTrue(

                '<option selected="" value="Fixed">Fixed</option>'

                in output.data)

            # Add new comment

            data = {

                'csrf_token': csrf_token,

                'status': 'Fixed',

                'comment': 'Woohoo a second comment !',

            }

            output = self.app.post(

                '/test/issue/1/update', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
test project #1
' in output.data)

            self.assertTrue(

                '
Comment added
' in output.data)

            self.assertTrue(

                'li class="message">No changes to edit' in output.data)

            self.assertTrue(

                '
Woohoo a second comment !
' in output.data)

            self.assertEqual(

                output.data.count('
'), 2)

            self.assertTrue(

                '<option selected="" value="Fixed">Fixed</option>'

                in output.data)

            # Add new tag

            data = {

                'csrf_token': csrf_token,

                'status': 'Fixed',

                'tag': 'tag#2',

            }

            output = self.app.post(

                '/test/issue/1/update', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
test project #1
' in output.data)

            self.assertTrue(

                '
Tag added
' in output.data)

            self.assertTrue(

                'li class="message">No changes to edit' in output.data)

            self.assertTrue(

                '
Woohoo a second comment !
' in output.data)

            self.assertEqual(

                output.data.count('
'), 2)

            self.assertTrue(

                '<option selected="" value="Fixed">Fixed</option>'

                in output.data)

            # Assign issue to an non-existent user

            data = {

                'csrf_token': csrf_token,

                'status': 'Fixed',

                'assignee': 'ralph',

            }

            output = self.app.post(

                '/test/issue/1/update', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
test project #1
' in output.data)

            self.assertTrue(

                '
No user "ralph" found
'

                in output.data)

            self.assertTrue(

                '
Woohoo a second comment !
' in output.data)

            self.assertEqual(

                output.data.count('
'), 2)

            self.assertTrue(

                '<option selected="" value="Fixed">Fixed</option>'

                in output.data)

            # Assign issue properly

            data = {

                'csrf_token': csrf_token,

                'status': 'Fixed',

                'assignee': 'pingou',

            }

            output = self.app.post(

                '/test/issue/1/update', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
test project #1
' in output.data)

            self.assertTrue(

                '
Issue assigned
' in output.data)

            self.assertTrue(

                '' in output.data)

            self.assertTrue(

                '
Woohoo a second comment !
' in output.data)

            self.assertEqual(

                output.data.count('
'), 2)

            self.assertTrue(

                '<option selected="" value="Fixed">Fixed</option>'

                in output.data)

        # Create another issue with a dependency

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

        msg = progit.lib.new_issue(

            session=self.session,

            repo=repo,

            title='Test issue',

            content='We should work on this',

            user='pingou',

            ticketfolder=None

        )

        self.session.commit()

        self.assertEqual(msg, 'Issue created')

        # Reset the status of the first issue

        parent_issue = progit.lib.search_issues(

            self.session, repo, issueid=2)

        parent_issue.status = 'Open'

        # Add the dependency relationship

        self.session.add(parent_issue)

        issue = progit.lib.search_issues(self.session, repo, issueid=2)

        issue.parents.append(parent_issue)

        self.session.add(issue)

        self.session.commit()

        with tests.user_set(progit.APP, user):

            data['csrf_token'] = csrf_token

            output = self.app.post(

                '/test/issue/2/update', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
test project #1
' in output.data)

            self.assertTrue(

                '
You cannot close a ticket that has ticket '

                'depending that are still open.' in output.data)

            self.assertTrue(

                '<option selected="" value="Open">Open</option>'

                in output.data)

        # Project w/o issue tracker

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

        repo.issue_tracker = False

        self.session.add(repo)

        self.session.commit()

        with tests.user_set(progit.APP, user):

            output = self.app.get('/test/issue/1/update')

            self.assertEqual(output.status_code, 404)

    @patch('progit.lib.git.update_git_ticket')

    @patch('progit.lib.notify.send_email')

    def test_edit_issue(self, p_send_email, p_ugt):

        """ Test the edit_issue endpoint. """

        p_send_email.return_value = True

        p_ugt.return_value = True

        output = self.app.get('/foo/issue/1/edit')

        self.assertEqual(output.status_code, 302)

        user = tests.FakeUser()

        with tests.user_set(progit.APP, user):

            output = self.app.get('/foo/issue/1/edit')

            self.assertEqual(output.status_code, 404)

            tests.create_projects(self.session)

            output = self.app.get('/test/issue/1/edit')

            self.assertEqual(output.status_code, 403)

        user.username = 'pingou'

        with tests.user_set(progit.APP, user):

            output = self.app.get('/test/issue/1/edit')

            self.assertEqual(output.status_code, 404)

        # Create issues to play with

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

        msg = progit.lib.new_issue(

            session=self.session,

            repo=repo,

            title='Test issue',

            content='We should work on this',

            user='pingou',

            ticketfolder=None

        )

        self.session.commit()

        self.assertEqual(msg, 'Issue created')

        user = tests.FakeUser()

        with tests.user_set(progit.APP, user):

            output = self.app.get('/test/issue/1/edit')

            self.assertEqual(output.status_code, 403)

        user.username = 'pingou'

        with tests.user_set(progit.APP, user):

            output = self.app.get('/test/issue/1/edit')

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
Edit issue #1
' in output.data)

            csrf_token = output.data.split(

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

            data = {

                'issue_content': 'We should work on this!'

            }

            output = self.app.post('/test/issue/1/edit', data=data)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
Edit issue #1
' in output.data)

            self.assertEqual(output.data.count(

                'This field is required.'), 1)

            self.assertEqual(output.data.count(

                'Not a valid choice'), 1)

            data['status'] = 'Open'

            data['title'] = 'Test issue #1'

            output = self.app.post('/test/issue/1/edit', data=data)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
Edit issue #1
' in output.data)

            self.assertEqual(output.data.count(

                'This field is required.'), 0)

            self.assertEqual(output.data.count(

                'Not a valid choice'), 0)

            data['csrf_token'] = csrf_token

            output = self.app.post(

                '/test/issue/1/edit', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue(

                '#1 Test issue #1'

                in output.data)

            self.assertEqual(output.data.count(

                '<option selected="" value="Open">Open</option>'), 1)

            self.assertEqual(output.data.count(

                '
\n        '

                '
We should work on this!
'), 1)

        # Project w/o issue tracker

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

        repo.issue_tracker = False

        self.session.add(repo)

        self.session.commit()

        user.username = 'pingou'

        with tests.user_set(progit.APP, user):

            output = self.app.post('/test/issue/1/edit', data=data)

            self.assertEqual(output.status_code, 404)

    @patch('progit.lib.git.update_git_ticket')

    @patch('progit.lib.notify.send_email')

    def test_edit_tag(self, p_send_email, p_ugt):

        """ Test the edit_tag endpoint. """

        p_send_email.return_value = True

        p_ugt.return_value = True

        output = self.app.get('/foo/tag/foo/edit')

        self.assertEqual(output.status_code, 302)

        user = tests.FakeUser()

        with tests.user_set(progit.APP, user):

            output = self.app.get('/foo/tag/foo/edit')

            self.assertEqual(output.status_code, 404)

            tests.create_projects(self.session)

            output = self.app.get('/test/tag/foo/edit')

            self.assertEqual(output.status_code, 403)

        # Create issues to play with

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

        msg = progit.lib.new_issue(

            session=self.session,

            repo=repo,

            title='Test issue',

            content='We should work on this',

            user='pingou',

            ticketfolder=None

        )

        self.session.commit()

        self.assertEqual(msg, 'Issue created')

        # Add a tag to the issue

        issue = progit.lib.search_issues(self.session, repo, issueid=1)

        msg = progit.lib.add_issue_tag(

            session=self.session,

            issue=issue,

            tag='tag1',

            user='pingou',

            ticketfolder=None)

        self.session.commit()

        self.assertEqual(msg, 'Tag added')

        # Before edit, list tags

        tags = progit.lib.get_tags_of_project(self.session, repo)

        self.assertEqual([tag.tag for tag in tags], ['tag1'])

        # Edit tag

        user.username = 'pingou'

        with tests.user_set(progit.APP, user):

            output = self.app.get('/test/tag/tag1/edit')

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
Edit tag: tag1
' in output.data)

            self.assertTrue(

                '
Enter in the field below the new name for the tag: '

                '"tag1"
' in output.data)

            csrf_token = output.data.split(

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

            data = {'tag': 'tag2'}

            output = self.app.post('/test/tag/tag1/edit', data=data)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
Edit tag: tag1
' in output.data)

            self.assertTrue(

                '
Enter in the field below the new name for the tag: '

                '"tag1"
' in output.data)

            data['csrf_token'] = csrf_token

            output = self.app.post(

                '/test/tag/tag1/edit', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
test project #1
' in output.data)

            self.assertTrue(

                '
Edited tag: tag1 to tag2
'

                in output.data)

        # After edit, list tags

        tags = progit.lib.get_tags_of_project(self.session, repo)

        self.assertEqual([tag.tag for tag in tags], ['tag2'])

    @patch('progit.lib.git.update_git_ticket')

    @patch('progit.lib.notify.send_email')

    def test_remove_tag(self, p_send_email, p_ugt):

        """ Test the remove_tag endpoint. """

        p_send_email.return_value = True

        p_ugt.return_value = True

        output = self.app.post('/foo/droptag/')

        self.assertEqual(output.status_code, 302)

        user = tests.FakeUser()

        with tests.user_set(progit.APP, user):

            output = self.app.post('/foo/droptag/')

            self.assertEqual(output.status_code, 404)

            tests.create_projects(self.session)

            output = self.app.post('/test/droptag/')

            self.assertEqual(output.status_code, 403)

        # Create issues to play with

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

        msg = progit.lib.new_issue(

            session=self.session,

            repo=repo,

            title='Test issue',

            content='We should work on this',

            user='pingou',

            ticketfolder=None

        )

        self.session.commit()

        self.assertEqual(msg, 'Issue created')

        # Add a tag to the issue

        issue = progit.lib.search_issues(self.session, repo, issueid=1)

        msg = progit.lib.add_issue_tag(

            session=self.session,

            issue=issue,

            tag='tag1',

            user='pingou',

            ticketfolder=None)

        self.session.commit()

        self.assertEqual(msg, 'Tag added')

        # Before edit, list tags

        tags = progit.lib.get_tags_of_project(self.session, repo)

        self.assertEqual([tag.tag for tag in tags], ['tag1'])

        # Edit tag

        user.username = 'pingou'

        with tests.user_set(progit.APP, user):

            output = self.app.post(

                '/test/droptag/', data={}, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
Settings
' in output.data)

            self.assertTrue(

                '
\n                
' in output.data)

            csrf_token = output.data.split(

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

            data = {'tag': 'tag1'}

            output = self.app.post(

                '/test/droptag/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
Settings
' in output.data)

            self.assertTrue(

                '
\n                
' in output.data)

            data['csrf_token'] = csrf_token

            output = self.app.post(

                '/test/droptag/', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            self.assertTrue('
Settings
' in output.data)

            self.assertTrue(

                '
Removed tag: tag1
' in output.data)

if __name__ == '__main__':

    SUITE = unittest.TestLoader().loadTestsFromTestCase(ProgitFlaskIssuestests)

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