def get_request(self, sig, Request=Request, **kwargs): return Request(body_from_sig(self.app, sig), on_ack=Mock(), eventer=Mock(), app=self.app, connection_errors=(socket.error, ), task=sig.type, **kwargs)
def get_request(self, sig, Request=Request, **kwargs): return Request( body_from_sig(self.app, sig), on_ack=Mock(), eventer=Mock(), 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.disable_rate_limits = not rate_limits consumer.event_dispatcher.enabled = events s = sig.type.start_strategy(self.c, consumer, task_reserved=reserved) self.assertTrue(s) message = Mock() body = body_from_sig(self.c, sig, utc=utc) yield self.Context(sig, s, reserved, consumer, message, body)
def task_context(self, sig, **kwargs): x, on_task = get_task_callback(**kwargs) body = body_from_sig(self.app, sig) message = Mock() strategy = x.strategies[sig.task] = Mock() return x, on_task, body, message, strategy
def task_context(self, sig, **kwargs): x, on_task = get_task_callback(**kwargs) body = body_from_sig(self.app, sig) message = Mock() strategy = x.obj.strategies[sig.task] = Mock() return x, on_task, body, message, strategy