19074c535cFor now subscriptions do nothing, but if we go the route of handing out subscribe tokens, I have layed the groundwork for validating that
113 lines
3.6 KiB
Python
113 lines
3.6 KiB
Python
from __future__ import unicode_literals
|
|
import boto3
|
|
import sure # noqa
|
|
|
|
from moto import mock_sns
|
|
from botocore.exceptions import ClientError
|
|
from nose.tools import assert_raises
|
|
|
|
|
|
@mock_sns
|
|
def test_set_sms_attributes():
|
|
conn = boto3.client('sns', region_name='us-east-1')
|
|
|
|
conn.set_sms_attributes(attributes={'DefaultSMSType': 'Transactional', 'test': 'test'})
|
|
|
|
response = conn.get_sms_attributes()
|
|
response.should.contain('attributes')
|
|
response['attributes'].should.contain('DefaultSMSType')
|
|
response['attributes'].should.contain('test')
|
|
response['attributes']['DefaultSMSType'].should.equal('Transactional')
|
|
response['attributes']['test'].should.equal('test')
|
|
|
|
|
|
@mock_sns
|
|
def test_get_sms_attributes_filtered():
|
|
conn = boto3.client('sns', region_name='us-east-1')
|
|
|
|
conn.set_sms_attributes(attributes={'DefaultSMSType': 'Transactional', 'test': 'test'})
|
|
|
|
response = conn.get_sms_attributes(attributes=['DefaultSMSType'])
|
|
response.should.contain('attributes')
|
|
response['attributes'].should.contain('DefaultSMSType')
|
|
response['attributes'].should_not.contain('test')
|
|
response['attributes']['DefaultSMSType'].should.equal('Transactional')
|
|
|
|
|
|
@mock_sns
|
|
def test_check_not_opted_out():
|
|
conn = boto3.client('sns', region_name='us-east-1')
|
|
response = conn.check_if_phone_number_is_opted_out(phoneNumber='+447428545375')
|
|
|
|
response.should.contain('isOptedOut')
|
|
response['isOptedOut'].should.be(False)
|
|
|
|
|
|
@mock_sns
|
|
def test_check_opted_out(): # Ends in 99 so is opted out
|
|
conn = boto3.client('sns', region_name='us-east-1')
|
|
response = conn.check_if_phone_number_is_opted_out(phoneNumber='+447428545399')
|
|
|
|
response.should.contain('isOptedOut')
|
|
response['isOptedOut'].should.be(True)
|
|
|
|
|
|
@mock_sns
|
|
def test_check_opted_out_invalid():
|
|
conn = boto3.client('sns', region_name='us-east-1')
|
|
|
|
# Invalid phone number
|
|
with assert_raises(ClientError):
|
|
conn.check_if_phone_number_is_opted_out(phoneNumber='+44742LALALA')
|
|
|
|
|
|
@mock_sns
|
|
def test_list_opted_out():
|
|
conn = boto3.client('sns', region_name='us-east-1')
|
|
response = conn.list_phone_numbers_opted_out()
|
|
|
|
response.should.contain('phoneNumbers')
|
|
len(response['phoneNumbers']).should.be.greater_than(0)
|
|
|
|
|
|
@mock_sns
|
|
def test_opt_in():
|
|
conn = boto3.client('sns', region_name='us-east-1')
|
|
response = conn.list_phone_numbers_opted_out()
|
|
current_len = len(response['phoneNumbers'])
|
|
assert current_len > 0
|
|
|
|
conn.opt_in_phone_number(phoneNumber=response['phoneNumbers'][0])
|
|
|
|
response = conn.list_phone_numbers_opted_out()
|
|
len(response['phoneNumbers']).should.be.greater_than(0)
|
|
len(response['phoneNumbers']).should.be.lower_than(current_len)
|
|
|
|
|
|
@mock_sns
|
|
def test_add_remove_permissions():
|
|
conn = boto3.client('sns', region_name='us-east-1')
|
|
response = conn.create_topic(Name='testpermissions')
|
|
|
|
conn.add_permission(
|
|
TopicArn=response['TopicArn'],
|
|
Label='Test1234',
|
|
AWSAccountId=['999999999999'],
|
|
ActionName=['AddPermission']
|
|
)
|
|
conn.remove_permission(
|
|
TopicArn=response['TopicArn'],
|
|
Label='Test1234'
|
|
)
|
|
|
|
|
|
@mock_sns
|
|
def test_confirm_subscription():
|
|
conn = boto3.client('sns', region_name='us-east-1')
|
|
response = conn.create_topic(Name='testconfirm')
|
|
|
|
conn.confirm_subscription(
|
|
TopicArn=response['TopicArn'],
|
|
Token='2336412f37fb687f5d51e6e241d59b68c4e583a5cee0be6f95bbf97ab8d2441cf47b99e848408adaadf4c197e65f03473d53c4ba398f6abbf38ce2e8ebf7b4ceceb2cd817959bcde1357e58a2861b05288c535822eb88cac3db04f592285249971efc6484194fc4a4586147f16916692',
|
|
AuthenticateOnUnsubscribe='true'
|
|
)
|