def _context(self,
sig,
rate_limits=True,
events=True,
utc=True,
limit=None):
self.assertTrue(sig.type.Strategy)
reserved = Mock()
consumer = Mock()
consumer.task_buckets = defaultdict(lambda: None)
if limit:
bucket = TokenBucket(rate(limit), capacity=1)
consumer.task_buckets[sig.task] = bucket
consumer.controller.state.revoked = set()
consumer.disable_rate_limits = not rate_limits
consumer.event_dispatcher.enabled = events
s = sig.type.start_strategy(self.app, consumer, task_reserved=reserved)
self.assertTrue(s)
message = task_message_from_sig(
self.app,
sig,
utc=utc,
TaskMessage=self.get_message_class(),
)
message = self.prepare_message(message)
yield self.Context(sig, s, reserved, consumer, message)
def get_request(self, sig, Request=Request, **kwargs):
return Request(task_message_from_sig(self.app, sig),
on_ack=Mock(name='on_ack'),
on_reject=Mock(name='on_reject'),
eventer=Mock(name='eventer'),
app=self.app,
connection_errors=(socket.error, ),
task=sig.type,
**kwargs)
def get_request(self, sig, Request=Request, **kwargs):
return Request(
task_message_from_sig(self.app, sig),
on_ack=Mock(name='on_ack'),
on_reject=Mock(name='on_reject'),
eventer=Mock(name='eventer'),
app=self.app,
connection_errors=(socket.error,),
task=sig.type,
**kwargs
)
def _context(self, sig,
rate_limits=True, events=True, utc=True, limit=None):
self.assertTrue(sig.type.Strategy)
reserved = Mock()
consumer = Mock()
consumer.task_buckets = defaultdict(lambda: None)
if limit:
bucket = TokenBucket(rate(limit), capacity=1)
consumer.task_buckets[sig.task] = bucket
consumer.controller.state.revoked = set()
consumer.disable_rate_limits = not rate_limits
consumer.event_dispatcher.enabled = events
s = sig.type.start_strategy(self.app, consumer, task_reserved=reserved)
self.assertTrue(s)
message = task_message_from_sig(self.app, sig, utc=utc)
yield self.Context(sig, s, reserved, consumer, message)
def task_context(self, sig, **kwargs):
x, on_task = get_task_callback(self.app, **kwargs)
message = task_message_from_sig(self.app, sig)
strategy = x.obj.strategies[sig.task] = Mock(name='strategy')
return x, on_task, message, strategy
def task_context(self, sig, **kwargs):
x, on_task = get_task_callback(self.app, **kwargs)
message = task_message_from_sig(self.app, sig)
strategy = x.obj.strategies[sig.task] = Mock(name='strategy')
return x, on_task, message, strategy
