Show More
test_mako_emails.py
143 lines
| 4.8 KiB
| text/x-python
|
PythonLexer
r1 | import collections | |||
r2310 | # -*- coding: utf-8 -*- | |||
r2487 | # Copyright (C) 2010-2018 RhodeCode GmbH | |||
r2310 | # | |||
# This program is free software: you can redistribute it and/or modify | ||||
# it under the terms of the GNU Affero General Public License, version 3 | ||||
# (only), as published by the Free Software Foundation. | ||||
# | ||||
# This program is distributed in the hope that it will be useful, | ||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||||
# GNU General Public License for more details. | ||||
# | ||||
# You should have received a copy of the GNU Affero General Public License | ||||
# along with this program. If not, see <http://www.gnu.org/licenses/>. | ||||
# | ||||
# This program is dual-licensed. If you wish to learn more about the | ||||
# RhodeCode Enterprise Edition, including its added features, Support services, | ||||
# and proprietary license terms, please see https://rhodecode.com/licenses/ | ||||
r1 | ||||
import pytest | ||||
r2310 | from rhodecode.lib.partial_renderer import PyramidPartialRenderer | |||
r1728 | from rhodecode.lib.utils2 import AttributeDict | |||
r1 | from rhodecode.model.notification import EmailNotificationModel | |||
r2310 | def test_get_template_obj(app, request_stub): | |||
r1 | template = EmailNotificationModel().get_renderer( | |||
r2310 | EmailNotificationModel.TYPE_TEST, request_stub) | |||
assert isinstance(template, PyramidPartialRenderer) | ||||
r1 | ||||
r1774 | def test_render_email(app, http_host_only_stub): | |||
r1 | kwargs = {} | |||
subject, headers, body, body_plaintext = EmailNotificationModel().render_email( | ||||
EmailNotificationModel.TYPE_TEST, **kwargs) | ||||
# subject | ||||
assert subject == 'Test "Subject" hello "world"' | ||||
# headers | ||||
assert headers == 'X=Y' | ||||
# body plaintext | ||||
assert body_plaintext == 'Email Plaintext Body' | ||||
# body | ||||
r1774 | notification_footer = 'This is a notification from RhodeCode. http://%s/' \ | |||
% http_host_only_stub | ||||
assert notification_footer in body | ||||
r512 | assert 'Email Body' in body | |||
r1 | ||||
r1774 | def test_render_pr_email(app, user_admin): | |||
r1 | ||||
ref = collections.namedtuple('Ref', | ||||
'name, type')( | ||||
'fxies123', 'book' | ||||
) | ||||
pr = collections.namedtuple('PullRequest', | ||||
'pull_request_id, title, description, source_ref_parts, source_ref_name, target_ref_parts, target_ref_name')( | ||||
200, 'Example Pull Request', 'Desc of PR', ref, 'bookmark', ref, 'Branch') | ||||
source_repo = target_repo = collections.namedtuple('Repo', | ||||
'type, repo_name')( | ||||
'hg', 'pull_request_1') | ||||
kwargs = { | ||||
'user': '<marcin@rhodecode.com> Marcin Kuzminski', | ||||
'pull_request': pr, | ||||
'pull_request_commits': [], | ||||
'pull_request_target_repo': target_repo, | ||||
'pull_request_target_repo_url': 'x', | ||||
'pull_request_source_repo': source_repo, | ||||
'pull_request_source_repo_url': 'x', | ||||
'pull_request_url': 'http://localhost/pr1', | ||||
} | ||||
subject, headers, body, body_plaintext = EmailNotificationModel().render_email( | ||||
EmailNotificationModel.TYPE_PULL_REQUEST, **kwargs) | ||||
# subject | ||||
assert subject == 'Marcin Kuzminski wants you to review pull request #200: "Example Pull Request"' | ||||
r1728 | ||||
@pytest.mark.parametrize('mention', [ | ||||
True, | ||||
False | ||||
]) | ||||
@pytest.mark.parametrize('email_type', [ | ||||
EmailNotificationModel.TYPE_COMMIT_COMMENT, | ||||
EmailNotificationModel.TYPE_PULL_REQUEST_COMMENT | ||||
]) | ||||
r1774 | def test_render_comment_subject_no_newlines(app, mention, email_type): | |||
r1728 | ref = collections.namedtuple('Ref', | |||
'name, type')( | ||||
'fxies123', 'book' | ||||
) | ||||
pr = collections.namedtuple('PullRequest', | ||||
'pull_request_id, title, description, source_ref_parts, source_ref_name, target_ref_parts, target_ref_name')( | ||||
200, 'Example Pull Request', 'Desc of PR', ref, 'bookmark', ref, 'Branch') | ||||
source_repo = target_repo = collections.namedtuple('Repo', | ||||
'type, repo_name')( | ||||
'hg', 'pull_request_1') | ||||
kwargs = { | ||||
'user': '<marcin@rhodecode.com> Marcin Kuzminski', | ||||
'commit': AttributeDict(raw_id='a'*40, message='Commit message'), | ||||
'status_change': 'approved', | ||||
'commit_target_repo': AttributeDict(), | ||||
'repo_name': 'test-repo', | ||||
'comment_file': 'test-file.py', | ||||
'comment_line': 'n100', | ||||
'comment_type': 'note', | ||||
'commit_comment_url': 'http://comment-url', | ||||
'instance_url': 'http://rc-instance', | ||||
'comment_body': 'hello world', | ||||
'mention': mention, | ||||
'pr_comment_url': 'http://comment-url', | ||||
'pr_source_repo': AttributeDict(repo_name='foobar'), | ||||
'pr_source_repo_url': 'http://soirce-repo/url', | ||||
'pull_request': pr, | ||||
'pull_request_commits': [], | ||||
'pull_request_target_repo': target_repo, | ||||
'pull_request_target_repo_url': 'x', | ||||
'pull_request_source_repo': source_repo, | ||||
'pull_request_source_repo_url': 'x', | ||||
} | ||||
subject, headers, body, body_plaintext = EmailNotificationModel().render_email( | ||||
email_type, **kwargs) | ||||
assert '\n' not in subject | ||||