# -*- 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.add_content_git_repo(

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

        )

        # Fork

        project = pagure.lib.query.get_authorized_project(self.session, "test")

        task = pagure.lib.query.fork_project(

            session=self.session, user="pingou", repo=project

        )

        self.session.commit()

        self.assertEqual(

            task.get(),

            {

                "endpoint": "ui_ns.view_repo",

                "repo": "test",

                "namespace": None,

                "username": "pingou",

            },

        )

        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()}

            # Check the PR view on UI

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

            self.assertEqual(output.status_code, 200)

            # 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,

                {

                    "code": "FFORWARD",

                    "message": "The pull-request can be merged and "

                    "fast-forwarded",

                    "short_code": "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,

                {

                    "code": "FFORWARD",

                    "message": "The pull-request can be merged and "

                    "fast-forwarded",

                    "short_code": "Ok",

                },

            )

if __name__ == "__main__":

    unittest.main(verbosity=2)