def test_EmailMessage_attributes(self):
"""Test that send_at and categories attributes are correctly written through to output."""
msg = EmailMessage(
subject="Hello, World!",
body="Hello, World!",
from_email="Sam Smith <*****@*****.**>",
to=["John Doe <*****@*****.**>", "*****@*****.**"],
)
# Set new attributes as message property
msg.send_at = 1518108670
if SENDGRID_VERSION < '6':
msg.categories = ['mammal', 'dog']
else:
msg.categories = ['dog', 'mammal']
msg.ip_pool_name = 'some-name'
result = self.backend._build_sg_mail(msg)
expected = {
"personalizations": [{
"to": [{
"email": "*****@*****.**",
"name": "John Doe"
}, {
"email": "*****@*****.**",
}],
"subject":
"Hello, World!",
"send_at":
1518108670,
}],
"from": {
"email": "*****@*****.**",
"name": "Sam Smith"
},
"mail_settings": {
"sandbox_mode": {
"enable": False
}
},
"subject":
"Hello, World!",
"tracking_settings": {
"open_tracking": {
"enable": True
}
},
"content": [{
"type": "text/plain",
"value": "Hello, World!"
}],
"categories": ['mammal', 'dog'],
"ip_pool_name":
"some-name"
}
self.assertDictEqual(result, expected)
def test_EmailMessage_attributes(self):
"""Test that send_at and categories attributes are correctly written through to output."""
msg = EmailMessage(
subject="Hello, World!",
body="Hello, World!",
from_email="Sam Smith <*****@*****.**>",
to=["John Doe <*****@*****.**>", "*****@*****.**"],
)
# Set new attributes as message property
msg.send_at = 1518108670
msg.categories = ['mammal', 'dog']
msg.ip_pool_name = 'some-name'
result = self.backend._build_sg_mail(msg)
expected = {
"personalizations": [{
"to": [{
"email": "*****@*****.**",
"name": "John Doe"
}, {
"email": "*****@*****.**",
}],
"subject": "Hello, World!",
"send_at": 1518108670,
}],
"from": {
"email": "*****@*****.**",
"name": "Sam Smith"
},
"mail_settings": {
"sandbox_mode": {
"enable": False
}
},
"subject": "Hello, World!",
"tracking_settings": {"open_tracking": {"enable": True}},
"content": [{
"type": "text/plain",
"value": "Hello, World!"
}],
"categories": ['mammal', 'dog'],
"ip_pool_name": "some-name"
}
self.assertDictEqual(result, expected)
