def mock_req(event, url, **kwargs):
kwargs.setdefault('on_success', Mock(name='on_success'))
kwargs.setdefault('on_timeout', Mock(name='on_timeout'))
kwargs.setdefault('on_error', Mock(name='on_error'))
subscriber = Mock(name='subscriber')
subscriber.url = url
subscriber.content_type = MIME_JSON
return Request(
event, 'data', 'george', subscriber,
timeout=3.03,
**kwargs
)
def test_flush_buffer(self, group, app):
g = [None]
subscriber = Mock(name='subscriber')
subscriber.url = 'http://example.com/?e=1'
def group_consume_generator(arg):
g[0] = list(arg)
return group.return_value
group.side_effect = group_consume_generator
d = Dispatcher()
d.pending_outbound.extend(
Request('order.created', {}, None, subscriber)
for i in range(100)
)
d.flush_buffer()
assert g[0]
assert len(g[0]) == 100 / app.settings.THORN_CHUNKSIZE
group.return_value.delay.assert_called_once_with()
assert not d.pending_outbound
def MockQueue(self, url):
q = Mock(name='Queue')
q.url = url
return q
