def test_send_reply(self):
req = Mock()
req.content_type = 'application/json'
req.content_encoding = 'binary'
req.properties = {'reply_to': 'hello',
'correlation_id': 'world'}
channel = Mock()
exchange = Mock()
exchange.is_bound = True
exchange.channel = channel
producer = Mock()
producer.channel = channel
producer.channel.connection.client.declared_entities = set()
send_reply(exchange, req, {'hello': 'world'}, producer)
assert producer.publish.call_count
args = producer.publish.call_args
assert args[0][0] == {'hello': 'world'}
assert args[1] == {
'exchange': exchange,
'routing_key': 'hello',
'correlation_id': 'world',
'serializer': 'json',
'retry': False,
'retry_policy': None,
'content_encoding': 'binary',
}
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 TaskMessage(name, id=None, args=(), kwargs={}, callbacks=None,
errbacks=None, chain=None, shadow=None, utc=None, **options):
from celery import uuid
from kombu.serialization import dumps
id = id or uuid()
message = Mock(name='TaskMessage-{0}'.format(id))
message.headers = {
'id': id,
'task': name,
'shadow': shadow,
}
embed = {'callbacks': callbacks, 'errbacks': errbacks, 'chain': chain}
message.headers.update(options)
message.content_type, message.content_encoding, message.body = dumps(
(args, kwargs, embed), serializer='json',
)
message.payload = (args, kwargs, embed)
return message
def TaskMessage1(name, id=None, args=(), kwargs={}, callbacks=None,
errbacks=None, chain=None, **options):
from celery import uuid
from kombu.serialization import dumps
id = id or uuid()
message = Mock(name='TaskMessage-{0}'.format(id))
message.headers = {}
message.payload = {
'task': name,
'id': id,
'args': args,
'kwargs': kwargs,
'callbacks': callbacks,
'errbacks': errbacks,
}
message.payload.update(options)
message.content_type, message.content_encoding, message.body = dumps(
message.payload,
)
return message
def TaskMessage(name, id=None, args=(), kwargs={}, callbacks=None,
errbacks=None, chain=None, shadow=None, utc=None, **options):
# type: (str, str, Sequence, Mapping, Sequence[Signature],
# Sequence[Signature], Sequence[Signature],
# str, bool, **Any) -> Any
"""Create task message in protocol 2 format."""
from celery import uuid
from kombu.serialization import dumps
id = id or uuid()
message = Mock(name='TaskMessage-{0}'.format(id))
message.headers = {
'id': id,
'task': name,
'shadow': shadow,
}
embed = {'callbacks': callbacks, 'errbacks': errbacks, 'chain': chain}
message.headers.update(options)
message.content_type, message.content_encoding, message.body = dumps(
(args, kwargs, embed), serializer='json',
)
message.payload = (args, kwargs, embed)
return message
def TaskMessage1(name, id=None, args=(), kwargs={}, callbacks=None,
errbacks=None, chain=None, **options):
# type: (str, str, Sequence, Mapping, Sequence[Signature],
# Sequence[Signature], Sequence[Signature]) -> Any
"""Create task message in protocol 1 format."""
from celery import uuid
from kombu.serialization import dumps
id = id or uuid()
message = Mock(name='TaskMessage-{0}'.format(id))
message.headers = {}
message.payload = {
'task': name,
'id': id,
'args': args,
'kwargs': kwargs,
'callbacks': callbacks,
'errbacks': errbacks,
}
message.payload.update(options)
message.content_type, message.content_encoding, message.body = dumps(
message.payload,
)
return message
def Message(body, content_type='text/plain', content_encoding='utf-8'):
m = Mock(name='Message')
m.body = body
m.content_type = content_type
m.content_encoding = content_encoding
return m