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

from __future__ import unicode_literals

__requires__ = ['SQLAlchemy >= 0.8']

import pkg_resources

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

import tests

from pagure.lib.repo import PagureRepo

FULL_ISSUE_LIST = [

    {

        "assignee": None,

        "blocks": [],

        "close_status": None,

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

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

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

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

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

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

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

        "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._get_project(self.session, 'pmc')

        req = pagure.lib.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._get_project(self.session, 'test3')

        msg = pagure.lib.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._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._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._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._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._get_project(self.session, 'pmc')

        msg = pagure.lib.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))

            self.assertIn(

                """

                Forks 

                  0

              """, output.get_data(as_text=True))

            # Shows on the front page

            output = self.app.get('/dashboard/projects')

            self.assertEqual(output.status_code, 200)

            self.assertIn(