def test_target():
batch = MailBatch(MockTemplateStore())
batch.target(Recipient("1@mail", "1", timezone.utc, False), "template-author")
batch.target(Recipient("2@mail", "2", timezone.utc, False), "template-reviewer")
emails = batch.process(PUBLIC_REVISION, "actor", 0, 0, EVENT)
assert len(emails) == 2
assert emails[0].to == "1@mail"
assert emails[1].to == "2@mail"
def test_only_sends_to_each_recipient_once():
batch = MailBatch(MockTemplateStore())
batch.target(Recipient("1@mail", "1", timezone.utc, False),
"template-author")
batch.target_many(
[
Recipient("1@mail", "1", timezone.utc, False),
Recipient("2@mail", "2", timezone.utc, False),
],
"template-reviewer",
)
emails = batch.process(PUBLIC_REVISION, ACTOR, 0, 0, EVENT)
assert len(emails) == 2
assert emails[0].to == "1@mail"
assert emails[1].to == "2@mail"
def test_target_many():
batch = MailBatch(MockTemplateStore())
batch.target_many(
[
Recipient("1@mail", "1", timezone.utc, False),
Recipient("2@mail", "2", timezone.utc, False),
],
"template-reviewer",
)
batch.target_many(
[
Recipient("3@mail", "3", timezone.utc, False),
Recipient("4@mail", "4", timezone.utc, False),
],
"template-reviewer",
)
emails = batch.process(PUBLIC_REVISION, "actor", 0, 0, EVENT)
assert len(emails) == 4
assert emails[0].to == "1@mail"
assert emails[1].to == "2@mail"
assert emails[2].to == "3@mail"
assert emails[3].to == "4@mail"
def test_integration_templates():
template_store = TemplateStore("", "", False)
template = template_store.get(PUBLIC_TEMPLATE_PATH_PREFIX + "pinged")
html, text = template.render(
{
"revision": Revision(1, "revision", "link", None),
"actor_name": "actor",
"recipient_username": "******",
"unique_number": 0,
"event": RevisionCommentPinged(
Recipient("1@mail", "1", timezone.utc, False),
"you've been pinged",
[],
"link",
),
}
)
assert "actor mentioned you" in html
assert "you've been pinged" in html
assert "actor mentioned you" in text
assert "you've been pinged" in text
def test_filter_target_is_actor():
batch = MailBatch(MockTemplateStore())
batch.target(Recipient("1@mail", "1", timezone.utc, True), "template-author")
emails = batch.process(PUBLIC_REVISION, "actor", 0, 0, EVENT)
assert len(emails) == 0
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
from datetime import timezone
from phabricatoremails.render.events.common import Recipient
from phabricatoremails.render.events.phabricator import Revision, RevisionCreated
from phabricatoremails.render.events.phabricator_secure import SecureRevision, SecureBug
from phabricatoremails.render.mailbatch import (
MailBatch,
PUBLIC_TEMPLATE_PATH_PREFIX,
SECURE_TEMPLATE_PATH_PREFIX,
)
from tests.render.mock_template import MockTemplateStore
NON_ACTOR_RECIPIENT = Recipient("1@mail", "1", timezone.utc, False)
PUBLIC_REVISION = Revision(1, "revision", "link", None)
EVENT = RevisionCreated([], [])
def test_target():
batch = MailBatch(MockTemplateStore())
batch.target(Recipient("1@mail", "1", timezone.utc, False), "template-author")
batch.target(Recipient("2@mail", "2", timezone.utc, False), "template-reviewer")
emails = batch.process(PUBLIC_REVISION, "actor", 0, 0, EVENT)
assert len(emails) == 2
assert emails[0].to == "1@mail"
assert emails[1].to == "2@mail"
def test_target_many():