Cleanup SNS exceptions. Closes #751.

This commit is contained in:
Steve Pulec 2017-03-16 22:28:30 -04:00
commit c207963a86
4 changed files with 87 additions and 3 deletions

View file

@ -297,7 +297,7 @@ def test_publish_to_platform_endpoint():
token="some_unique_id",
custom_user_data="some user data",
attributes={
"Enabled": False,
"Enabled": True,
},
)

View file

@ -142,6 +142,35 @@ def test_create_platform_endpoint():
"arn:aws:sns:us-east-1:123456789012:endpoint/APNS/my-application/")
@mock_sns
def test_create_duplicate_platform_endpoint():
conn = boto3.client('sns', region_name='us-east-1')
platform_application = conn.create_platform_application(
Name="my-application",
Platform="APNS",
Attributes={},
)
application_arn = platform_application['PlatformApplicationArn']
endpoint = conn.create_platform_endpoint(
PlatformApplicationArn=application_arn,
Token="some_unique_id",
CustomUserData="some user data",
Attributes={
"Enabled": 'false',
},
)
endpoint = conn.create_platform_endpoint.when.called_with(
PlatformApplicationArn=application_arn,
Token="some_unique_id",
CustomUserData="some user data",
Attributes={
"Enabled": 'false',
},
).should.throw(ClientError)
@mock_sns
def test_get_list_endpoints_by_platform_application():
conn = boto3.client('sns', region_name='us-east-1')
@ -256,7 +285,7 @@ def test_publish_to_platform_endpoint():
Token="some_unique_id",
CustomUserData="some user data",
Attributes={
"Enabled": 'false',
"Enabled": 'true',
},
)
@ -264,3 +293,31 @@ def test_publish_to_platform_endpoint():
conn.publish(Message="some message",
MessageStructure="json", TargetArn=endpoint_arn)
@mock_sns
def test_publish_to_disabled_platform_endpoint():
conn = boto3.client('sns', region_name='us-east-1')
platform_application = conn.create_platform_application(
Name="my-application",
Platform="APNS",
Attributes={},
)
application_arn = platform_application['PlatformApplicationArn']
endpoint = conn.create_platform_endpoint(
PlatformApplicationArn=application_arn,
Token="some_unique_id",
CustomUserData="some user data",
Attributes={
"Enabled": 'false',
},
)
endpoint_arn = endpoint['EndpointArn']
conn.publish.when.called_with(
Message="some message",
MessageStructure="json",
TargetArn=endpoint_arn,
).should.throw(ClientError)