Created
August 10, 2014 22:28
-
-
Save chrishenry/3b45327e2c94de8dcf49 to your computer and use it in GitHub Desktop.
Attempt to mock AWSQueryConnection
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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