def test_request_reviews__token(self):
github = GitlabService()
with self.assertRaises(GitlabConnectionError) as context:
github.request_reviews(user_name=self.config['user_name'],
token=self.config['token'],
host=self.config['host'])
self.assertTrue('NewConnectionError' in str(context.exception))
def test_get_last_comment(self, mock_mr):
gitlab = GitlabService()
now = datetime.datetime.now()
mock_mr.notes.list.return_value = [
test_mock.FakeProjectMergeRequestNote(
system=True,
author="Gitlab",
created_at=now.strftime("%Y-%m-%dT%H:%M:%S.%fZ"),
body="System comment",
),
test_mock.FakeProjectMergeRequestNote(
system=False,
author="user",
created_at=now.strftime("%Y-%m-%dT%H:%M:%S.%fZ"),
body="last comment by user",
),
test_mock.FakeProjectMergeRequestNote(
system=False,
author="user2",
created_at=(now - datetime.timedelta(minutes=1)).strftime(
"%Y-%m-%dT%H:%M:%S.%fZ"
),
body="user comment",
),
]
last_comment = gitlab.get_last_comment(mock_mr)
self.assertEqual(
last_comment,
LastComment(
author="user", body="last comment by user", created_at=now
),
)
def test_request_reviews_with_repo_success(self,
mock_get_reviews,
mock_gitlab):
"""
Tests 'request_reviews' function where we have repos and no errors
"""
# Set up mock return values and side effects
mock_gitlab_instance = MagicMock(name='mock_gitlab_instance')
mock_gitlab_instance.projects.get.return_value = 'dummy_project'
mock_gitlab.return_value = mock_gitlab_instance
mock_get_reviews.return_value = '1'
# Call function
response = GitlabService().request_reviews(
user_name='dummy_user',
token='dummy_token',
host='dummy.com',
ssl_verify=True,
repo_name='dummy_repo'
)
# Validate function calls and response
mock_gitlab.assert_called_with('dummy.com', 'dummy_token', ssl_verify=True)
mock_gitlab_instance.groups.get.assert_not_called()
mock_gitlab_instance.auth.assert_called_once()
mock_gitlab_instance.projects.get.assert_called_with('dummy_user/dummy_repo')
mock_get_reviews.assert_called_with(
uname='dummy_user',
project='dummy_project',
age=None,
show_last_comment=None
)
self.assertEqual(['1'], response)
def test_request_reviews_GitlabGetError_with_repo(self,
mock_get_reviews,
mock_gitlab):
"""
Tests 'request_reviews' function where we have repos
and gl.projects.get throws an exception
"""
# Set up mock return values and side effects
mock_gitlab_instance = MagicMock(name='mock_gitlab_instance')
mock_gitlab_instance.projects.get.side_effect = GitlabGetError
mock_gitlab.return_value = mock_gitlab_instance
# Call function
with self.assertRaises(Exception):
GitlabService().request_reviews(
user_name='dummy_user',
token='dummy_token',
host='dummy.com',
ssl_verify=True,
repo_name='dummy_repo'
)
# Validate function calls and response
mock_gitlab.assert_called_with('dummy.com', 'dummy_token', ssl_verify=True)
mock_gitlab_instance.groups.get.assert_not_called()
mock_gitlab_instance.auth.assert_called_once()
mock_gitlab_instance.projects.get.assert_called_with('dummy_user/dummy_repo')
mock_get_reviews.assert_not_called()
def test_request_reviews_with_repo(self, mock_auth, mock_gitlab, mock_get_reviews):
mock_gitlab().projects.get.side_effect = test_mock.mock_projects_get_()
res = GitlabService().request_reviews(user_name=self.config['user_name'],
repo_name=self.config['repo_name'],
token=self.config['token'],
host=self.config['host'])
self.assertEqual([self.config['msg']], res)
def test_request_reviews_token(self):
with self.assertRaises(Exception) as context:
GitlabService().request_reviews(user_name=self.config['user_name'],
token=self.config['token'],
host=self.config['host'])
self.assertTrue('NewConnectionError' in str(context.exception), msg=context.exception)
def test_get_reviews_GitlabListError(self, mock_GitlabReview,
mock_has_new_comments,
mock_check_request_state,
mock_get_last_comment):
"""
Test 'get_reviews' where getting merge requests throws GitlabListError
"""
# Set up mock return values and side effects
self.mock_project.mergerequests.list.side_effect = GitlabListError()
# Call function
response = GitlabService().get_reviews(
uname='dummy_user',
project=self.mock_project,
)
# Validate function calls and response
self.mock_project.mergerequests.list.assert_called_with(
project_id=1,
state='opened'
)
mock_GitlabReview.assert_not_called()
mock_has_new_comments.assert_not_called()
mock_check_request_state.assert_not_called()
mock_get_last_comment.assert_not_called()
self.assertEqual(response, [])
def test_request_reviews_no_group_no_repo(self,
mock_get_reviews,
mock_gitlab):
"""
Test 'request_reviews' function where gl.groups.get returns None
"""
# Set up mock return values and side effects
mock_gitlab_instance = MagicMock(name='mock_gitlab_instance')
mock_gitlab_instance.projects.get.return_value = 'dummy_project'
mock_gitlab_instance.groups.get.return_value = None
mock_gitlab.return_value = mock_gitlab_instance
# Call function
with self.assertRaises(Exception):
GitlabService().request_reviews(
user_name='dummy_user',
token='dummy_token',
host='dummy.com',
ssl_verify=True
)
# Validate function calls and response
mock_gitlab.assert_called_with('dummy.com', 'dummy_token', ssl_verify=True)
mock_gitlab_instance.auth.assert_called_once()
mock_get_reviews.assert_not_called()
mock_gitlab_instance.groups.get.assert_called_with('dummy_user')
mock_gitlab_instance.projects.get.assert_not_called()
def test_request_reviews_groups_search(self, mock_auth, mock_gitlab):
with self.assertRaises(Exception) as context:
mock_gitlab().groups.get.return_value = []
GitlabService().request_reviews(user_name=self.config['user_name'],
token=self.config['token'],
host=self.config['host'])
msg = 'Invalid user/group name: %s' % self.config['user_name']
self.assertTrue(msg in str(context.exception))
def test_request_reviews_with_repo(self, mock_projects_get_,
mock_gitlab_get_reviews, mock_auth):
res = GitlabService().request_reviews(
user_name=self.config['user_name'],
repo_name=self.config['repo_name'],
token=self.config['token'],
host=self.config['host'])
self.assertEqual([self.config['msg']], res)
def test_request_reviews_projects_get(self, mock_projects_get, mock_auth):
with self.assertRaises(Exception) as context:
GitlabService().request_reviews(user_name=self.config['user_name'],
repo_name=self.config['repo_name'],
token=self.config['token'],
host=self.config['host'])
msg = 'Project %s not found for user %s' % (self.config['repo_name'],
self.config['user_name'])
self.assertTrue(msg in str(context.exception))
def test_request_reviews_with_repo(self, mock_auth, mock_gitlab,
mock_get_reviews):
mock_gitlab().projects.get.side_effect = test_mock.mock_projects_get_()
res = GitlabService().request_reviews(
user_name=self.config["user_name"],
repo_name=self.config["repo_name"],
token=self.config["token"],
host=self.config["host"],
)
self.assertEqual([self.config["msg"]], res)
def test_request_reviews_token(self):
with self.assertRaises(Exception) as context:
GitlabService().request_reviews(
user_name=self.config["user_name"],
token=self.config["token"],
host=self.config["host"],
)
self.assertTrue(
"NewConnectionError" in str(context.exception),
msg=context.exception,
)
def test_get_reviews_ValueError(self, mock_GitlabReview,
mock_has_new_comments,
mock_check_request_state,
mock_get_last_comment):
"""
Test 'get_reviews' where we have no age and no last comment
and datetime.strptime throws a ValueError
"""
# Set up mock return values and side effects
self.mock_mr.updated_at = '2010-10-04T03:41:22Z'
self.mock_mr.created_at = '2010-10-04T03:41:22Z'
expected_date = datetime.strptime(
'2010-10-04T03:41:22Z', '%Y-%m-%dT%H:%M:%SZ')
self.mock_project.mergerequests.list.return_value = [self.mock_mr]
mock_get_last_comment.return_value = self.mock_last_comment
mock_check_request_state.return_value = True
mock_GitlabReview.return_value = "Successful Call!"
mock_GitlabReview.logo = 'dummy_logo'
# Call function
response = GitlabService().get_reviews(
uname='dummy_user',
project=self.mock_project,
)
# Validate function calls and response
self.mock_project.mergerequests.list.assert_called_with(
project_id=1,
state='opened'
)
mock_check_request_state.assert_called_with(
expected_date,
None,
)
mock_get_last_comment.assert_called_with(
self.mock_mr
)
mock_GitlabReview.assert_called_with(
user='******',
title='dummy_title',
url='dummy_url',
time=expected_date,
updated_time=expected_date,
comments=1,
image='dummy_logo',
last_comment=self.mock_last_comment,
project_name='dummy_project',
project_url='dummy_url'
)
mock_has_new_comments.assert_not_called()
self.assertEqual(response, ["Successful Call!"])
def test_get_last_comment_with_note_system(self, mock_LastComment):
"""
Test 'get_last_comment' function with all notes in
mr.notes.list() not having note.system
"""
# Set up mock return values and side effects
mock_merge_requests = MagicMock()
mock_merge_requests.notes.list().return_value = ['test']
# Call function
response = GitlabService().get_last_comment(mock_merge_requests)
# Validate function calls and response
mock_LastComment.assert_not_called()
self.assertEqual(None, response)
def test_request_reviews_projects_get(self, mock_auth, mock_gitlab):
with self.assertRaises(Exception) as context:
mock_gitlab().projects.get.side_effect = (
test_mock.mock_projects_get())
GitlabService().request_reviews(
user_name=self.config["user_name"],
repo_name=self.config["repo_name"],
token=self.config["token"],
host=self.config["host"],
)
msg = "Project %s not found for user %s" % (
self.config["repo_name"],
self.config["user_name"],
)
self.assertTrue(msg in str(context.exception),
msg=context.exception)
def get_git_service(git):
"""
Returns git service as per requested.
Args:
git (str): String indicating git service requested.
Returns:
Returns desired git service
"""
if git == "github":
return GithubService()
elif git == "gitlab":
return GitlabService()
elif git == "pagure":
return PagureService()
elif git == "gerrit":
return GerritService()
else:
raise ValueError('requested git service %s is not valid' % (git))
def test_get_reviews_no_age_with_last_comment(self, mock_GitlabReview,
mock_has_new_comments,
mock_check_request_state,
mock_get_last_comment):
"""
Test 'get_reviews' where we have no age and with last comment
"""
# Set up mock return values and side effects
expected_date = datetime.strptime(
self.mock_mr.updated_at, '%Y-%m-%dT%H:%M:%S.%fZ')
self.mock_project.mergerequests.list.return_value = [self.mock_mr]
mock_get_last_comment.return_value = self.mock_last_comment
mock_check_request_state.return_value = True
mock_GitlabReview.return_value = "Successful Call!"
mock_GitlabReview.logo = 'dummy_logo'
# Call function
response = GitlabService().get_reviews(
uname='dummy_user',
project=self.mock_project,
show_last_comment=True
)
# Validate function calls and response
self.mock_project.mergerequests.list.assert_called_with(
project_id=1,
state='opened'
)
mock_check_request_state.assert_called_with(
expected_date,
None,
)
mock_get_last_comment.assert_called_with(
self.mock_mr
)
mock_GitlabReview.assert_not_called()
mock_has_new_comments.assert_called_with(
'dummy_created_at', True
)
self.assertEqual(response, [])
def test_get_reviews_with_age_no_last_comment(self, mock_GitlabReview,
mock_has_new_comments,
mock_check_request_state,
mock_get_last_comment):
"""
Test 'get_reviews' where we have a age and no last comment
"""
# Set up mock return values and side effects
expected_date = datetime.strptime(
self.mock_mr.updated_at, '%Y-%m-%dT%H:%M:%S.%fZ')
self.mock_project.mergerequests.list.return_value = [self.mock_mr]
mock_age = MagicMock()
mock_age.state = 'mock_state '
mock_get_last_comment.return_value = 'last_comment'
mock_check_request_state.return_value = False
mock_GitlabReview.logo = 'dummy_logo'
# Call function
response = GitlabService().get_reviews(
uname='dummy_user',
project=self.mock_project,
age=mock_age
)
# Validate function calls and response
self.mock_project.mergerequests.list.assert_called_with(
project_id=1,
state='opened'
)
mock_check_request_state.assert_called_with(
expected_date,
mock_age,
)
mock_get_last_comment.assert_called_with(
self.mock_mr
)
mock_GitlabReview.assert_not_called()
mock_has_new_comments.assert_not_called()
self.assertEqual(response, [])
def test_get_last_comment_no_note_system(self, mock_LastComment):
"""
Test 'get_last_comment' function with all notes
in mr.notes.list() having note.system
"""
# Set up mock return values and side effects
mock_merge_requests = MagicMock(name="mock_merge_requests")
created_at_expected_response = datetime.strptime(
self.mock_merge_request.created_at, '%Y-%m-%dT%H:%M:%S.%fZ')
mock_LastComment.return_value = 'Successful Call!'
mock_merge_requests.notes.list.return_value = [self.mock_merge_request]
# Call function
response = GitlabService().get_last_comment(mock_merge_requests)
# Validate function calls and response
mock_LastComment.assert_called_with(
author='dummy_author',
body='dummy_body',
created_at=created_at_expected_response
)
self.assertEqual('Successful Call!', response)
def test_request_reviews_SSLError_no_repo(self,
mock_get_reviews,
mock_gitlab):
"""
Test 'request_reviews' function where there is an SSL error and no repos
"""
# Set up mock return values and side effects
mock_gitlab_group = MagicMock(name='mock_gitlab_group')
mock_gitlab_group.id = 1
mock_gitlab_group_projects = MagicMock()
mock_gitlab_group_projects.projects.list.return_value = [mock_gitlab_group]
mock_gitlab_instance = MagicMock(name='mock_gitlab_instance')
mock_gitlab_instance.groups.get.return_value = mock_gitlab_group_projects
mock_gitlab_instance.projects.get.return_value = 'dummy_project'
mock_gitlab_instance.auth.side_effect = SSLError
mock_get_reviews.return_value = '1'
mock_gitlab.return_value = mock_gitlab_instance
# Call function
response = GitlabService().request_reviews(
user_name='dummy_user',
token='dummy_token',
host='dummy.com',
ssl_verify=True
)
# Validate function calls and response
mock_gitlab.assert_called_with('dummy.com', 'dummy_token', ssl_verify=True)
mock_gitlab_instance.auth.assert_called_once()
mock_gitlab_instance.groups.get.assert_called_with('dummy_user')
mock_gitlab_instance.projects.get.assert_called_with(1)
mock_get_reviews.assert_called_with(
uname='dummy_user',
project='dummy_project',
age=None)
self.assertEqual(['1'], response)
