示例#1
0
    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',
        }
示例#2
0
 def test_message_to_python(self):
     message = Mock()
     message.headers = {}
     message.properties = {}
     assert self.channel.message_to_python(message)
示例#3
0
文件: test_amqp.py 项目: Scalr/celery
 def test_dump_message(self):
     m = Mock()
     m.body = 'the quick brown fox'
     m.properties = {'a': 1}
     m.delivery_info = {'exchange': 'bar'}
     assert dump_message(m)