class test_AsyncAWSQueryConnection(AWSCase):
def setup(self):
session = boto3.session.Session(
aws_access_key_id='AAA',
aws_secret_access_key='AAAA',
region_name='us-west-2',
)
sqs_client = session.client('sqs')
self.x = AsyncAWSQueryConnection(sqs_client,
http_client=Mock(name='client'))
def test_make_request(self):
_mexe, self.x._mexe = self.x._mexe, Mock(name='_mexe')
Conn = self.x.get_http_connection = Mock(name='get_http_connection')
callback = PromiseMock(name='callback')
self.x.make_request(
'action', {'foo': 1}, 'https://foo.com/', 'GET', callback=callback,
)
self.x._mexe.assert_called()
request = self.x._mexe.call_args[0][0]
parsed = urlparse(request.url)
params = parse_qs(parsed.query)
assert params['Action'][0] == 'action'
ret = _mexe(request, callback=callback)
assert ret is callback
Conn.return_value.request.assert_called()
Conn.return_value.getresponse.assert_called_with(
callback=callback,
)
def test_make_request__no_action(self):
self.x._mexe = Mock(name='_mexe')
self.x.get_http_connection = Mock(name='get_http_connection')
callback = PromiseMock(name='callback')
self.x.make_request(
None, {'foo': 1}, 'http://foo.com/', 'GET', callback=callback,
)
self.x._mexe.assert_called()
request = self.x._mexe.call_args[0][0]
parsed = urlparse(request.url)
params = parse_qs(parsed.query)
assert 'Action' not in params
def Response(self, status, body):
r = Mock(name='response')
r.status = status
r.read.return_value = body
return r
@contextmanager
def mock_make_request(self):
self.x.make_request = Mock(name='make_request')
callback = PromiseMock(name='callback')
yield callback
def assert_make_request_called(self):
self.x.make_request.assert_called()
return self.x.make_request.call_args[1]['callback']
def __init__(self, sqs_connection, debug=0, region=None, **kwargs):
if boto3 is None:
raise ImportError('boto3 is not installed')
AsyncAWSQueryConnection.__init__(self,
sqs_connection,
region_name=region,
debug=debug,
**kwargs)
def setup(self):
session = boto3.session.Session(
aws_access_key_id='AAA',
aws_secret_access_key='AAAA',
region_name='us-west-2',
)
sqs_client = session.client('sqs')
self.x = AsyncAWSQueryConnection(sqs_client,
http_client=Mock(name='client'))
def __init__(self, sqs_connection, debug=0, region=None, **kwargs):
if boto3 is None:
raise ImportError('boto3 is not installed')
AsyncAWSQueryConnection.__init__(
self,
sqs_connection,
region_name=region, debug=debug,
**kwargs
)
def setup(self):
session = boto3.session.Session(
aws_access_key_id='AAA',
aws_secret_access_key='AAAA',
region_name='us-west-2',
)
sqs_client = session.client('sqs')
self.x = AsyncAWSQueryConnection(sqs_client,
http_client=Mock(name='client'))
class test_AsyncAWSQueryConnection(AWSCase):
def setup(self):
session = boto3.session.Session(
aws_access_key_id='AAA',
aws_secret_access_key='AAAA',
region_name='us-west-2',
)
sqs_client = session.client('sqs')
self.x = AsyncAWSQueryConnection(sqs_client,
http_client=Mock(name='client'))
def test_make_request(self):
_mexe, self.x._mexe = self.x._mexe, Mock(name='_mexe')
Conn = self.x.get_http_connection = Mock(name='get_http_connection')
callback = PromiseMock(name='callback')
self.x.make_request(
'action',
{'foo': 1},
'https://foo.com/',
'GET',
callback=callback,
)
self.x._mexe.assert_called()
request = self.x._mexe.call_args[0][0]
parsed = urlparse(request.url)
params = parse_qs(parsed.query)
assert params['Action'][0] == 'action'
ret = _mexe(request, callback=callback)
assert ret is callback
Conn.return_value.request.assert_called()
Conn.return_value.getresponse.assert_called_with(callback=callback, )
def test_make_request__no_action(self):
self.x._mexe = Mock(name='_mexe')
self.x.get_http_connection = Mock(name='get_http_connection')
callback = PromiseMock(name='callback')
self.x.make_request(
None,
{'foo': 1},
'http://foo.com/',
'GET',
callback=callback,
)
self.x._mexe.assert_called()
request = self.x._mexe.call_args[0][0]
parsed = urlparse(request.url)
params = parse_qs(parsed.query)
assert 'Action' not in params
@pytest.mark.parametrize('error_status_code', [
AsyncAWSQueryConnection.STATUS_CODE_REQUEST_TIMEOUT,
AsyncAWSQueryConnection.STATUS_CODE_NETWORK_CONNECT_TIMEOUT_ERROR,
AsyncAWSQueryConnection.STATUS_CODE_INTERNAL_ERROR,
AsyncAWSQueryConnection.STATUS_CODE_BAD_GATEWAY,
AsyncAWSQueryConnection.STATUS_CODE_SERVICE_UNAVAILABLE_ERROR,
AsyncAWSQueryConnection.STATUS_CODE_GATEWAY_TIMEOUT
])
def test_on_list_ready_error_response(self, error_status_code):
mocked_response_error = self.Response(error_status_code,
"error_status_code")
result = self.x._on_list_ready("parent", "markers", "operation",
mocked_response_error)
assert result == []
def Response(self, status, body):
r = Mock(name='response')
r.status = status
r.read.return_value = body
return r
@contextmanager
def mock_make_request(self):
self.x.make_request = Mock(name='make_request')
callback = PromiseMock(name='callback')
yield callback
def assert_make_request_called(self):
self.x.make_request.assert_called()
return self.x.make_request.call_args[1]['callback']
class test_AsyncAWSQueryConnection(AWSCase):
def setup(self):
session = boto3.session.Session(
aws_access_key_id='AAA',
aws_secret_access_key='AAAA',
region_name='us-west-2',
)
sqs_client = session.client('sqs')
self.x = AsyncAWSQueryConnection(sqs_client,
http_client=Mock(name='client'))
def test_make_request(self):
_mexe, self.x._mexe = self.x._mexe, Mock(name='_mexe')
Conn = self.x.get_http_connection = Mock(name='get_http_connection')
callback = PromiseMock(name='callback')
self.x.make_request(
'action',
{'foo': 1},
'https://foo.com/',
'GET',
callback=callback,
)
self.x._mexe.assert_called()
request = self.x._mexe.call_args[0][0]
parsed = urlparse(request.url)
params = parse_qs(parsed.query)
assert params['Action'][0] == 'action'
ret = _mexe(request, callback=callback)
assert ret is callback
Conn.return_value.request.assert_called()
Conn.return_value.getresponse.assert_called_with(callback=callback, )
def test_make_request__no_action(self):
self.x._mexe = Mock(name='_mexe')
self.x.get_http_connection = Mock(name='get_http_connection')
callback = PromiseMock(name='callback')
self.x.make_request(
None,
{'foo': 1},
'http://foo.com/',
'GET',
callback=callback,
)
self.x._mexe.assert_called()
request = self.x._mexe.call_args[0][0]
parsed = urlparse(request.url)
params = parse_qs(parsed.query)
assert 'Action' not in params
def Response(self, status, body):
r = Mock(name='response')
r.status = status
r.read.return_value = body
return r
@contextmanager
def mock_make_request(self):
self.x.make_request = Mock(name='make_request')
callback = PromiseMock(name='callback')
yield callback
def assert_make_request_called(self):
self.x.make_request.assert_called()
return self.x.make_request.call_args[1]['callback']
