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)
示例#2
0
 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
     )
示例#3
0
    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.app, consumer, task_reserved=reserved)
        self.assertTrue(s)

        message = Mock()
        body = body_from_sig(self.app, sig, utc=utc)

        yield self.Context(sig, s, reserved, consumer, message, body)
示例#4
0
    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)
示例#5
0
 def task_context(self, sig, **kwargs):
     x, on_task = get_task_callback(self.app, **kwargs)
     body = body_from_sig(self.app, sig)
     message = Mock()
     strategy = x.obj.strategies[sig.task] = Mock()
     return x, on_task, body, message, strategy
示例#6
0
 def task_context(self, sig, **kwargs):
     x, on_task = get_task_callback(self.app, **kwargs)
     body = body_from_sig(self.app, sig)
     message = Mock()
     strategy = x.obj.strategies[sig.task] = Mock()
     return x, on_task, body, message, strategy