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

"""

 (c) 2018 - Copyright Red Hat Inc

 Authors:

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

"""

from __future__ import unicode_literals, absolute_import

import unittest

import sys

import os

import json

from mock import patch, MagicMock

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

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

import pagure.config

import pagure.lib.query

import pagure.lib.tasks

import tests

class PagureMergePrNoForkTest(tests.Modeltests):

    """ Tests merging a PR in pagure when the fork no longer exists """

    maxDiff = None

    def setUp(self):

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

        super(PagureMergePrNoForkTest, self).setUp()

        tests.create_projects(self.session)

        tests.create_projects_git(

            os.path.join(self.path, "repos"),

            bare=True

        )

        tests.add_content_git_repo(

            os.path.join(self.path, "repos", "test.git"))

        tests.create_projects(

            self.session,

            is_fork=True,

            hook_token_suffix='fork')

        tests.create_projects_git(

            os.path.join(self.path, "repos", "forks", "pingou"),

            bare=True

        )

        tests.add_content_git_repo(

            os.path.join(self.path, "repos", "forks", "pingou", "test.git"))

        tests.add_readme_git_repo(

            os.path.join(self.path, "repos", "forks", "pingou", "test.git"))

        project = pagure.lib.query.get_authorized_project(

            self.session, 'test')

        fork = pagure.lib.query.get_authorized_project(

            self.session,

            'test',

            user='pingou',

        )

        tests.create_tokens(self.session)

        tests.create_tokens_acl(self.session)

        req = pagure.lib.query.new_pull_request(

            session=self.session,

            repo_from=fork,

            branch_from='master',

            repo_to=project,

            branch_to='master',

            title='test pull-request',

            user='pingou',

        )

        self.session.commit()

        self.assertEqual(req.id, 1)

        self.assertEqual(req.title, 'test pull-request')

        # Assert the PR is open

        self.session = pagure.lib.query.create_session(self.dbpath)

        project = pagure.lib.query.get_authorized_project(

            self.session, 'test')

        self.assertEqual(len(project.requests), 1)

        self.assertEqual(project.requests[0].status, "Open")

        # Check how the PR renders in the API and the UI

        output = self.app.get('/api/0/test/pull-request/1')

        self.assertEqual(output.status_code, 200)

        output = self.app.get('/test/pull-request/1')

        self.assertEqual(output.status_code, 200)

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

    def test_api_pull_request_diffstats(self):

        """ Test the api_pull_request_merge method of the flask api. """

        # Check the PR stats in the API

        output = self.app.get(

            '/api/0/test/pull-request/1/diffstats')

        self.assertEqual(output.status_code, 200)

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

        self.assertDictEqual(

            data,

            {

                "README.rst": {

                    "lines_added": 16,

                    "lines_removed": 0,

                    "new_id": "fb7093d2ba1cf8f80d10b45e4f15b10240727db5",

                    "old_id": "0000000000000000000000000000000000000000",

                    "old_path": "README.rst",

                    "status": "A"

                }

            }

        )

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

    def test_api_pull_request_diffstats_no_fork(self):

        """ Test the api_pull_request_merge method of the flask api. """

        pagure.lib.tasks.delete_project(

            namespace=None,

            name="test",

            user="pingou",

            action_user="pingou",

        )

        # Check the PR stats in the API

        output = self.app.get(

            '/api/0/test/pull-request/1/diffstats')

        self.assertEqual(output.status_code, 200)

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

        self.assertDictEqual(

            data,

            {

                "README.rst": {

                    "lines_added": 16,

                    "lines_removed": 0,

                    "new_id": "fb7093d2ba1cf8f80d10b45e4f15b10240727db5",

                    "old_id": "0000000000000000000000000000000000000000",

                    "old_path": "README.rst",

                    "status": "A"

                }

            }

        )

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

    def test_api_pull_request_merge(self):

        """ Test the api_pull_request_merge method of the flask api. """

        headers = {'Authorization': 'token aaabbbcccddd'}

        # Merge PR

        output = self.app.post(

            '/api/0/test/pull-request/1/merge', headers=headers)

        self.assertEqual(output.status_code, 200)

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

        self.assertDictEqual(

            data,

            {"message": "Changes merged!"}

        )

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

    def test_api_pull_request_merge_no_fork(self):

        """ Test the api_pull_request_merge method of the flask api. """

        pagure.lib.tasks.delete_project(

            namespace=None,

            name="test",

            user="pingou",

            action_user="pingou",

        )

        headers = {'Authorization': 'token aaabbbcccddd'}

        # Merge PR

        output = self.app.post(

            '/api/0/test/pull-request/1/merge', headers=headers)

        self.assertEqual(output.status_code, 200)

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

        self.assertDictEqual(

            data,

            {"message": "Changes merged!"}

        )

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

    def test_ui_pull_request_merge(self):

        """ Test the api_pull_request_merge method of the flask UI. """

        user = tests.FakeUser(username='pingou')

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

            data = {

                'csrf_token': self.get_csrf()

            }

            # Merge PR

            output = self.app.post(

                '/test/pull-request/1/merge', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                "<title>Overview - test - Pagure</title>",

                output_text

            )

        self.session = pagure.lib.query.create_session(self.dbpath)

        project = pagure.lib.query.get_authorized_project(

            self.session, 'test')

        self.assertEqual(project.requests[0].status, "Merged")

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

    def test_ui_pull_request_merge_no_fork(self):

        """ Test the api_pull_request_merge method of the flask UI. """

        pagure.lib.tasks.delete_project(

            namespace=None,

            name="test",

            user="pingou",

            action_user="pingou",

        )

        user = tests.FakeUser(username='pingou')

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

            data = {

                'csrf_token': self.get_csrf()

            }

            # Merge PR

            output = self.app.post(

                '/test/pull-request/1/merge', data=data, follow_redirects=True)

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                "<title>Overview - test - Pagure</title>",

                output_text

            )

        self.session = pagure.lib.query.create_session(self.dbpath)

        project = pagure.lib.query.get_authorized_project(

            self.session, 'test')

        self.assertEqual(project.requests[0].status, "Merged")

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

    def test_internal_merge_status(self):

        """ Test the api_pull_request_merge method of the flask UI. """

        self.session = pagure.lib.query.create_session(self.dbpath)

        project = pagure.lib.query.get_authorized_project(

            self.session, 'test')

        user = tests.FakeUser(username='pingou')

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

            data = {

                'csrf_token': self.get_csrf(),

                'requestid': project.requests[0].uid,

            }

            # Merge PR

            output = self.app.post(

                '/pv/pull-request/merge', data=data)

            self.assertEqual(output.status_code, 200)

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

            self.assertDictEqual(

                data,

                {

                    u'code': u'FFORWARD',

                    u'message': u'The pull-request can be merged and '

                        'fast-forwarded',

                    u'short_code': u'Ok'

                }

            )

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

    def test_internal_merge_status_no_fork(self):

        """ Test the api_pull_request_merge method of the flask UI. """

        pagure.lib.tasks.delete_project(

            namespace=None,

            name="test",

            user="pingou",

            action_user="pingou",

        )

        self.session = pagure.lib.query.create_session(self.dbpath)

        project = pagure.lib.query.get_authorized_project(

            self.session, 'test')

        user = tests.FakeUser(username='pingou')

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

            data = {

                'csrf_token': self.get_csrf(),

                'requestid': project.requests[0].uid,

            }

            # Merge PR

            output = self.app.post(

                '/pv/pull-request/merge', data=data)

            self.assertEqual(output.status_code, 200)

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

            self.assertDictEqual(

                data,

                {

                    u'code': u'FFORWARD',

                    u'message': u'The pull-request can be merged and '

                        'fast-forwarded',

                    u'short_code': u'Ok'

                }

            )

if __name__ == '__main__':

    unittest.main(verbosity=2)