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

"""

 Authors:

   Julen Landa Alustiza <jlanda@fedoraproject.org></jlanda@fedoraproject.org>

"""

from __future__ import unicode_literals, absolute_import

import sys

import os

sys.path.insert(

    0, os.path.join(os.path.dirname(os.path.abspath(__file__)), "..")

)

import tests

import pagure.lib.query

import pygit2

class PagureFlaskPrEditSimpletests(tests.Modeltests):

    def test_pr_edit_no_project(self):

        """ Test the edit pull request endpoint """

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

        self.assertEqual(output.status_code, 404)

        output_text = output.get_data(as_text=True)

        self.assertIn(

            "<title>Page not found :'( - Pagure</title>", output_text

        )

        self.assertIn("
Page not found (404)
", output_text)

    def test_pr_edit_no_git_repo(self):

        """ Test the edit pull request endpoint """

        tests.create_projects(self.session)

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

        self.assertEqual(output.status_code, 404)

        output_text = output.get_data(as_text=True)

        self.assertIn(

            "<title>Page not found :'( - Pagure</title>", output_text

        )

        self.assertIn("
No git repo found
", output_text)

    def test_pr_edit_no_pull_requests_no_login(self):

        """ Test the edit pull request endpoint """

        tests.create_projects(self.session)

        tests.create_projects_git(os.path.join(self.path, "repos"), bare=True)

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

        self.assertEqual(output.status_code, 302)

    def test_pr_edit_no_pull_requests(self):

        """ Test the edit pull request endpoint """

        tests.create_projects(self.session)

        tests.create_projects_git(os.path.join(self.path, "repos"), bare=True)

        user = tests.FakeUser()

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

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

            self.assertEqual(output.status_code, 404)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                "<title>Page not found :'( - Pagure</title>", output_text

            )

            self.assertIn("
Pull-request not found
", output_text)

class PagureFlaskPrEdittests(tests.Modeltests):

    def setUp(self):

        super(PagureFlaskPrEdittests, self).setUp()

        tests.create_projects(self.session)

        tests.create_projects_git(os.path.join(self.path, "repos"), bare=True)

        # Create foo's fork of pingou's test project

        item = pagure.lib.model.Project(

            user_id=2,  # foo

            name="test",

            description="test project #1",

            hook_token="aaabbb",

            is_fork=True,

            parent_id=1,

        )

        self.session.add(item)

        self.session.commit()

        # Create the fork's git repo

        repo_path = os.path.join(self.path, "repos", item.path)

        pygit2.init_repository(repo_path, bare=True)

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

        fork = pagure.lib.query.get_authorized_project(

            self.session, "test", user="foo"

        )

        tests.add_pull_request_git_repo(

            self.path,

            self.session,

            project,

            fork,

            user="foo",

            allow_rebase=True,

        )

    def tearDown(self):

        super(PagureFlaskPrEdittests, self).tearDown()

        tests.clean_pull_requests_path()

    def test_pr_edit_pull_request_unauthenticated(self):

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

        self.assertEqual(output.status_code, 302)

    def test_pr_edit_pull_request_unauthorized(self):

        user = tests.FakeUser()

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

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

            self.assertEqual(output.status_code, 403)

            output_text = output.get_data(as_text=True)

            self.assertIn("<title>403 Forbidden</title>", output_text)

            self.assertIn(

                "
You are not allowed to edit this pull-request
",

                output_text,

            )

    def test_pr_edit_pull_request_view_author(self):

        user = tests.FakeUser(username="foo")

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

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

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            # Author is editing PR #1

            self.assertIn(

                "<title>Edit PR#1: PR from the feature branch - test - "</title>

                "Pagure",

                output_text,

            )

            # Author has a title input

            if self.get_wtforms_version() >= (2, 2):

                self.assertIn(

                    '

                    'required type="text" value="PR from the feature branch">',

                    output_text,

                )

            else:

                self.assertIn(

                    '

                    'type="text" value="PR from the feature branch">',

                    output_text,

                )

            # Author has an initial_commit textarea

            self.assertIn(

                '

                'id="initial_comment"\n                    '

                'name="initial_comment">',

                output_text,

            )

            # Author has an non-disabled allow_rebase input

            self.assertIn(

                '

                'type="checkbox" value="y" checked>',

                output_text,

            )

    def test_pr_edit_pull_request_post_author_no_csrf_token(self):

        user = tests.FakeUser(username="foo")

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

            data = {

                "title": "New title",

                "initial_comment": "New initial comment",

                "allow_rebase": False,

            }

            output = self.app.post(

                "/test/pull-request/1/edit", data=data, follow_redirects=True

            )

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            # Without CSRF token, we finish again on the form with new

            # values.

            self.assertIn(

                "<title>Edit PR#1: PR from the feature branch - test - "</title>

                "Pagure",

                output_text,

            )

            if self.get_wtforms_version() >= (2, 2):

                self.assertIn(

                    '

                    'required type="text" value="New title">',

                    output_text,

                )

            else:

                self.assertIn(

                    '

                    'type="text" value="New title">',

                    output_text,

                )

            self.assertIn(

                '

                'id="initial_comment"\n                    '

                'name="initial_comment">New initial comment',

                output_text,

            )

            self.assertIn(

                '

                ' value="y" checked>',

                output_text,

            )

            request = pagure.lib.query.search_pull_requests(

                self.session, project_id=1, requestid=1

            )

            # DB model has not been changed

            self.assertEqual("PR from the feature branch", request.title)

            self.assertEqual(None, request.initial_comment)

            self.assertEqual(True, request.allow_rebase)

    def test_pr_edit_pull_request_post_author(self):

        user = tests.FakeUser(username="foo")

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

            data = {

                "title": "New title",

                "initial_comment": "New initial comment",

                "allow_rebase": False,

                "csrf_token": self.get_csrf(),

            }

            output = self.app.post(

                "/test/pull-request/1/edit", data=data, follow_redirects=True

            )

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            # After successful edit, we end on pull_request view with new data

            self.assertIn(

                "<title>PR#1: New title - test\n - Pagure</title>", output_text

            )

            self.assertIn(

                '\n'

                "                  New title\n"

                "            ",

                output_text,

            )

            self.assertIn("
New initial comment
", output_text)

            request = pagure.lib.query.search_pull_requests(

                self.session, project_id=1, requestid=1

            )

            # DB model has been changed

            self.assertEqual("New title", request.title)

            self.assertEqual("New initial comment", request.initial_comment)

            self.assertEqual(False, request.allow_rebase)

    def test_pr_edit_pull_request_view_committer(self):

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

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

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

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            # Committer is editing PR #1

            self.assertIn(

                "<title>Edit PR#1: PR from the feature branch - test - "</title>

                "Pagure",

                output_text,

            )

            # Committer has a title input

            if self.get_wtforms_version() >= (2, 2):

                self.assertIn(

                    '

                    'required type="text" value="PR from the feature branch">',

                    output_text,

                )

            else:

                self.assertIn(

                    '

                    'type="text" value="PR from the feature branch">',

                    output_text,

                )

            # Committer has an initial_commit textarea

            self.assertIn(

                '

                'id="initial_comment"\n'

                '                    name="initial_comment">',

                output_text,

            )

            # Committer has an disabled allow_rebase input

            self.assertIn(

                '

                ' value="y" checked disabled>',

                output_text,

            )

    def test_pr_edit_pull_request_post_committer(self):

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

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

            data = {

                "title": "New title",

                "initial_comment": "New initial comment",

                "allow_rebase": False,

                "csrf_token": self.get_csrf(),

            }

            output = self.app.post(

                "/test/pull-request/1/edit", data=data, follow_redirects=True

            )

            self.assertEqual(output.status_code, 200)

            output_text = output.get_data(as_text=True)

            # After successful edit, we end on pull_request view with new data

            self.assertIn(

                "<title>PR#1: New title - test\n - Pagure</title>", output_text

            )

            self.assertIn(

                '\n'

                "                  New title\n"

                "            ",

                output_text,

            )

            self.assertIn("
New initial comment
", output_text)

            request = pagure.lib.query.search_pull_requests(

                self.session, project_id=1, requestid=1

            )

            # DB model has been changed

            self.assertEqual("New title", request.title)

            self.assertEqual("New initial comment", request.initial_comment)

            # But allow_rebase remains unchanged

            self.assertEqual(True, request.allow_rebase)