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)