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

259
tests/test_sns/test_application.py
View file

@ -17,10 +17,12 @@ def test_create_platform_application():
"PlatformPrincipal": "platform_principal",
},
)
application_arn = platform_application['CreatePlatformApplicationResponse'][
'CreatePlatformApplicationResult']['PlatformApplicationArn']
application_arn = platform_application["CreatePlatformApplicationResponse"][
"CreatePlatformApplicationResult"
]["PlatformApplicationArn"]
application_arn.should.equal(
'arn:aws:sns:us-east-1:123456789012:app/APNS/my-application')
"arn:aws:sns:us-east-1:123456789012:app/APNS/my-application"
)
@mock_sns_deprecated
@ -34,21 +36,26 @@ def test_get_platform_application_attributes():
"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",
})
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)
"a-fake-arn"
).should.throw(BotoServerError)
@mock_sns_deprecated
@ -62,60 +69,50 @@ def test_set_platform_application_attributes():
"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",
})
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",
)
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 = 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",
)
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 = applications_repsonse["ListPlatformApplicationsResponse"][
"ListPlatformApplicationsResult"
]["PlatformApplications"]
applications.should.have.length_of(2)
application_arn = applications[0]['PlatformApplicationArn']
application_arn = applications[0]["PlatformApplicationArn"]
conn.delete_platform_application(application_arn)
applications_repsonse = conn.list_platform_applications()
applications = applications_repsonse['ListPlatformApplicationsResponse'][
'ListPlatformApplicationsResult']['PlatformApplications']
applications = applications_repsonse["ListPlatformApplicationsResponse"][
"ListPlatformApplicationsResult"
]["PlatformApplications"]
applications.should.have.length_of(1)
@ -123,154 +120,152 @@ def test_delete_platform_application():
def test_create_platform_endpoint():
conn = boto.connect_sns()
platform_application = conn.create_platform_application(
name="my-application",
platform="APNS",
name="my-application", platform="APNS"
)
application_arn = platform_application['CreatePlatformApplicationResponse'][
'CreatePlatformApplicationResult']['PlatformApplicationArn']
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,
},
attributes={"Enabled": False},
)
endpoint_arn = endpoint['CreatePlatformEndpointResponse'][
'CreatePlatformEndpointResult']['EndpointArn']
endpoint_arn = endpoint["CreatePlatformEndpointResponse"][
"CreatePlatformEndpointResult"
]["EndpointArn"]
endpoint_arn.should.contain(
"arn:aws:sns:us-east-1:123456789012:endpoint/APNS/my-application/")
"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",
name="my-application", platform="APNS"
)
application_arn = platform_application['CreatePlatformApplicationResponse'][
'CreatePlatformApplicationResult']['PlatformApplicationArn']
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",
},
attributes={"CustomUserData": "some data"},
)
endpoint_arn = endpoint['CreatePlatformEndpointResponse'][
'CreatePlatformEndpointResult']['EndpointArn']
endpoint_arn = endpoint["CreatePlatformEndpointResponse"][
"CreatePlatformEndpointResult"
]["EndpointArn"]
endpoint_list = conn.list_endpoints_by_platform_application(
platform_application_arn=application_arn
)['ListEndpointsByPlatformApplicationResponse']['ListEndpointsByPlatformApplicationResult']['Endpoints']
)["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)
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",
name="my-application", platform="APNS"
)
application_arn = platform_application['CreatePlatformApplicationResponse'][
'CreatePlatformApplicationResult']['PlatformApplicationArn']
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",
},
attributes={"Enabled": False, "CustomUserData": "some data"},
)
endpoint_arn = endpoint['CreatePlatformEndpointResponse'][
'CreatePlatformEndpointResult']['EndpointArn']
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",
})
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)
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",
name="my-application", platform="APNS"
)
application_arn = platform_application['CreatePlatformApplicationResponse'][
'CreatePlatformApplicationResult']['PlatformApplicationArn']
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",
},
attributes={"Enabled": False, "CustomUserData": "some data"},
)
endpoint_arn = endpoint['CreatePlatformEndpointResponse'][
'CreatePlatformEndpointResult']['EndpointArn']
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",
})
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",
name="my-application", platform="APNS"
)
application_arn = platform_application['CreatePlatformApplicationResponse'][
'CreatePlatformApplicationResult']['PlatformApplicationArn']
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",
},
attributes={"Enabled": False, "CustomUserData": "some data"},
)
endpoint_arn = endpoint['CreatePlatformEndpointResponse'][
'CreatePlatformEndpointResult']['EndpointArn']
endpoint_arn = endpoint["CreatePlatformEndpointResponse"][
"CreatePlatformEndpointResult"
]["EndpointArn"]
endpoint_list = conn.list_endpoints_by_platform_application(
platform_application_arn=application_arn
)['ListEndpointsByPlatformApplicationResponse']['ListEndpointsByPlatformApplicationResult']['Endpoints']
)["ListEndpointsByPlatformApplicationResponse"][
"ListEndpointsByPlatformApplicationResult"
][
"Endpoints"
]
endpoint_list.should.have.length_of(1)
@ -278,7 +273,11 @@ def test_delete_endpoint():
endpoint_list = conn.list_endpoints_by_platform_application(
platform_application_arn=application_arn
)['ListEndpointsByPlatformApplicationResponse']['ListEndpointsByPlatformApplicationResult']['Endpoints']
)["ListEndpointsByPlatformApplicationResponse"][
"ListEndpointsByPlatformApplicationResult"
][
"Endpoints"
]
endpoint_list.should.have.length_of(0)
@ -286,23 +285,23 @@ def test_delete_endpoint():
def test_publish_to_platform_endpoint():
conn = boto.connect_sns()
platform_application = conn.create_platform_application(
name="my-application",
platform="APNS",
name="my-application", platform="APNS"
)
application_arn = platform_application['CreatePlatformApplicationResponse'][
'CreatePlatformApplicationResult']['PlatformApplicationArn']
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,
},
attributes={"Enabled": True},
)
endpoint_arn = endpoint['CreatePlatformEndpointResponse'][
'CreatePlatformEndpointResult']['EndpointArn']
endpoint_arn = endpoint["CreatePlatformEndpointResponse"][
"CreatePlatformEndpointResult"
]["EndpointArn"]
conn.publish(message="some message", message_structure="json",
target_arn=endpoint_arn)
conn.publish(
message="some message", message_structure="json", target_arn=endpoint_arn
)