def setUp(self):
self.sqsc = SQSConnection()
self.snsc = SNSConnection()
def setUp(self):
self.sqsc = SQSConnection()
self.snsc = SNSConnection()
class SNSSubcribeSQSTest(unittest.TestCase):
sqs = True
sns = True
def setUp(self):
self.sqsc = SQSConnection()
self.snsc = SNSConnection()
def get_policy_statements(self, queue):
attrs = queue.get_attributes('Policy')
policy = json.loads(attrs.get('Policy', "{}"))
return policy.get('Statement', {})
def test_correct_sid(self):
now = time.time()
topic_name = queue_name = "test_correct_sid%d" % (now)
timeout = 60
queue = self.sqsc.create_queue(queue_name, timeout)
self.addCleanup(self.sqsc.delete_queue, queue, True)
queue_arn = queue.arn
topic = self.snsc.create_topic(topic_name)
topic_arn = topic['CreateTopicResponse']['CreateTopicResult']\
['TopicArn']
self.addCleanup(self.snsc.delete_topic, topic_arn)
expected_sid = hashlib.md5((topic_arn + queue_arn).encode('utf-8')).hexdigest()
resp = self.snsc.subscribe_sqs_queue(topic_arn, queue)
found_expected_sid = False
statements = self.get_policy_statements(queue)
for statement in statements:
if statement['Sid'] == expected_sid:
found_expected_sid = True
break
self.assertTrue(found_expected_sid)
def test_idempotent_subscribe(self):
now = time.time()
topic_name = queue_name = "test_idempotent_subscribe%d" % (now)
timeout = 60
queue = self.sqsc.create_queue(queue_name, timeout)
self.addCleanup(self.sqsc.delete_queue, queue, True)
initial_statements = self.get_policy_statements(queue)
queue_arn = queue.arn
topic = self.snsc.create_topic(topic_name)
topic_arn = topic['CreateTopicResponse']['CreateTopicResult']\
['TopicArn']
self.addCleanup(self.snsc.delete_topic, topic_arn)
resp = self.snsc.subscribe_sqs_queue(topic_arn, queue)
time.sleep(3)
first_subscribe_statements = self.get_policy_statements(queue)
self.assertEqual(len(first_subscribe_statements),
len(initial_statements) + 1)
resp2 = self.snsc.subscribe_sqs_queue(topic_arn, queue)
time.sleep(3)
second_subscribe_statements = self.get_policy_statements(queue)
self.assertEqual(len(second_subscribe_statements),
len(first_subscribe_statements))
class SNSSubcribeSQSTest(unittest.TestCase):
sqs = True
sns = True
def setUp(self):
self.sqsc = SQSConnection()
self.snsc = SNSConnection()
def get_policy_statements(self, queue):
attrs = queue.get_attributes('Policy')
policy = json.loads(attrs.get('Policy', "{}"))
return policy.get('Statement', {})
def test_correct_sid(self):
now = time.time()
topic_name = queue_name = "test_correct_sid%d" % (now)
timeout = 60
queue = self.sqsc.create_queue(queue_name, timeout)
self.addCleanup(self.sqsc.delete_queue, queue, True)
queue_arn = queue.arn
topic = self.snsc.create_topic(topic_name)
topic_arn = topic['CreateTopicResponse']['CreateTopicResult']\
['TopicArn']
self.addCleanup(self.snsc.delete_topic, topic_arn)
expected_sid = hashlib.md5(
(topic_arn + queue_arn).encode('utf-8')).hexdigest()
resp = self.snsc.subscribe_sqs_queue(topic_arn, queue)
found_expected_sid = False
statements = self.get_policy_statements(queue)
for statement in statements:
if statement['Sid'] == expected_sid:
found_expected_sid = True
break
self.assertTrue(found_expected_sid)
def test_idempotent_subscribe(self):
now = time.time()
topic_name = queue_name = "test_idempotent_subscribe%d" % (now)
timeout = 60
queue = self.sqsc.create_queue(queue_name, timeout)
self.addCleanup(self.sqsc.delete_queue, queue, True)
initial_statements = self.get_policy_statements(queue)
queue_arn = queue.arn
topic = self.snsc.create_topic(topic_name)
topic_arn = topic['CreateTopicResponse']['CreateTopicResult']\
['TopicArn']
self.addCleanup(self.snsc.delete_topic, topic_arn)
resp = self.snsc.subscribe_sqs_queue(topic_arn, queue)
time.sleep(3)
first_subscribe_statements = self.get_policy_statements(queue)
self.assertEqual(len(first_subscribe_statements),
len(initial_statements) + 1)
resp2 = self.snsc.subscribe_sqs_queue(topic_arn, queue)
time.sleep(3)
second_subscribe_statements = self.get_policy_statements(queue)
self.assertEqual(len(second_subscribe_statements),
len(first_subscribe_statements))
def test_sns_expired(self):
c = SNSConnection()
self.assert_is_expired(c, SNS_EXPIRED, status=403)
def __init__(self, application, configuration=None, resource=None):
super(SNS, self).__init__(application, configuration, resource)
self.sns = SNSConnection()
def test_sns_expired(self):
c = SNSConnection(aws_access_key_id='aws_access_key_id',
aws_secret_access_key='aws_secret_access_key')
self.assert_is_expired(c, SNS_EXPIRED, status=403)
def create_sns_connection(key):
return SNSConnection(key.access_key_id, key.secret_key)
