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

"""

 (c) 2015 - Copyright Red Hat Inc

 Authors:

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

"""

__requires__ = ['SQLAlchemy >= 0.8']

import pkg_resources

import json

import unittest

import shutil

import sys

import os

import tempfile

import pygit2

from mock import patch

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

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

import pagure.lib.git

import tests

from pagure.lib.repo import PagureRepo

class PagureLibGittests(tests.Modeltests):

    """ Tests for pagure.lib.git """

    def setUp(self):

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

        super(PagureLibGittests, self).setUp()

        pagure.lib.git.SESSION = self.session

        pagure.APP.config['GIT_FOLDER'] = os.path.join(

            self.path, 'repos')

        pagure.APP.config['TICKETS_FOLDER'] = os.path.join(

            self.path, 'tickets')

        pagure.APP.config['DOCS_FOLDER'] = os.path.join(

            self.path, 'docs')

        pagure.APP.config['REQUESTS_FOLDER'] = os.path.join(

            self.path, 'requests')

    def test_write_gitolite_acls(self):

        """ Test the write_gitolite_acls function of pagure.lib.git. """

        tests.create_projects(self.session)

        repo = pagure.lib.get_project(self.session, 'test')

        # Add an user to a project

        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')

        # Add a forked project

        item = pagure.lib.model.Project(

            user_id=1,  # pingou

            name='test3',

            description='test project #2',

            is_fork=True,

            parent_id=1,

            hook_token='aaabbbvvv',

        )

        self.session.add(item)

        self.session.commit()

        outputconf = os.path.join(self.path, 'test_gitolite.conf')

        pagure.lib.git.write_gitolite_acls(self.session, outputconf)

        self.assertTrue(os.path.exists(outputconf))

        with open(outputconf) as stream:

            data = stream.read()

        exp = """

repo test

  R   = @all

  RW+ = pingou

  RW+ = foo

repo docs/test

  R   = @all

  RW+ = pingou

  RW+ = foo

repo tickets/test

  RW+ = pingou

  RW+ = foo

repo requests/test

  RW+ = pingou

  RW+ = foo

repo test2

  R   = @all

  RW+ = pingou

repo docs/test2

  R   = @all

  RW+ = pingou

repo tickets/test2

  RW+ = pingou

repo requests/test2

  RW+ = pingou

repo somenamespace/test3

  R   = @all

  RW+ = pingou

repo docs/somenamespace/test3

  R   = @all

  RW+ = pingou

repo tickets/somenamespace/test3

  RW+ = pingou

repo requests/somenamespace/test3

  RW+ = pingou

repo forks/pingou/test3

  R   = @all

  RW+ = pingou

repo docs/forks/pingou/test3

  R   = @all

  RW+ = pingou

repo tickets/forks/pingou/test3

  RW+ = pingou

repo requests/forks/pingou/test3

  RW+ = pingou

"""

        #print data

        self.assertEqual(data, exp)

        os.unlink(outputconf)

        self.assertFalse(os.path.exists(outputconf))

    def test_write_gitolite_acls_groups(self):

        """ Test the write_gitolite_acls function of pagure.lib.git with

        groups.

        """

        tests.create_projects(self.session)

        repo = pagure.lib.get_project(self.session, 'test')

        # Add a couple of groups

        msg = pagure.lib.add_group(

            self.session,

            group_name='sysadmin',

            display_name='sysadmin group',

            description=None,

            group_type='user',

            user='pingou',

            is_admin=False,

            blacklist=[],

        )

        self.session.commit()

        self.assertEqual(msg, 'User `pingou` added to the group `sysadmin`.')

        msg = pagure.lib.add_group(

            self.session,

            group_name='devs',

            display_name='devs group',

            description=None,

            group_type='user',

            user='pingou',

            is_admin=False,

            blacklist=[],

        )

        self.session.commit()

        self.assertEqual(msg, 'User `pingou` added to the group `devs`.')

        # Associate these groups to a project

        msg = pagure.lib.add_group_to_project(

            session=self.session,

            project=repo,

            new_group='sysadmin',

            user='pingou',

        )

        self.session.commit()

        self.assertEqual(msg, 'Group added')

        msg = pagure.lib.add_group_to_project(

            session=self.session,

            project=repo,

            new_group='devs',

            user='pingou',

        )

        self.session.commit()

        self.assertEqual(msg, 'Group added')

        # Add an user to a project

        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')

        # Add a forked project

        item = pagure.lib.model.Project(

            user_id=1,  # pingou

            name='test2',

            description='test project #2',

            is_fork=True,

            parent_id=1,

            hook_token='aaabbbvvv',

        )

        self.session.add(item)

        self.session.commit()

        outputconf = os.path.join(self.path, 'test_gitolite.conf')

        pagure.lib.git.write_gitolite_acls(self.session, outputconf)

        self.assertTrue(os.path.exists(outputconf))

        with open(outputconf) as stream:

            data = stream.read()

        exp = """@sysadmin   = pingou

@devs   = pingou

repo test

  R   = @all

  RW+ = @sysadmin @devs

  RW+ = pingou

  RW+ = foo

repo docs/test

  R   = @all

  RW+ = @sysadmin @devs

  RW+ = pingou

  RW+ = foo

repo tickets/test

  RW+ = @sysadmin @devs

  RW+ = pingou

  RW+ = foo

repo requests/test

  RW+ = @sysadmin @devs

  RW+ = pingou

  RW+ = foo

repo test2

  R   = @all

  RW+ = pingou

repo docs/test2

  R   = @all

  RW+ = pingou

repo tickets/test2

  RW+ = pingou

repo requests/test2

  RW+ = pingou

repo somenamespace/test3

  R   = @all

  RW+ = pingou

repo docs/somenamespace/test3

  R   = @all

  RW+ = pingou

repo tickets/somenamespace/test3

  RW+ = pingou

repo requests/somenamespace/test3

  RW+ = pingou

repo forks/pingou/test2

  R   = @all

  RW+ = pingou

repo docs/forks/pingou/test2

  R   = @all

  RW+ = pingou

repo tickets/forks/pingou/test2

  RW+ = pingou

repo requests/forks/pingou/test2

  RW+ = pingou

"""

        #print data

        self.assertEqual(data.split('\n'), exp.split('\n'))

        os.unlink(outputconf)

        self.assertFalse(os.path.exists(outputconf))

    def test_commit_to_patch(self):

        """ Test the commit_to_patch function of pagure.lib.git. """

        # Create a git repo to play with

        self.gitrepo = os.path.join(self.path, 'test_repo.git')

        os.makedirs(self.gitrepo)

        repo = pygit2.init_repository(self.gitrepo)

        # Create a file in that git repo

        with open(os.path.join(self.gitrepo, 'sources'), 'w') as stream:

            stream.write('foo\n bar')

        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/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

            []

        )

        first_commit = repo.revparse_single('HEAD')

        # Edit the sources file again

        with open(os.path.join(self.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/master',  # the name of the reference to update

            author,

            committer,

            'Add baz and boose to the sources\n\n There are more objects to '

            'consider',

            # binary string representing the tree object ID

            tree,

            # list of binary strings representing parents of the new commit

            [first_commit.oid.hex]

        )

        second_commit = repo.revparse_single('HEAD')

        # Generate a patch for 2 commits

        patch = pagure.lib.git.commit_to_patch(

            repo, [first_commit, second_commit])

        exp = """Mon Sep 17 00:00:00 2001

From: Alice Author <alice@authors.tld></alice@authors.tld>

Subject: [PATCH 1/2] Add sources file for testing

---

diff --git a/sources b/sources

new file mode 100644

index 0000000..9f44358

--- /dev/null

+++ b/sources

@@ -0,0 +1,2 @@

+foo

+ bar

\ No newline at end of file

Mon Sep 17 00:00:00 2001

From: Alice Author <alice@authors.tld></alice@authors.tld>

Subject: [PATCH 2/2] Add baz and boose to the sources

 There are more objects to consider

---

diff --git a/sources b/sources

index 9f44358..2a552bb 100644

--- a/sources

+++ b/sources

@@ -1,2 +1,4 @@

 foo

- bar

\ No newline at end of file

+ bar

+baz

+ boose

\ No newline at end of file

"""

        npatch = []

        for row in patch.split('\n'):

            if row.startswith('Date:'):

                continue

            if row.startswith('From '):

                row = row.split(' ', 2)[2]

            npatch.append(row)

        patch = '\n'.join(npatch)

        self.assertEqual(patch, exp)

        # Generate a patch for a single commit

        patch = pagure.lib.git.commit_to_patch(repo, second_commit)

        exp = """Mon Sep 17 00:00:00 2001

From: Alice Author <alice@authors.tld></alice@authors.tld>

Subject: Add baz and boose to the sources

 There are more objects to consider

---

diff --git a/sources b/sources

index 9f44358..2a552bb 100644

--- a/sources

+++ b/sources

@@ -1,2 +1,4 @@

 foo

- bar

\ No newline at end of file

+ bar

+baz

+ boose

\ No newline at end of file

"""

        npatch = []

        for row in patch.split('\n'):

            if row.startswith('Date:'):

                continue

            if row.startswith('From '):

                row = row.split(' ', 2)[2]

            npatch.append(row)

        patch = '\n'.join(npatch)

        self.assertEqual(patch, exp)

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

    def test_update_git(self, email_f):

        """ Test the update_git of pagure.lib.git. """

        email_f.return_value = True

        # Create project

        item = pagure.lib.model.Project(

            user_id=1,  # pingou

            name='test_ticket_repo',

            description='test project for ticket',

            hook_token='aaabbbwww',

        )

        self.session.add(item)

        self.session.commit()

        # Create repo

        self.gitrepo = os.path.join(self.path, 'test_ticket_repo.git')

        os.makedirs(self.gitrepo)

        repo_obj = pygit2.init_repository(self.gitrepo, bare=True)

        repo = pagure.lib.get_project(self.session, 'test_ticket_repo')

        # Create an issue to play with

        msg = pagure.lib.new_issue(

            session=self.session,

            repo=repo,

            title='Test issue',

            content='We should work on this',

            user='pingou',

            ticketfolder=self.path

        )

        self.assertEqual(msg.title, 'Test issue')

        issue = pagure.lib.search_issues(self.session, repo, issueid=1)

        pagure.lib.git.update_git(issue, repo, self.path)

        repo = pygit2.Repository(self.gitrepo)

        commit = repo.revparse_single('HEAD')

        # Use patch to validate the repo

        patch = pagure.lib.git.commit_to_patch(repo, commit)

        exp = """Mon Sep 17 00:00:00 2001

From: pagure <pagure></pagure>

Subject: Updated issue <hash>: Test issue</hash>

---

diff --git a/123 b/456

new file mode 100644

index 0000000..60f7480

--- /dev/null

+++ b/456

@@ -0,0 +1,28 @@

+{

+    "assignee": null,

+    "blocks": [],

+    "close_status": null,

+    "closed_at": null,

+    "comments": [],

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

+    "custom_fields": [],

+    "date_created": null,

+    "depends": [],

+    "id": 1,

+    "last_updated": null,

+    "milestone": null,

+    "priority": null,

+    "private": false,

+    "status": "Open",

+    "tags": [],

+    "title": "Test issue",

+    "user": {

+        "default_email": "bar@pingou.com",

+        "emails": [

+            "bar@pingou.com",

+            "foo@pingou.com"

+        ],

+        "fullname": "PY C",

+        "name": "pingou"

+    }

+}

\ No newline at end of file

"""

        npatch = []

        for row in patch.split('\n'):

            if row.startswith('Date:'):

                continue

            elif row.startswith('From '):

                row = row.split(' ', 2)[2]

            elif row.startswith('diff --git '):

                row = row.split(' ')

                row[2] = 'a/123'

                row[3] = 'b/456'

                row = ' '.join(row)

            elif 'Updated issue' in row:

                row = row.split()

                row[3] = '<hash>:'</hash>

                row = ' '.join(row)

            elif 'date_created' in row:

                t = row.split(': ')[0]

                row = '%s: null,' % t

            elif 'last_updated' in row:

                t = row.split(': ')[0]

                row = '%s: null,' % t

            elif 'closed_at' in row:

                t = row.split(': ')[0]

                row = '%s: null,' % t

            elif row.startswith('index 00'):

                row = 'index 0000000..60f7480'

            elif row.startswith('+++ b/'):

                row = '+++ b/456'

            npatch.append(row)

        patch = '\n'.join(npatch)

        #print patch

        self.assertEqual(patch, exp)

        # Test again after adding a comment

        msg = pagure.lib.add_issue_comment(

            session=self.session,

            issue=issue,

            comment='Hey look a comment!',

            user='foo',

            ticketfolder=self.path

        )

        self.session.commit()

        self.assertEqual(msg, 'Comment added')

        # Use patch to validate the repo

        repo = pygit2.Repository(self.gitrepo)

        commit = repo.revparse_single('HEAD')

        patch = pagure.lib.git.commit_to_patch(repo, commit)

        exp = """Mon Sep 17 00:00:00 2001

From: pagure <pagure></pagure>

Subject: Updated issue <hash>: Test issue</hash>

---

diff --git a/123 b/456

index 458821a..77674a8

--- a/123

+++ b/456

@@ -3,7 +3,25 @@

     "blocks": [],

     "close_status": null,

     "closed_at": null,

-    "comments": [],

+    "comments": [

+        {

+            "comment": "Hey look a comment!",

+            "date_created": null,

+            "edited_on": null,

+            "editor": null,

+            "id": 1,

+            "notification": false,

+            "parent": null,

+            "user": {

+                "default_email": "foo@bar.com",

+                "emails": [

+                    "foo@bar.com"

+                ],

+                "fullname": "foo bar",

+                "name": "foo"

+            }

+        }

+    ],

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

     "custom_fields": [],

     "date_created": null,

"""

        npatch = []

        for row in patch.split('\n'):

            if row.startswith('Date:'):

                continue

            elif row.startswith('From '):

                row = row.split(' ', 2)[2]

            elif row.startswith('diff --git '):

                row = row.split(' ')

                row[2] = 'a/123'

                row[3] = 'b/456'

                row = ' '.join(row)

            elif 'Updated issue' in row:

                row = row.split()

                row[3] = '<hash>:'</hash>

                row = ' '.join(row)

            elif 'date_created' in row:

                t = row.split(': ')[0]

                row = '%s: null,' % t

            elif 'closed_at' in row:

                t = row.split(': ')[0]

                row = '%s: null,' % t

            elif row.startswith('index'):

                row = 'index 458821a..77674a8'

            elif row.startswith('--- a/'):

                row = '--- a/123'

            elif row.startswith('+++ b/'):

                row = '+++ b/456'

            npatch.append(row)

        patch = '\n'.join(npatch)

        #print patch

        self.assertEqual(patch, exp)

    def test_clean_git(self):

        """ Test the clean_git method of pagure.lib.git. """

        pagure.lib.git.clean_git(None, None, None)

        self.test_update_git()

        gitpath = os.path.join(self.path, 'test_ticket_repo.git')

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

        # Get the uid of the ticket created

        commit = gitrepo.revparse_single('HEAD')

        patch = pagure.lib.git.commit_to_patch(gitrepo, commit)

        hash_file = None

        for row in patch.split('\n'):

            if row.startswith('+++ b/'):

                hash_file = row.split('+++ b/')[-1]

                break

        # The only file in git is the one of that ticket

        files = [entry.name for entry in commit.tree]

        self.assertEqual(files, [hash_file])

        repo = pagure.lib.get_project(self.session, 'test_ticket_repo')

        issue = pagure.lib.search_issues(self.session, repo, issueid=1)

        pagure.lib.git.clean_git(issue, repo, self.path)

        # No more files in the git repo

        commit = gitrepo.revparse_single('HEAD')

        files = [entry.name for entry in commit.tree]

        self.assertEqual(files, [])

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

    def test_update_git_requests(self, email_f):

        """ Test the update_git of pagure.lib.git for pull-requests. """

        email_f.return_value = True

        # Create project

        item = pagure.lib.model.Project(

            user_id=1,  # pingou

            name='test_ticket_repo',

            description='test project for ticket',

            hook_token='aaabbbxxx',

        )

        self.session.add(item)

        self.session.commit()

        # Create repo

        self.gitrepo = os.path.join(self.path, 'test_ticket_repo.git')

        os.makedirs(self.gitrepo)

        repo_obj = pygit2.init_repository(self.gitrepo, bare=True)

        repo = pagure.lib.get_project(self.session, 'test_ticket_repo')

        # Create an issue to play with

        req = pagure.lib.new_pull_request(

            session=self.session,

            repo_from=repo,

            branch_from='feature',

            repo_to=repo,

            branch_to='master',

            title='test PR',

            user='pingou',

            requestfolder=self.path,

            requestuid='foobar',

            requestid=None,

            status='Open',

            notify=True

        )

        self.assertEqual(req.id, 1)

        self.assertEqual(req.title, 'test PR')

        request = repo.requests[0]

        self.assertEqual(request.title, 'test PR')

        pagure.lib.git.update_git(request, request.project, self.path)

        repo = pygit2.Repository(self.gitrepo)

        commit = repo.revparse_single('HEAD')

        # Use patch to validate the repo

        patch = pagure.lib.git.commit_to_patch(repo, commit)

        exp = """Mon Sep 17 00:00:00 2001

From: pagure <pagure></pagure>

Subject: Updated pull-request <hash>: test PR</hash>

---

diff --git a/123 b/456

new file mode 100644

index 0000000..60f7480

--- /dev/null

+++ b/456

@@ -0,0 +1,96 @@

+{

+    "assignee": null,

+    "branch": "master",

+    "branch_from": "feature",

+    "closed_at": null,

+    "closed_by": null,

+    "comments": [],

+    "commit_start": null,

+    "commit_stop": null,

+    "date_created": null,

+    "id": 1,

+    "initial_comment": null,

+    "last_updated": null,

+    "project": {

+        "close_status": [],

+        "custom_keys": [],

+        "date_created": null,

+        "description": "test project for ticket",

+        "id": 1,

+        "milestones": {},

+        "name": "test_ticket_repo",

+        "namespace": null,

+        "parent": null,

+        "priorities": {},

+        "settings": {

+            "Enforce_signed-off_commits_in_pull-request": false,

+            "Minimum_score_to_merge_pull-request": -1,

+            "Only_assignee_can_merge_pull-request": false,

+            "Web-hooks": null,

+            "always_merge": false,

+            "fedmsg_notifications": true,

+            "issue_tracker": true,

+            "issues_default_to_private": false,

+            "project_documentation": false,

+            "pull_requests": true

+        },

+        "tags": [],

+        "user": {

+            "default_email": "bar@pingou.com",

+            "emails": [

+                "bar@pingou.com",

+                "foo@pingou.com"

+            ],

+            "fullname": "PY C",

+            "name": "pingou"

+        }

+    },

+    "remote_git": null,

+    "repo_from": {

+        "close_status": [],

+        "custom_keys": [],

+        "date_created": null,

+        "description": "test project for ticket",

+        "id": 1,

+        "milestones": {},

+        "name": "test_ticket_repo",

+        "namespace": null,

+        "parent": null,

+        "priorities": {},

+        "settings": {

+            "Enforce_signed-off_commits_in_pull-request": false,

+            "Minimum_score_to_merge_pull-request": -1,

+            "Only_assignee_can_merge_pull-request": false,

+            "Web-hooks": null,

+            "always_merge": false,

+            "fedmsg_notifications": true,

+            "issue_tracker": true,

+            "issues_default_to_private": false,

+            "project_documentation": false,

+            "pull_requests": true

+        },

+        "tags": [],

+        "user": {

+            "default_email": "bar@pingou.com",

+            "emails": [

+                "bar@pingou.com",

+                "foo@pingou.com"

+            ],

+            "fullname": "PY C",

+            "name": "pingou"

+        }

+    },

+    "status": "Open",

+    "title": "test PR",

+    "uid": "foobar",

+    "updated_on": null,

+    "user": {

+        "default_email": "bar@pingou.com",

+        "emails": [

+            "bar@pingou.com",

+            "foo@pingou.com"

+        ],

+        "fullname": "PY C",

+        "name": "pingou"

+    }

+}

\ No newline at end of file

"""

        npatch = []

        for row in patch.split('\n'):

            if row.startswith('Date:'):

                continue

            elif row.startswith('From '):

                row = row.split(' ', 2)[2]

            elif row.startswith('diff --git '):

                row = row.split(' ')

                row[2] = 'a/123'

                row[3] = 'b/456'

                row = ' '.join(row)

            elif 'Updated pull-request' in row:

                row = row.split()

                row[3] = '<hash>:'</hash>

                row = ' '.join(row)

            elif 'date_created' in row:

                t = row.split(': ')[0]

                row = '%s: null,' % t

            elif 'last_updated' in row:

                t = row.split(': ')[0]

                row = '%s: null,' % t

            elif 'updated_on' in row:

                t = row.split(': ')[0]

                row = '%s: null,' % t

            elif row.startswith('index 00'):

                row = 'index 0000000..60f7480'

            elif row.startswith('+++ b/'):

                row = '+++ b/456'

            npatch.append(row)

        patch = '\n'.join(npatch)

        #print patch

        self.assertEqual(patch, exp)