Skip to content

Instantly share code, notes, and snippets.

@chrishenry
Created August 10, 2014 22:28
Show Gist options
  • Save chrishenry/3b45327e2c94de8dcf49 to your computer and use it in GitHub Desktop.
Save chrishenry/3b45327e2c94de8dcf49 to your computer and use it in GitHub Desktop.
Attempt to mock AWSQueryConnection
class SQSProfileName(AWSMockServiceTestCase):
connection_class = SQSConnection
profile_name = 'prod'
def setUp(self):
super(SQSProfileName, self).setUp()
def mock_init(self, aws_access_key_id=None, aws_secret_access_key=None,
is_secure=True, port=None, proxy=None, proxy_port=None,
proxy_user=None, proxy_pass=None, host=None, debug=0,
https_connection_factory=None, path='/', security_token=None,
validate_certs=True, profile_name=None):
print 'in mock'
super(AWSQueryConnection, self).__init__(host, aws_access_key_id,
aws_secret_access_key,
is_secure, port, proxy,
proxy_port, proxy_user, proxy_pass,
debug, https_connection_factory, path,
security_token=security_token,
validate_certs=validate_certs,
profile_name=profile_name)
@attr(sqs=True)
@attr(chris=True)
def test_profile_name_gets_passed(self):
my_mock = mock.MagicMock(wraps=AWSQueryConnection.__init__)
my_mock = self.mock_init
with mock.patch('boto.connection.AWSQueryConnection.__init__', my_mock):
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)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment