update SES backend to support domain identities and multiple recipients
includes test cases for ses client with boto3
This commit is contained in:
parent
af04bb2681
commit
fb06c6517e
5 changed files with 239 additions and 43 deletions
|
|
@ -75,11 +75,10 @@ def test_send_html_email():
|
|||
def test_send_raw_email():
|
||||
conn = boto.connect_ses('the_key', 'the_secret')
|
||||
|
||||
to = 'to@example.com'
|
||||
message = email.mime.multipart.MIMEMultipart()
|
||||
message['Subject'] = 'Test'
|
||||
message['From'] = 'test@example.com'
|
||||
message['To'] = to
|
||||
message['To'] = 'to@example.com'
|
||||
|
||||
# Message body
|
||||
part = email.mime.text.MIMEText('test file attached')
|
||||
|
|
@ -93,14 +92,12 @@ def test_send_raw_email():
|
|||
conn.send_raw_email.when.called_with(
|
||||
source=message['From'],
|
||||
raw_message=message.as_string(),
|
||||
destinations=message['To']
|
||||
).should.throw(BotoServerError)
|
||||
|
||||
conn.verify_email_identity("test@example.com")
|
||||
conn.send_raw_email(
|
||||
source=message['From'],
|
||||
raw_message=message.as_string(),
|
||||
destinations=message['To']
|
||||
)
|
||||
|
||||
send_quota = conn.get_send_quota()
|
||||
|
|
|
|||
131
tests/test_ses/test_ses_boto3.py
Normal file
131
tests/test_ses/test_ses_boto3.py
Normal file
|
|
@ -0,0 +1,131 @@
|
|||
from __future__ import unicode_literals
|
||||
|
||||
import boto3
|
||||
from botocore.exceptions import ClientError
|
||||
from six.moves.email_mime_multipart import MIMEMultipart
|
||||
from six.moves.email_mime_text import MIMEText
|
||||
|
||||
import sure # noqa
|
||||
|
||||
from moto import mock_ses
|
||||
|
||||
|
||||
@mock_ses
|
||||
def test_verify_email_identity():
|
||||
conn = boto3.client('ses', region_name='us-east-1')
|
||||
conn.verify_email_identity(EmailAddress="test@example.com")
|
||||
|
||||
identities = conn.list_identities()
|
||||
address = identities['Identities'][0]
|
||||
address.should.equal('test@example.com')
|
||||
|
||||
|
||||
@mock_ses
|
||||
def test_domain_verify():
|
||||
conn = boto3.client('ses', region_name='us-east-1')
|
||||
|
||||
conn.verify_domain_dkim(Domain="domain1.com")
|
||||
conn.verify_domain_identity(Domain="domain2.com")
|
||||
|
||||
identities = conn.list_identities()
|
||||
domains = list(identities['Identities'])
|
||||
domains.should.equal(['domain1.com', 'domain2.com'])
|
||||
|
||||
|
||||
@mock_ses
|
||||
def test_delete_identity():
|
||||
conn = boto3.client('ses', region_name='us-east-1')
|
||||
conn.verify_email_identity(EmailAddress="test@example.com")
|
||||
|
||||
conn.list_identities()['Identities'].should.have.length_of(1)
|
||||
conn.delete_identity(Identity="test@example.com")
|
||||
conn.list_identities()['Identities'].should.have.length_of(0)
|
||||
|
||||
|
||||
@mock_ses
|
||||
def test_send_email():
|
||||
conn = boto3.client('ses', region_name='us-east-1')
|
||||
|
||||
kwargs = dict(
|
||||
Source="test@example.com",
|
||||
Destination={
|
||||
"ToAddresses": ["test_to@example.com"],
|
||||
"CcAddresses": ["test_cc@example.com"],
|
||||
"BccAddresses": ["test_bcc@example.com"],
|
||||
},
|
||||
Message={
|
||||
"Subject": {"Data": "test subject"},
|
||||
"Body": {"Text": {"Data": "test body"}}
|
||||
}
|
||||
)
|
||||
conn.send_email.when.called_with(**kwargs).should.throw(ClientError)
|
||||
|
||||
conn.verify_domain_identity(Domain='example.com')
|
||||
conn.send_email(**kwargs)
|
||||
|
||||
too_many_addresses = list('to%s@example.com' % i for i in range(51))
|
||||
conn.send_email.when.called_with(
|
||||
**dict(kwargs, Destination={'ToAddresses': too_many_addresses})
|
||||
).should.throw(ClientError)
|
||||
|
||||
send_quota = conn.get_send_quota()
|
||||
sent_count = int(send_quota['SentLast24Hours'])
|
||||
sent_count.should.equal(3)
|
||||
|
||||
|
||||
@mock_ses
|
||||
def test_send_html_email():
|
||||
conn = boto3.client('ses', region_name='us-east-1')
|
||||
|
||||
kwargs = dict(
|
||||
Source="test@example.com",
|
||||
Destination={
|
||||
"ToAddresses": ["test_to@example.com"]
|
||||
},
|
||||
Message={
|
||||
"Subject": {"Data": "test subject"},
|
||||
"Body": {"Html": {"Data": "test body"}}
|
||||
}
|
||||
)
|
||||
|
||||
conn.send_email.when.called_with(**kwargs).should.throw(ClientError)
|
||||
|
||||
conn.verify_email_identity(EmailAddress="test@example.com")
|
||||
conn.send_email(**kwargs)
|
||||
|
||||
send_quota = conn.get_send_quota()
|
||||
sent_count = int(send_quota['SentLast24Hours'])
|
||||
sent_count.should.equal(1)
|
||||
|
||||
|
||||
@mock_ses
|
||||
def test_send_raw_email():
|
||||
conn = boto3.client('ses', region_name='us-east-1')
|
||||
|
||||
message = MIMEMultipart()
|
||||
message['Subject'] = 'Test'
|
||||
message['From'] = 'test@example.com'
|
||||
message['To'] = 'to@example.com, foo@example.com'
|
||||
|
||||
# Message body
|
||||
part = MIMEText('test file attached')
|
||||
message.attach(part)
|
||||
|
||||
# Attachment
|
||||
part = MIMEText('contents of test file here')
|
||||
part.add_header('Content-Disposition', 'attachment; filename=test.txt')
|
||||
message.attach(part)
|
||||
|
||||
kwargs = dict(
|
||||
Source=message['From'],
|
||||
RawMessage={'Data': message.as_string()},
|
||||
)
|
||||
|
||||
conn.send_raw_email.when.called_with(**kwargs).should.throw(ClientError)
|
||||
|
||||
conn.verify_email_identity(EmailAddress="test@example.com")
|
||||
conn.send_raw_email(**kwargs)
|
||||
|
||||
send_quota = conn.get_send_quota()
|
||||
sent_count = int(send_quota['SentLast24Hours'])
|
||||
sent_count.should.equal(2)
|
||||
Loading…
Add table
Add a link
Reference in a new issue