309 lines
11 KiB
Python
309 lines
11 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import boto
|
|
from boto.exception import BotoServerError
|
|
from moto import mock_sns_deprecated
|
|
import sure # noqa
|
|
|
|
|
|
@mock_sns_deprecated
|
|
def test_create_platform_application():
|
|
conn = boto.connect_sns()
|
|
platform_application = conn.create_platform_application(
|
|
name="my-application",
|
|
platform="APNS",
|
|
attributes={
|
|
"PlatformCredential": "platform_credential",
|
|
"PlatformPrincipal": "platform_principal",
|
|
},
|
|
)
|
|
application_arn = platform_application['CreatePlatformApplicationResponse'][
|
|
'CreatePlatformApplicationResult']['PlatformApplicationArn']
|
|
application_arn.should.equal(
|
|
'arn:aws:sns:us-east-1:123456789012:app/APNS/my-application')
|
|
|
|
|
|
@mock_sns_deprecated
|
|
def test_get_platform_application_attributes():
|
|
conn = boto.connect_sns()
|
|
platform_application = conn.create_platform_application(
|
|
name="my-application",
|
|
platform="APNS",
|
|
attributes={
|
|
"PlatformCredential": "platform_credential",
|
|
"PlatformPrincipal": "platform_principal",
|
|
},
|
|
)
|
|
arn = platform_application['CreatePlatformApplicationResponse'][
|
|
'CreatePlatformApplicationResult']['PlatformApplicationArn']
|
|
attributes = conn.get_platform_application_attributes(arn)['GetPlatformApplicationAttributesResponse'][
|
|
'GetPlatformApplicationAttributesResult']['Attributes']
|
|
attributes.should.equal({
|
|
"PlatformCredential": "platform_credential",
|
|
"PlatformPrincipal": "platform_principal",
|
|
})
|
|
|
|
|
|
@mock_sns_deprecated
|
|
def test_get_missing_platform_application_attributes():
|
|
conn = boto.connect_sns()
|
|
conn.get_platform_application_attributes.when.called_with(
|
|
"a-fake-arn").should.throw(BotoServerError)
|
|
|
|
|
|
@mock_sns_deprecated
|
|
def test_set_platform_application_attributes():
|
|
conn = boto.connect_sns()
|
|
platform_application = conn.create_platform_application(
|
|
name="my-application",
|
|
platform="APNS",
|
|
attributes={
|
|
"PlatformCredential": "platform_credential",
|
|
"PlatformPrincipal": "platform_principal",
|
|
},
|
|
)
|
|
arn = platform_application['CreatePlatformApplicationResponse'][
|
|
'CreatePlatformApplicationResult']['PlatformApplicationArn']
|
|
conn.set_platform_application_attributes(arn,
|
|
{"PlatformPrincipal": "other"}
|
|
)
|
|
attributes = conn.get_platform_application_attributes(arn)['GetPlatformApplicationAttributesResponse'][
|
|
'GetPlatformApplicationAttributesResult']['Attributes']
|
|
attributes.should.equal({
|
|
"PlatformCredential": "platform_credential",
|
|
"PlatformPrincipal": "other",
|
|
})
|
|
|
|
|
|
@mock_sns_deprecated
|
|
def test_list_platform_applications():
|
|
conn = boto.connect_sns()
|
|
conn.create_platform_application(
|
|
name="application1",
|
|
platform="APNS",
|
|
)
|
|
conn.create_platform_application(
|
|
name="application2",
|
|
platform="APNS",
|
|
)
|
|
|
|
applications_repsonse = conn.list_platform_applications()
|
|
applications = applications_repsonse['ListPlatformApplicationsResponse'][
|
|
'ListPlatformApplicationsResult']['PlatformApplications']
|
|
applications.should.have.length_of(2)
|
|
|
|
|
|
@mock_sns_deprecated
|
|
def test_delete_platform_application():
|
|
conn = boto.connect_sns()
|
|
conn.create_platform_application(
|
|
name="application1",
|
|
platform="APNS",
|
|
)
|
|
conn.create_platform_application(
|
|
name="application2",
|
|
platform="APNS",
|
|
)
|
|
|
|
applications_repsonse = conn.list_platform_applications()
|
|
applications = applications_repsonse['ListPlatformApplicationsResponse'][
|
|
'ListPlatformApplicationsResult']['PlatformApplications']
|
|
applications.should.have.length_of(2)
|
|
|
|
application_arn = applications[0]['PlatformApplicationArn']
|
|
conn.delete_platform_application(application_arn)
|
|
|
|
applications_repsonse = conn.list_platform_applications()
|
|
applications = applications_repsonse['ListPlatformApplicationsResponse'][
|
|
'ListPlatformApplicationsResult']['PlatformApplications']
|
|
applications.should.have.length_of(1)
|
|
|
|
|
|
@mock_sns_deprecated
|
|
def test_create_platform_endpoint():
|
|
conn = boto.connect_sns()
|
|
platform_application = conn.create_platform_application(
|
|
name="my-application",
|
|
platform="APNS",
|
|
)
|
|
application_arn = platform_application['CreatePlatformApplicationResponse'][
|
|
'CreatePlatformApplicationResult']['PlatformApplicationArn']
|
|
|
|
endpoint = conn.create_platform_endpoint(
|
|
platform_application_arn=application_arn,
|
|
token="some_unique_id",
|
|
custom_user_data="some user data",
|
|
attributes={
|
|
"Enabled": False,
|
|
},
|
|
)
|
|
|
|
endpoint_arn = endpoint['CreatePlatformEndpointResponse'][
|
|
'CreatePlatformEndpointResult']['EndpointArn']
|
|
endpoint_arn.should.contain(
|
|
"arn:aws:sns:us-east-1:123456789012:endpoint/APNS/my-application/")
|
|
|
|
|
|
@mock_sns_deprecated
|
|
def test_get_list_endpoints_by_platform_application():
|
|
conn = boto.connect_sns()
|
|
platform_application = conn.create_platform_application(
|
|
name="my-application",
|
|
platform="APNS",
|
|
)
|
|
application_arn = platform_application['CreatePlatformApplicationResponse'][
|
|
'CreatePlatformApplicationResult']['PlatformApplicationArn']
|
|
|
|
endpoint = conn.create_platform_endpoint(
|
|
platform_application_arn=application_arn,
|
|
token="some_unique_id",
|
|
custom_user_data="some user data",
|
|
attributes={
|
|
"CustomUserData": "some data",
|
|
},
|
|
)
|
|
endpoint_arn = endpoint['CreatePlatformEndpointResponse'][
|
|
'CreatePlatformEndpointResult']['EndpointArn']
|
|
|
|
endpoint_list = conn.list_endpoints_by_platform_application(
|
|
platform_application_arn=application_arn
|
|
)['ListEndpointsByPlatformApplicationResponse']['ListEndpointsByPlatformApplicationResult']['Endpoints']
|
|
|
|
endpoint_list.should.have.length_of(1)
|
|
endpoint_list[0]['Attributes']['CustomUserData'].should.equal('some data')
|
|
endpoint_list[0]['EndpointArn'].should.equal(endpoint_arn)
|
|
|
|
|
|
@mock_sns_deprecated
|
|
def test_get_endpoint_attributes():
|
|
conn = boto.connect_sns()
|
|
platform_application = conn.create_platform_application(
|
|
name="my-application",
|
|
platform="APNS",
|
|
)
|
|
application_arn = platform_application['CreatePlatformApplicationResponse'][
|
|
'CreatePlatformApplicationResult']['PlatformApplicationArn']
|
|
|
|
endpoint = conn.create_platform_endpoint(
|
|
platform_application_arn=application_arn,
|
|
token="some_unique_id",
|
|
custom_user_data="some user data",
|
|
attributes={
|
|
"Enabled": False,
|
|
"CustomUserData": "some data",
|
|
},
|
|
)
|
|
endpoint_arn = endpoint['CreatePlatformEndpointResponse'][
|
|
'CreatePlatformEndpointResult']['EndpointArn']
|
|
|
|
attributes = conn.get_endpoint_attributes(endpoint_arn)['GetEndpointAttributesResponse'][
|
|
'GetEndpointAttributesResult']['Attributes']
|
|
attributes.should.equal({
|
|
"Token": "some_unique_id",
|
|
"Enabled": 'False',
|
|
"CustomUserData": "some data",
|
|
})
|
|
|
|
|
|
@mock_sns_deprecated
|
|
def test_get_missing_endpoint_attributes():
|
|
conn = boto.connect_sns()
|
|
conn.get_endpoint_attributes.when.called_with(
|
|
"a-fake-arn").should.throw(BotoServerError)
|
|
|
|
|
|
@mock_sns_deprecated
|
|
def test_set_endpoint_attributes():
|
|
conn = boto.connect_sns()
|
|
platform_application = conn.create_platform_application(
|
|
name="my-application",
|
|
platform="APNS",
|
|
)
|
|
application_arn = platform_application['CreatePlatformApplicationResponse'][
|
|
'CreatePlatformApplicationResult']['PlatformApplicationArn']
|
|
|
|
endpoint = conn.create_platform_endpoint(
|
|
platform_application_arn=application_arn,
|
|
token="some_unique_id",
|
|
custom_user_data="some user data",
|
|
attributes={
|
|
"Enabled": False,
|
|
"CustomUserData": "some data",
|
|
},
|
|
)
|
|
endpoint_arn = endpoint['CreatePlatformEndpointResponse'][
|
|
'CreatePlatformEndpointResult']['EndpointArn']
|
|
|
|
conn.set_endpoint_attributes(endpoint_arn,
|
|
{"CustomUserData": "other data"}
|
|
)
|
|
attributes = conn.get_endpoint_attributes(endpoint_arn)['GetEndpointAttributesResponse'][
|
|
'GetEndpointAttributesResult']['Attributes']
|
|
attributes.should.equal({
|
|
"Token": "some_unique_id",
|
|
"Enabled": 'False',
|
|
"CustomUserData": "other data",
|
|
})
|
|
|
|
|
|
@mock_sns_deprecated
|
|
def test_delete_endpoint():
|
|
conn = boto.connect_sns()
|
|
platform_application = conn.create_platform_application(
|
|
name="my-application",
|
|
platform="APNS",
|
|
)
|
|
application_arn = platform_application['CreatePlatformApplicationResponse'][
|
|
'CreatePlatformApplicationResult']['PlatformApplicationArn']
|
|
|
|
endpoint = conn.create_platform_endpoint(
|
|
platform_application_arn=application_arn,
|
|
token="some_unique_id",
|
|
custom_user_data="some user data",
|
|
attributes={
|
|
"Enabled": False,
|
|
"CustomUserData": "some data",
|
|
},
|
|
)
|
|
endpoint_arn = endpoint['CreatePlatformEndpointResponse'][
|
|
'CreatePlatformEndpointResult']['EndpointArn']
|
|
|
|
endpoint_list = conn.list_endpoints_by_platform_application(
|
|
platform_application_arn=application_arn
|
|
)['ListEndpointsByPlatformApplicationResponse']['ListEndpointsByPlatformApplicationResult']['Endpoints']
|
|
|
|
endpoint_list.should.have.length_of(1)
|
|
|
|
conn.delete_endpoint(endpoint_arn)
|
|
|
|
endpoint_list = conn.list_endpoints_by_platform_application(
|
|
platform_application_arn=application_arn
|
|
)['ListEndpointsByPlatformApplicationResponse']['ListEndpointsByPlatformApplicationResult']['Endpoints']
|
|
endpoint_list.should.have.length_of(0)
|
|
|
|
|
|
@mock_sns_deprecated
|
|
def test_publish_to_platform_endpoint():
|
|
conn = boto.connect_sns()
|
|
platform_application = conn.create_platform_application(
|
|
name="my-application",
|
|
platform="APNS",
|
|
)
|
|
application_arn = platform_application['CreatePlatformApplicationResponse'][
|
|
'CreatePlatformApplicationResult']['PlatformApplicationArn']
|
|
|
|
endpoint = conn.create_platform_endpoint(
|
|
platform_application_arn=application_arn,
|
|
token="some_unique_id",
|
|
custom_user_data="some user data",
|
|
attributes={
|
|
"Enabled": True,
|
|
},
|
|
)
|
|
|
|
endpoint_arn = endpoint['CreatePlatformEndpointResponse'][
|
|
'CreatePlatformEndpointResult']['EndpointArn']
|
|
|
|
conn.publish(message="some message", message_structure="json",
|
|
target_arn=endpoint_arn)
|