def test_task_properties(self):
"""Setup task properties."""
c = RpcClient()
c._send_command = MagicMock()
c.task('whatever', {"data": "x"})
c._send_command.assert_called_with(ANY,
'whatever',
declare_queue=False)
# This part shouldn't matter if legacy or not.
c = RpcClient(client_queue='test.my.queue')
c._send_command = MagicMock()
c.task('whatever', {"data": "x"})
c._send_command.assert_called_with(ANY,
'whatever',
declare_queue=False)
def test_rpc_properties(self):
"""Test setup rpc properties. """
c = RpcClient(client_queue='whatever.client')
c._send_command = MagicMock()
c.retrieve_messages = MagicMock()
c.rpc('whatever', {"data": "x"})
c._send_command.assert_called_with(ANY, 'whatever', {
"reply_to": "whatever.client",
"correlation_id": ANY
})
# This part shouldn't matter if legacy or not.
c = RpcClient(client_queue='test.my.queue')
c._send_command = MagicMock()
c.retrieve_messages = MagicMock()
c.rpc('whatever', {"data": "x"})
c._send_command.assert_called_with(ANY, 'whatever', {
"reply_to": "test.my.queue",
"correlation_id": ANY
})
def test_payload_standard_rpc(self):
"""Test command for standard RPC."""
c = RpcClient()
c._send_command = MagicMock()
c.retrieve_messages = MagicMock()
call_payload = {"test": 2}
c.rpc('do_that', call_payload)
c._send_command.assert_called_with(
call_payload,
'do_that', # routing key
ANY) # properties
from kombu import Exchange
from qurator.rpc.client import RpcClient
if __name__ == '__main__':
client = RpcClient()
reply = client.rpc(
'personal_handle',
{"handles": {
"personal": {
'id': '001bffe546d6705f4e004a80e30769bd'
}
}},
server_routing_key='api.handle')
print("Got reply: {!r}".format(reply))
import time
from kombu import Exchange as task_exchange
from qurator.rpc.client import RpcClient
if __name__ == '__main__':
client = RpcClient(exchange=task_exchange)
reply = client.task('missing_handle',
{'user_id': "4084a589401ebe665c313edecf001fcb"},
server_routing_key='api.mailer')
print("Got reply: {!r}".format(reply))
