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