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

from __future__ import unicode_literals, absolute_import

import datetime

import unittest

import shutil

import sys

import tempfile

import os

import json

import pygit2

from mock import patch, MagicMock

sys.path.insert(

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

)

import pagure.lib.query

import tests

from pagure.lib.repo import PagureRepo

FULL_ISSUE_LIST = [

    {

        "assignee": None,

        "blocks": [],

        "close_status": None,

        "closed_at": None,

        "closed_by": None,

        "comments": [],

        "content": "We should work on this",

        "custom_fields": [],

        "date_created": "1431414800",

        "depends": [],

        "id": 8,

        "last_updated": "1431414800",

        "milestone": None,

        "priority": None,

        "private": True,

        "status": "Open",

        "tags": [],

        "title": "Test issue",

        "user": {"fullname": "PY C", "name": "pingou"},

    },

    {

        "assignee": None,

        "blocks": [],

        "close_status": None,

        "closed_at": None,

        "closed_by": None,

        "comments": [],

        "content": "This issue needs attention",

        "custom_fields": [],

        "date_created": "1431414800",

        "depends": [],

        "id": 7,

        "last_updated": "1431414800",

        "milestone": None,

        "priority": None,

        "private": True,

        "status": "Open",

        "tags": [],

        "title": "test issue",

        "user": {"fullname": "PY C", "name": "pingou"},

    },

    {

        "assignee": None,

        "blocks": [],

        "close_status": None,

        "closed_at": None,

        "closed_by": None,

        "comments": [],

        "content": "This issue needs attention",

        "custom_fields": [],

        "date_created": "1431414800",

        "depends": [],

        "id": 6,

        "last_updated": "1431414800",

        "milestone": None,

        "priority": None,

        "private": True,

        "status": "Open",

        "tags": [],

        "title": "test issue",

        "user": {"fullname": "PY C", "name": "pingou"},

    },

    {

        "assignee": None,

        "blocks": [],

        "close_status": None,

        "closed_at": None,

        "closed_by": None,

        "comments": [],

        "content": "This issue needs attention",

        "custom_fields": [],

        "date_created": "1431414800",

        "depends": [],

        "id": 5,

        "last_updated": "1431414800",

        "milestone": None,

        "priority": None,

        "private": False,

        "status": "Open",

        "tags": [],

        "title": "test issue",

        "user": {"fullname": "PY C", "name": "pingou"},

    },

    {

        "assignee": None,

        "blocks": [],

        "close_status": None,

        "closed_at": None,

        "closed_by": None,

        "comments": [],

        "content": "This issue needs attention",

        "custom_fields": [],

        "date_created": "1431414800",

        "depends": [],

        "id": 4,

        "last_updated": "1431414800",

        "milestone": None,

        "priority": None,

        "private": False,

        "status": "Open",

        "tags": [],

        "title": "test issue",

        "user": {"fullname": "PY C", "name": "pingou"},

    },

    {

        "assignee": None,

        "blocks": [],

        "close_status": None,

        "closed_at": None,

        "closed_by": None,

        "comments": [],

        "content": "This issue needs attention",

        "custom_fields": [],

        "date_created": "1431414800",

        "depends": [],

        "id": 3,

        "last_updated": "1431414800",

        "milestone": None,

        "priority": None,

        "private": False,

        "status": "Open",

        "tags": [],

        "title": "test issue",

        "user": {"fullname": "PY C", "name": "pingou"},

    },

    {

        "assignee": None,

        "blocks": [],

        "close_status": None,

        "closed_at": None,

        "closed_by": None,

        "comments": [],

        "content": "This issue needs attention",

        "custom_fields": [],

        "date_created": "1431414800",

        "depends": [],

        "id": 2,

        "last_updated": "1431414800",

        "milestone": None,

        "priority": None,

        "private": False,

        "status": "Open",

        "tags": [],

        "title": "test issue",

        "user": {"fullname": "PY C", "name": "pingou"},

    },

    {

        "assignee": None,

        "blocks": [],

        "close_status": None,

        "closed_at": None,

        "closed_by": None,

        "comments": [],

        "content": "This issue needs attention",

        "custom_fields": [],

        "date_created": "1431414800",

        "depends": [],

        "id": 1,

        "last_updated": "1431414800",

        "milestone": None,

        "priority": None,

        "private": False,

        "status": "Open",

        "tags": [],

        "title": "test issue",

        "user": {"fullname": "PY C", "name": "pingou"},

    },

]

class PagurePrivateRepotest(tests.Modeltests):

    """ Tests for private repo in pagure """

    maxDiff = None

    def setUp(self):

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

        super(PagurePrivateRepotest, self).setUp()

        pagure.config.config["TESTING"] = True

        pagure.config.config["DATAGREPPER_URL"] = None

        pagure.config.config["PRIVATE_PROJECTS"] = True

        pagure.config.config["VIRUS_SCAN_ATTACHMENTS"] = False

    def set_up_git_repo(

        self, new_project=None, branch_from="feature", mtype="FF"

    ):

        """ Set up the git repo and create the corresponding PullRequest

        object.

        """

        # Create a git repo to play with

        gitrepo = os.path.join(self.path, "repos", "pmc.git")

        repo = pygit2.init_repository(gitrepo, bare=True)

        newpath = tempfile.mkdtemp(prefix="pagure-private-test")

        repopath = os.path.join(newpath, "test")

        clone_repo = pygit2.clone_repository(gitrepo, repopath)

        # Create a file in that git repo

        with open(os.path.join(repopath, "sources"), "w") as stream:

            stream.write("foo\n bar")

        clone_repo.index.add("sources")

        clone_repo.index.write()

        # Commits the files added

        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 sources file for testing",

            # binary string representing the tree object ID

            tree,

            # list of binary strings representing parents of the new commit

            [],

        )

        refname = "refs/heads/master:refs/heads/master"

        ori_remote = clone_repo.remotes[0]

        PagureRepo.push(ori_remote, refname)

        first_commit = repo.revparse_single("HEAD")

        if mtype == "merge":

            with open(os.path.join(repopath, ".gitignore"), "w") as stream:

                stream.write("*~")

            clone_repo.index.add(".gitignore")

            clone_repo.index.write()

            # Commits the files added

            tree = clone_repo.index.write_tree()

            author = pygit2.Signature("Alice Äuthòr", "alice@äuthòrs.tld")

            committer = pygit2.Signature(

                "Cecil Cõmmîttër", "cecil@cõmmîttërs.tld"

            )

            clone_repo.create_commit(

                "refs/heads/master",

                author,

                committer,

                "Add .gitignore file for testing",

                # binary string representing the tree object ID

                tree,

                # list of binary strings representing parents of the new commit

                [first_commit.oid.hex],

            )

            refname = "refs/heads/master:refs/heads/master"

            ori_remote = clone_repo.remotes[0]

            PagureRepo.push(ori_remote, refname)

        if mtype == "conflicts":

            with open(os.path.join(repopath, "sources"), "w") as stream:

                stream.write("foo\n bar\nbaz")

            clone_repo.index.add("sources")

            clone_repo.index.write()

            # Commits the files added

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

                author,

                committer,

                "Add sources conflicting",

                # binary string representing the tree object ID

                tree,

                # list of binary strings representing parents of the new commit

                [first_commit.oid.hex],

            )

            refname = "refs/heads/master:refs/heads/master"

            ori_remote = clone_repo.remotes[0]

            PagureRepo.push(ori_remote, refname)

        # Set the second repo

        new_gitrepo = repopath

        if new_project:

            # Create a new git repo to play with

            new_gitrepo = os.path.join(newpath, new_project.fullname)

            if not os.path.exists(new_gitrepo):

                os.makedirs(new_gitrepo)

                new_repo = pygit2.clone_repository(gitrepo, new_gitrepo)

        repo = pygit2.Repository(new_gitrepo)

        if mtype != "nochanges":

            # Edit the sources file again

            with open(os.path.join(new_gitrepo, "sources"), "w") as stream:

                stream.write("foo\n bar\nbaz\n boose")

            repo.index.add("sources")

            repo.index.write()

            # Commits the files added

            tree = repo.index.write_tree()

            author = pygit2.Signature("Alice Author", "alice@authors.tld")

            committer = pygit2.Signature(

                "Cecil Committer", "cecil@committers.tld"

            )

            repo.create_commit(

                "refs/heads/%s" % branch_from,

                author,

                committer,

                "A commit on branch %s" % branch_from,

                tree,

                [first_commit.oid.hex],

            )

            refname = "refs/heads/%s" % (branch_from)

            ori_remote = repo.remotes[0]

            PagureRepo.push(ori_remote, refname)

        # Create a PR for these changes

        project = pagure.lib.query._get_project(self.session, "pmc")

        req = pagure.lib.query.new_pull_request(

            session=self.session,

            repo_from=project,

            branch_from=branch_from,

            repo_to=project,

            branch_to="master",

            title="PR from the %s branch" % branch_from,

            user="pingou",

        )

        self.session.commit()

        self.assertEqual(req.id, 1)

        self.assertEqual(req.title, "PR from the %s branch" % branch_from)

        shutil.rmtree(newpath)

    def test_index(self):

        """ Test the index endpoint. """

        output = self.app.get("/")

        self.assertEqual(output.status_code, 200)

        self.assertIn(

            '
All Projects '

            '0',

            output.get_data(as_text=True),

        )

        # Add a private project

        item = pagure.lib.model.Project(

            user_id=2,  # foo

            name="test3",

            description="test project description",

            hook_token="aaabbbeee",

            private=True,

        )

        self.session.add(item)

        # Add a public project

        item = pagure.lib.model.Project(

            user_id=2,  # foo

            name="test4",

            description="test project description",

            hook_token="aaabbbeeeccceee",

        )

        self.session.add(item)

        self.session.commit()

        output = self.app.get("/?page=abc")

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn(

            '
All Projects '

            '1',

            output_text,

        )

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

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

            output = self.app.get("/", follow_redirects=True)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '
My Projects
',

                output_text,

            )

            self.assertIn("2 Projects", output_text)

            self.assertNotIn(

                'Forks', output_text

            )

            self.assertEqual(

                output_text.count('Groups'), 0

            )

    def test_view_user(self):

        """ Test the view_user endpoint. """

        output = self.app.get("/user/foo?repopage=abc&forkpage=def")

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn(

            """

                Projects 

                  0

              """,

            output_text,

        )

        self.assertIn(

            """

                Forks 

                  0

              """,

            output_text,

        )

        self.assertIn(

            """

                Groups 

                  0

              """,

            output_text,

        )

        # Add a private project

        item = pagure.lib.model.Project(

            user_id=2,  # foo

            name="test3",

            description="test project description",

            hook_token="aaabbbeee",

            private=True,

        )

        self.session.add(item)

        # Add a public project

        item = pagure.lib.model.Project(

            user_id=2,  # foo

            name="test4",

            description="test project description",

            hook_token="aaabbbeeeccceee",

        )

        self.session.add(item)

        self.session.commit()

        self.gitrepos = tests.create_projects_git(

            pagure.config.config["GIT_FOLDER"]

        )

        output = self.app.get("/user/foo")

        self.assertEqual(output.status_code, 200)

        output_text = output.get_data(as_text=True)

        self.assertIn(

            """

                Projects 

                  1

              """,

            output_text,

        )

        self.assertIn(

            """

                Forks 

                  0

              """,

            output_text,

        )

        self.assertIn(

            """

                Groups 

                  0

              """,

            output_text,

        )

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

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

            output = self.app.get("/user/foo")

            output_text = output.get_data(as_text=True)

            self.assertIn(

                """

                Projects 

                  1

              """,

                output_text,

            )

            self.assertIn(

                """

                Forks 

                  0

              """,

                output_text,

            )

            self.assertIn(

                """

                Groups 

                  0

              """,

                output_text,

            )

        user.username = "pingou"

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

            output = self.app.get("/user/foo")

            output_text = output.get_data(as_text=True)

            self.assertIn(

                """

                Projects 

                  1

              """,

                output_text,

            )

            self.assertIn(

                """

                Forks 

                  0

              """,

                output_text,

            )

            self.assertIn(

                """

                Groups 

                  0

              """,

                output_text,

            )

        # Check pingou has 0 projects

        user.username = "pingou"

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

            output = self.app.get("/", follow_redirects=True)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '
My Projects
',

                output_text,

            )

            self.assertIn("0 Projects", output_text)

            self.assertNotIn(

                'Forks', output_text

            )

            self.assertEqual(

                output_text.count('Groups'), 0

            )

        repo = pagure.lib.query._get_project(self.session, "test3")

        msg = pagure.lib.query.add_user_to_project(

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

        )

        self.session.commit()

        self.assertEqual(msg, "User added")

        # New user added to private projects

        user.username = "pingou"

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

            output = self.app.get("/", follow_redirects=True)

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '
My Projects
',

                output_text,

            )

            self.assertIn("1 Projects", output_text)

            self.assertNotIn(

                'Forks', output_text

            )

            self.assertEqual(

                output_text.count('Groups'), 0

            )

    @patch("pagure.decorators.admin_session_timedout")

    def test_private_settings_ui(self, ast):

        """ Test UI for private repo"""

        ast.return_value = False

        # Add private repo

        item = pagure.lib.model.Project(

            user_id=1,  # pingou

            name="test4",

            description="test project description",

            hook_token="aaabbbeeeceee",

            private=True,

        )

        self.session.add(item)

        self.session.commit()

        # Add a git repo

        repo_path = os.path.join(

            pagure.config.config.get("GIT_FOLDER"), "test4.git"

        )

        if not os.path.exists(repo_path):

            os.makedirs(repo_path)

        pygit2.init_repository(repo_path)

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

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

            tests.create_projects(self.session)

            tests.create_projects_git(pagure.config.config.get("GIT_FOLDER"))

            output = self.app.get("/test/settings")

            # Check for a public repo

            self.assertEqual(output.status_code, 200)

            self.assertNotIn(

                '

                output.get_data(as_text=True),

            )

            output = self.app.get("/test4/settings")

            # Check for private repo

            self.assertEqual(output.status_code, 200)

            self.assertIn(

                '<input checked="" name="private" type="checkbox" value="private">',

                output.get_data(as_text=True),

            )

            # Check the new project form has 'private' checkbox

            output = self.app.get("/new")

            self.assertEqual(output.status_code, 200)

            self.assertIn(

                '<input id="private" name="private" type="checkbox" value="y">',

                output.get_data(as_text=True),

            )

    @patch("pagure.decorators.admin_session_timedout")

    def test_private_settings_ui_update_privacy_false(self, ast):

        """ Test UI for private repo"""

        ast.return_value = False

        # Add private repo

        item = pagure.lib.model.Project(

            user_id=1,  # pingou

            name="test4",

            description="test project description",

            hook_token="aaabbbeeeceee",

            private=True,

        )

        self.session.add(item)

        self.session.commit()

        # Add a git repo

        repo_path = os.path.join(

            pagure.config.config.get("GIT_FOLDER"), "test4.git"

        )

        pygit2.init_repository(repo_path)

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

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

            # Check for private repo

            output = self.app.get("/test4/settings")

            self.assertEqual(output.status_code, 200)

            self.assertIn(

                '<input checked="" name="private" type="checkbox" value="private">',

                output.get_data(as_text=True),

            )

            self.session.commit()

            repo = pagure.lib.query._get_project(self.session, "test4")

            self.assertTrue(repo.private)

            # Make the project public

            data = {

                "description": "test project description",

                "private": False,

                "csrf_token": self.get_csrf(),

            }

            output = self.app.post(

                "/test4/update", data=data, follow_redirects=True

            )

            self.assertEqual(output.status_code, 200)

            self.assertIn("Project updated", output.get_data(as_text=True))

            self.assertNotIn(

                '<input checked="" name="private" type="checkbox" value="private">',

                output.get_data(as_text=True),

            )

            self.session.commit()

            repo = pagure.lib.query._get_project(self.session, "test4")

            self.assertFalse(repo.private)

    @patch("pagure.decorators.admin_session_timedout")

    def test_private_settings_ui_update_privacy_true(self, ast):

        """ Test UI for private repo"""

        ast.return_value = False

        # Add private repo

        item = pagure.lib.model.Project(

            user_id=1,  # pingou

            name="test4",

            description="test project description",

            hook_token="aaabbbeeeceee",

            private=False,

        )

        self.session.add(item)

        self.session.commit()

        # Add a git repo

        repo_path = os.path.join(

            pagure.config.config.get("GIT_FOLDER"), "test4.git"

        )

        pygit2.init_repository(repo_path)

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

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

            # Check for public repo

            output = self.app.get("/test4/settings")

            self.assertEqual(output.status_code, 200)

            self.assertNotIn(

                '<input checked="" name="private" type="checkbox" value="private">',

                output.get_data(as_text=True),

            )

            self.session.commit()

            repo = pagure.lib.query._get_project(self.session, "test4")

            self.assertFalse(repo.private)

            # Make the project private

            data = {

                "description": "test project description",

                "private": True,

                "csrf_token": self.get_csrf(),

            }

            output = self.app.post(

                "/test4/update", data=data, follow_redirects=True

            )

            self.assertEqual(output.status_code, 200)

            self.assertIn("Project updated", output.get_data(as_text=True))

            self.assertNotIn(

                '<input checked="" name="private" type="checkbox" value="private">',

                output.get_data(as_text=True),

            )

            # No change since we can't do public -> private

            self.session.commit()

            repo = pagure.lib.query._get_project(self.session, "test4")

            self.assertFalse(repo.private)

    @patch("pagure.lib.notify.send_email")

    def test_private_pr(self, send_email):

        """Test pull request made to the private repo"""

        send_email.return_value = True

        # Add a private project

        item = pagure.lib.model.Project(

            user_id=1,  # pingou

            name="pmc",

            description="test project description",

            hook_token="aaabbbeeeceee",

            private=True,

        )

        self.session.add(item)

        self.session.commit()

        repo = pagure.lib.query._get_project(self.session, "pmc")

        msg = pagure.lib.query.add_user_to_project(

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

        )

        self.session.commit()

        self.assertEqual(msg, "User added")

        # Create all the git repos

        tests.create_projects_git(

            os.path.join(self.path, "requests"), bare=True

        )

        # Add a git repo

        repo_path = os.path.join(

            pagure.config.config.get("REQUESTS_FOLDER"), "pmc.git"

        )

        if not os.path.exists(repo_path):

            os.makedirs(repo_path)

        pygit2.init_repository(repo_path, bare=True)

        # Check repo was created - Doesn't show on the public page

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

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

            output = self.app.get("/user/pingou/")

            self.assertEqual(output.status_code, 200)

            self.assertIn(

                """

                Projects 

                  0

              """,

                output.get_data(as_text=True),

            )