pagure/tests/test_tasks.py

from __future__ import unicode_literals, absolute_import

from mock import patch, MagicMock, Mock
from collections import namedtuple
import os
import unittest

from pagure.lib import tasks


MockUser = namedtuple('MockUser', ['fullname', 'default_email'])


class MockCommit(object):
    def __init__(self, name, email, time='1970-01-01 00:00'):
        self.author = Mock(email=email)
        self.author.name = name
        self.commit_time = time


@patch('pagure.lib.query.create_session', new=Mock())
class TestCommitsAuthorStats(unittest.TestCase):

    def setUp(self):
        self.search_user_patcher = patch('pagure.lib.query.search_user')
        mock_search_user = self.search_user_patcher.start()
        mock_search_user.side_effect = lambda _, email: self.authors.get(email)

        self.pygit_patcher = patch('pygit2.Repository')
        mock_repo = self.pygit_patcher.start().return_value

        def mock_walk_impl(*args, **kwargs):
            for commit in self.commits:
                yield commit

        mock_repo.walk.side_effect = mock_walk_impl

        self.repopath = Mock()
        exists = os.path.exists

        def mock_exists_impl(path):
            if path == self.repopath:
                return True
            return exists(path)

        self.exists_patcher = patch('os.path.exists')
        mock_exists = self.exists_patcher.start()
        mock_exists.side_effect = mock_exists_impl

    def tearDown(self):
        self.search_user_patcher.stop()
        self.pygit_patcher.stop()
        self.exists_patcher.stop()

    def test_no_change(self):
        self.commits = [
            MockCommit('Alice', 'alice@example.com', '2018-01-01 00:00'),
        ]
        self.authors = {
            'alice@example.com': MockUser('Alice', 'alice@example.com'),
        }

        num_commits, authors, num_authors, last_time = \
            tasks.commits_author_stats(self.repopath)

        self.assertEqual(num_commits, 1)
        self.assertEqual(num_authors, 1)
        self.assertEqual(last_time, '2018-01-01 00:00')
        self.assertIn(
            authors,
            [
                [(1, [(
                'Alice', 'alice@example.com',
                'https://seccdn.libravatar.org/avatar/'
                'ff8d9819fc0e12bf0d24892e45987e249a28dce836a85cad60e28eaaa8c6d976'
                '?s=32&d=retro'
                )])],
                [(1, [(
                'Alice', 'alice@example.com',
                'https://seccdn.libravatar.org/avatar/'
                'ff8d9819fc0e12bf0d24892e45987e249a28dce836a85cad60e28eaaa8c6d976'
                '?d=retro&s=32'
                )])]
            ]
        )

    def test_rename_user_and_merge(self):
        self.commits = [
            MockCommit('Alice', 'alice@example.com'),
            MockCommit('Bad name', 'alice@example.com', '2018-01-01 00:00'),
        ]
        self.authors = {
            'alice@example.com': MockUser('Alice', 'alice@example.com'),
        }

        num_commits, authors, num_authors, last_time = \
            tasks.commits_author_stats(self.repopath)

        self.assertEqual(num_commits, 2)
        self.assertEqual(num_authors, 1)
        self.assertEqual(last_time, '2018-01-01 00:00')
        self.assertIn(
            authors,
            [
                [(2, [(
                'Alice', 'alice@example.com',
                'https://seccdn.libravatar.org/avatar/'
                'ff8d9819fc0e12bf0d24892e45987e249a28dce836a85cad60e28eaaa8c6d976'
                '?s=32&d=retro'
                )])],
                [(2, [(
                'Alice', 'alice@example.com',
                'https://seccdn.libravatar.org/avatar/'
                'ff8d9819fc0e12bf0d24892e45987e249a28dce836a85cad60e28eaaa8c6d976'
                '?d=retro&s=32'
                )])]
            ]
        )

    def test_preserve_unknown_author(self):
        self.commits = [
            MockCommit('Alice', 'alice@example.com', '2018-01-01 00:00'),
        ]
        self.authors = {}

        num_commits, authors, num_authors, last_time = \
            tasks.commits_author_stats(self.repopath)

        self.assertEqual(num_commits, 1)
        self.assertEqual(num_authors, 1)
        self.assertEqual(last_time, '2018-01-01 00:00')
        self.assertIn(
            authors,
            [
                [(1, [(
                'Alice', 'alice@example.com',
                'https://seccdn.libravatar.org/avatar/'
                'ff8d9819fc0e12bf0d24892e45987e249a28dce836a85cad60e28eaaa8c6d976'
                '?s=32&d=retro'
                )])],
                [(1, [(
                'Alice', 'alice@example.com',
                'https://seccdn.libravatar.org/avatar/'
                'ff8d9819fc0e12bf0d24892e45987e249a28dce836a85cad60e28eaaa8c6d976'
                '?d=retro&s=32'
                )])]
            ]
        )

    def test_handle_empty_email(self):
        self.commits = [
            # Two commits for Alice to ensure order of the result.
            MockCommit('Alice', None),
            MockCommit('Alice', None),
            MockCommit('Bob', '', '2018-01-01 00:00'),
        ]
        self.authors = {}

        num_commits, authors, num_authors, last_time = \
            tasks.commits_author_stats(self.repopath)

        self.assertEqual(num_commits, 3)
        self.assertEqual(num_authors, 2)
        self.assertEqual(last_time, '2018-01-01 00:00')
        self.assertEqual(authors, [(2, [('Alice', None, None)]),
                                   (1, [('Bob', '', None)])])


class TestGitolitePostCompileOnly(object):
    @patch('pagure.lib.git_auth.get_git_auth_helper')
    def test_backend_has_post_compile_only(self, get_helper):
        helper = MagicMock()
        get_helper.return_value = helper
        helper.post_compile_only = MagicMock()
        tasks.gitolite_post_compile_only()
        helper.post_compile_only.assert_called_once()

    @patch('pagure.lib.git_auth.get_git_auth_helper')
    def test_backend_doesnt_have_post_compile_only(self, get_helper):
        helper = MagicMock()
        get_helper.return_value = helper
        helper.generate_acls = MagicMock()
        del helper.post_compile_only
        tasks.gitolite_post_compile_only()
        helper.generate_acls.assert_called_once_with(project=None)