def test_getresponse__real_response(self):
client = Mock(name='client')
client.add_request = passthrough(name='client.add_request')
callback = PromiseMock(name='callback')
x = AsyncHTTPConnection('aws.vandelay.com', http_client=client)
request = x.getresponse(callback)
x.http_client.add_request.assert_called_with(request)
buf = WhateverIO()
buf.write('The quick brown fox jumps')
headers = http.Headers({'X-Foo': 'Hello', 'X-Bar': 'World'})
response = http.Response(request, 200, headers, buf)
request.on_ready(response)
callback.assert_called()
wresponse = callback.call_args[0][0]
self.assertEqual(wresponse.read(), 'The quick brown fox jumps')
self.assertEqual(wresponse.status, 200)
self.assertEqual(wresponse.getheader('X-Foo'), 'Hello')
self.assertDictEqual(dict(wresponse.getheaders()), headers)
self.assertTrue(wresponse.msg)
self.assertTrue(wresponse.msg)
self.assertTrue(repr(wresponse))
def test_then(self):
callback = PromiseMock(name='callback')
x = http.Request('http://foo')
x.then(callback)
x.on_ready(1)
callback.assert_called_with(1)
def test_make_request(self, _):
x = AsyncAWSAuthConnection('aws.vandelay.com',
http_client=Mock(name='client'))
Conn = x.get_http_connection = Mock(name='get_http_connection')
callback = PromiseMock(name='callback')
ret = x.make_request('GET', '/foo', callback=callback)
self.assertIs(ret, 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},
'/',
'GET',
callback=callback,
)
self.x._mexe.assert_called()
request = self.x._mexe.call_args[0][0]
self.assertNotIn('Action', request.params)
self.assertEqual(request.params['Version'], self.x.APIVersion)
def test_mexe__with_sender(self, _):
x = AsyncAWSAuthConnection('aws.vandelay.com',
http_client=Mock(name='client'))
Conn = x.get_http_connection = Mock(name='get_http_connection')
request = x.build_base_http_request('GET', 'foo', '/auth')
sender = Mock(name='sender')
callback = PromiseMock(name='callback')
x._mexe(request, sender=sender, callback=callback)
sender.assert_called_with(
Conn.return_value,
request.method,
request.path,
request.body,
request.headers,
callback,
)
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},
'/',
'GET',
callback=callback,
)
self.x._mexe.assert_called()
request = self.x._mexe.call_args[0][0]
self.assertEqual(request.params['Action'], 'action')
self.assertEqual(request.params['Version'], self.x.APIVersion)
ret = _mexe(request, callback=callback)
self.assertIs(ret, callback)
Conn.return_value.request.assert_called()
Conn.return_value.getresponse.assert_called_with(callback=callback, )
def test_mexe(self, _):
x = AsyncAWSAuthConnection('aws.vandelay.com',
http_client=Mock(name='client'))
Conn = x.get_http_connection = Mock(name='get_http_connection')
request = x.build_base_http_request('GET', 'foo', '/auth')
callback = PromiseMock(name='callback')
x._mexe(request, callback=callback)
Conn.return_value.request.assert_called_with(
request.method,
request.path,
request.body,
request.headers,
)
Conn.return_value.getresponse.assert_called_with(callback=callback, )
no_callback_ret = x._mexe(request)
self.assertIsInstance(
no_callback_ret,
Thenable,
'_mexe always returns promise',
)
def setup(self):
self.x = AsyncSQSConnection('ak', 'sk')
self.x.get_object = Mock(name='X.get_object')
self.x.get_status = Mock(name='X.get_status')
self.x.get_list = Mock(nanme='X.get_list')
self.callback = PromiseMock(name='callback')
def setup(self):
self.queue = Mock(name='queue')
self.callback = PromiseMock(name='callback')
self.x = AsyncMessage(self.queue, 'body')
self.x.receipt_handle = uuid()
def mock_make_request(self):
self.x.make_request = Mock(name='make_request')
callback = PromiseMock(name='callback')
yield callback
def setup(self):
self.conn = Mock(name='connection')
self.x = AsyncQueue(self.conn, '/url')
self.callback = PromiseMock(name='callback')
