diff options
Diffstat (limited to 'tests/unit/sqs/test_connection.py')
| -rw-r--r-- | tests/unit/sqs/test_connection.py | 51 |
1 files changed, 43 insertions, 8 deletions
diff --git a/tests/unit/sqs/test_connection.py b/tests/unit/sqs/test_connection.py index 62dce12b..b1735de1 100644 --- a/tests/unit/sqs/test_connection.py +++ b/tests/unit/sqs/test_connection.py @@ -22,14 +22,17 @@ # IN THE SOFTWARE. # -from tests.unit import unittest -from tests.unit import AWSMockServiceTestCase +from tests.unit import AWSMockServiceTestCase, MockServiceWithConfigTestCase + +from tests.compat import mock from boto.sqs.connection import SQSConnection from boto.sqs.regioninfo import SQSRegionInfo from boto.sqs.message import RawMessage from boto.sqs.queue import Queue +from boto.connection import AWSQueryConnection +from nose.plugins.attrib import attr class SQSAuthParams(AWSMockServiceTestCase): connection_class = SQSConnection @@ -50,6 +53,7 @@ class SQSAuthParams(AWSMockServiceTestCase): </ResponseMetadata> </CreateQueueResponse>""" + @attr(sqs=True) def test_auth_service_name_override(self): self.set_http_response(status_code=200) # We can use the auth_service_name to change what service @@ -61,6 +65,7 @@ class SQSAuthParams(AWSMockServiceTestCase): self.assertIn('us-east-1/service_override/aws4_request', self.actual_request.headers['Authorization']) + @attr(sqs=True) def test_class_attribute_can_set_service_name(self): self.set_http_response(status_code=200) # The SQS class has an 'AuthServiceName' param of 'sqs': @@ -72,6 +77,7 @@ class SQSAuthParams(AWSMockServiceTestCase): self.assertIn('us-east-1/sqs/aws4_request', self.actual_request.headers['Authorization']) + @attr(sqs=True) def test_auth_region_name_is_automatically_updated(self): region = SQSRegionInfo(name='us-west-2', endpoint='us-west-2.queue.amazonaws.com') @@ -84,12 +90,12 @@ class SQSAuthParams(AWSMockServiceTestCase): self.set_http_response(status_code=200) self.service_connection.create_queue('my_queue') - + # Note the region name below is 'us-west-2'. self.assertIn('us-west-2/sqs/aws4_request', self.actual_request.headers['Authorization']) - - + + @attr(sqs=True) def test_set_get_auth_service_and_region_names(self): self.service_connection.auth_service_name = 'service_name' self.service_connection.auth_region_name = 'region_name' @@ -98,16 +104,43 @@ class SQSAuthParams(AWSMockServiceTestCase): 'service_name') self.assertEqual(self.service_connection.auth_region_name, 'region_name') + @attr(sqs=True) def test_get_queue_with_owner_account_id_returns_queue(self): - + self.set_http_response(status_code=200) self.service_connection.create_queue('my_queue') - + self.service_connection.get_queue('my_queue', '599169622985') assert 'QueueOwnerAWSAccountId' in self.actual_request.params.keys() self.assertEquals(self.actual_request.params['QueueOwnerAWSAccountId'], '599169622985') +class SQSProfileName(MockServiceWithConfigTestCase): + connection_class = SQSConnection + profile_name = 'prod' + + def setUp(self): + super(SQSProfileName, self).setUp() + self.config = { + "profile prod": { + 'aws_access_key_id': 'access_key', + 'aws_secret_access_key': 'secret_access', + } + } + + @attr(sqs=True) + def test_profile_name_gets_passed(self): + + region = SQSRegionInfo(name='us-west-2', + endpoint='us-west-2.queue.amazonaws.com') + self.service_connection = SQSConnection( + https_connection_factory=self.https_connection_factory, + region=region, + profile_name=self.profile_name) + self.initialize_service_connection() + self.set_http_response(status_code=200) + + self.assertEquals(self.service_connection.profile_name, self.profile_name) class SQSMessageAttributesParsing(AWSMockServiceTestCase): connection_class = SQSConnection @@ -143,6 +176,7 @@ class SQSMessageAttributesParsing(AWSMockServiceTestCase): </ResponseMetadata> </ReceiveMessageResponse>""" + @attr(sqs=True) def test_message_attribute_response(self): self.set_http_response(status_code=200) @@ -187,6 +221,7 @@ class SQSSendMessageAttributes(AWSMockServiceTestCase): </SendMessageResponse> """ + @attr(sqs=True) def test_send_message_attributes(self): self.set_http_response(status_code=200) @@ -242,6 +277,7 @@ class SQSSendBatchMessageAttributes(AWSMockServiceTestCase): </SendMessageBatchResponse> """ + @attr(sqs=True) def test_send_message_attributes(self): self.set_http_response(status_code=200) @@ -273,6 +309,5 @@ class SQSSendBatchMessageAttributes(AWSMockServiceTestCase): 'Version': '2012-11-05' }) - if __name__ == '__main__': unittest.main() |