catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Run black on moto & test directories.

Browse source
This commit is contained in:
Asher Foa 2019-10-31 08:44:26 -07:00
commit 96e5b1993d
507 changed files with 52541 additions and 47814 deletions
Download patch file Download diff file Expand all files Collapse all files

125
tests/test_ses/test_ses_boto3.py
View file

@ -12,46 +12,48 @@ from moto import mock_ses
@mock_ses
def test_verify_email_identity():
conn = boto3.client('ses', region_name='us-east-1')
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')
address = identities["Identities"][0]
address.should.equal("test@example.com")
@mock_ses
def test_verify_email_address():
conn = boto3.client('ses', region_name='us-east-1')
conn = boto3.client("ses", region_name="us-east-1")
conn.verify_email_address(EmailAddress="test@example.com")
email_addresses = conn.list_verified_email_addresses()
email = email_addresses['VerifiedEmailAddresses'][0]
email.should.equal('test@example.com')
email = email_addresses["VerifiedEmailAddresses"][0]
email.should.equal("test@example.com")
@mock_ses
def test_domain_verify():
conn = boto3.client('ses', region_name='us-east-1')
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'])
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 = 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.list_identities()["Identities"].should.have.length_of(1)
conn.delete_identity(Identity="test@example.com")
conn.list_identities()['Identities'].should.have.length_of(0)
conn.list_identities()["Identities"].should.have.length_of(0)
@mock_ses
def test_send_email():
conn = boto3.client('ses', region_name='us-east-1')
conn = boto3.client("ses", region_name="us-east-1")
kwargs = dict(
Source="test@example.com",
@ -62,27 +64,27 @@ def test_send_email():
},
Message={
"Subject": {"Data": "test subject"},
"Body": {"Text": {"Data": "test body"}}
}
"Body": {"Text": {"Data": "test body"}},
},
)
conn.send_email.when.called_with(**kwargs).should.throw(ClientError)
conn.verify_domain_identity(Domain='example.com')
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))
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})
**dict(kwargs, Destination={"ToAddresses": too_many_addresses})
).should.throw(ClientError)
send_quota = conn.get_send_quota()
sent_count = int(send_quota['SentLast24Hours'])
sent_count = int(send_quota["SentLast24Hours"])
sent_count.should.equal(3)
@mock_ses
def test_send_templated_email():
conn = boto3.client('ses', region_name='us-east-1')
conn = boto3.client("ses", region_name="us-east-1")
kwargs = dict(
Source="test@example.com",
@ -92,38 +94,35 @@ def test_send_templated_email():
"BccAddresses": ["test_bcc@example.com"],
},
Template="test_template",
TemplateData='{\"name\": \"test\"}'
TemplateData='{"name": "test"}',
)
conn.send_templated_email.when.called_with(
**kwargs).should.throw(ClientError)
conn.send_templated_email.when.called_with(**kwargs).should.throw(ClientError)
conn.verify_domain_identity(Domain='example.com')
conn.verify_domain_identity(Domain="example.com")
conn.send_templated_email(**kwargs)
too_many_addresses = list('to%s@example.com' % i for i in range(51))
too_many_addresses = list("to%s@example.com" % i for i in range(51))
conn.send_templated_email.when.called_with(
**dict(kwargs, Destination={'ToAddresses': too_many_addresses})
**dict(kwargs, Destination={"ToAddresses": too_many_addresses})
).should.throw(ClientError)
send_quota = conn.get_send_quota()
sent_count = int(send_quota['SentLast24Hours'])
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')
conn = boto3.client("ses", region_name="us-east-1")
kwargs = dict(
Source="test@example.com",
Destination={
"ToAddresses": ["test_to@example.com"]
},
Destination={"ToAddresses": ["test_to@example.com"]},
Message={
"Subject": {"Data": "test subject"},
"Body": {"Html": {"Data": "test body"}}
}
"Body": {"Html": {"Data": "test body"}},
},
)
conn.send_email.when.called_with(**kwargs).should.throw(ClientError)
@ -132,32 +131,29 @@ def test_send_html_email():
conn.send_email(**kwargs)
send_quota = conn.get_send_quota()
sent_count = int(send_quota['SentLast24Hours'])
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')
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["Subject"] = "Test"
message["From"] = "test@example.com"
message["To"] = "to@example.com, foo@example.com"
# Message body
part = MIMEText('test file attached')
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')
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()},
)
kwargs = dict(Source=message["From"], RawMessage={"Data": message.as_string()})
conn.send_raw_email.when.called_with(**kwargs).should.throw(ClientError)
@ -165,31 +161,29 @@ def test_send_raw_email():
conn.send_raw_email(**kwargs)
send_quota = conn.get_send_quota()
sent_count = int(send_quota['SentLast24Hours'])
sent_count = int(send_quota["SentLast24Hours"])
sent_count.should.equal(2)
@mock_ses
def test_send_raw_email_without_source():
conn = boto3.client('ses', region_name='us-east-1')
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["Subject"] = "Test"
message["From"] = "test@example.com"
message["To"] = "to@example.com, foo@example.com"
# Message body
part = MIMEText('test file attached')
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')
part = MIMEText("contents of test file here")
part.add_header("Content-Disposition", "attachment; filename=test.txt")
message.attach(part)
kwargs = dict(
RawMessage={'Data': message.as_string()},
)
kwargs = dict(RawMessage={"Data": message.as_string()})
conn.send_raw_email.when.called_with(**kwargs).should.throw(ClientError)
@ -197,29 +191,26 @@ def test_send_raw_email_without_source():
conn.send_raw_email(**kwargs)
send_quota = conn.get_send_quota()
sent_count = int(send_quota['SentLast24Hours'])
sent_count = int(send_quota["SentLast24Hours"])
sent_count.should.equal(2)
@mock_ses
def test_send_raw_email_without_source_or_from():
conn = boto3.client('ses', region_name='us-east-1')
conn = boto3.client("ses", region_name="us-east-1")
message = MIMEMultipart()
message['Subject'] = 'Test'
message['To'] = 'to@example.com, foo@example.com'
message["Subject"] = "Test"
message["To"] = "to@example.com, foo@example.com"
# Message body
part = MIMEText('test file attached')
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')
part = MIMEText("contents of test file here")
part.add_header("Content-Disposition", "attachment; filename=test.txt")
message.attach(part)
kwargs = dict(
RawMessage={'Data': message.as_string()},
)
kwargs = dict(RawMessage={"Data": message.as_string()})
conn.send_raw_email.when.called_with(**kwargs).should.throw(ClientError)