pagure/tests/test_tasks.py

from mock import patch, 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.create_session', new=Mock())
class TestCommitsAuthorStats(unittest.TestCase):

    def setUp(self):
        self.search_user_patcher = patch('pagure.lib.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.assertEqual(authors, [(1, [('Alice', 'alice@example.com')])])

    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.assertEqual(authors, [(2, [('Alice', 'alice@example.com')])])

    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.assertEqual(authors, [(1, [('Alice', 'alice@example.com')])])

    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)]),
                                   (1, [('Bob', '')])])