def test_send_rpc_multi_message_reply_ignores_all_but_last(get_connection):
queue_declared = Event()
def response_greenthread():
with get_connection() as conn:
with conn.channel() as chan:
queue = nova.get_topic_queue(
'test_rpc', 'test', channel=chan)
queue.declare()
queue_declared.send(True)
body, msg = ifirst(
queue_iterator(queue, no_ack=True, timeout=2))
msgid, _, _, args = nova.parse_message(body)
exchange = nova.get_reply_exchange(msgid)
producer = Producer(chan, exchange=exchange, routing_key=msgid)
for _ in range(3):
msg = dict(
result='should ignore this message',
failure=None, ending=False)
producer.publish(msg)
eventlet.sleep(0.1)
msg = dict(result=args, failure=None, ending=False)
producer.publish(msg)
msg = dict(result=None, failure=None, ending=True)
producer.publish(msg)
g = eventlet.spawn_n(response_greenthread)
eventlet.sleep()
with get_connection() as conn:
ctx = context.get_admin_context()
queue_declared.wait()
resp = nova.send_rpc(
conn,
context=ctx,
exchange='test_rpc',
topic='test',
method='test_method',
args={'spam': 'shrub', },
timeout=3)
assert resp == {'spam': 'shrub', }
eventlet.sleep()
def check_greenthread_dead():
assert not g
assert_stops_raising(check_greenthread_dead)
def test_send_rpc_unknown_service(get_connection):
with get_connection() as conn:
ctx = context.get_admin_context()
with pytest.raises(UnknownService):
nova.send_rpc(
conn,
context=ctx,
exchange='test_rpc',
topic='test',
method='test_method',
args={'foo': 'bar', },
timeout=3)
def test_send_rpc_errors(get_connection):
queue_declared = Event()
def response_greenthread():
with get_connection() as conn:
with conn.channel() as chan:
queue = nova.get_topic_queue(
'test_rpc', 'test', channel=chan)
queue.declare()
queue_declared.send(True)
body, msg = ifirst(
queue_iterator(queue, no_ack=True, timeout=2))
msgid, _, _, _ = nova.parse_message(body)
exchange = nova.get_reply_exchange(msgid)
producer = Producer(chan, exchange=exchange, routing_key=msgid)
exc = Exception('error')
failure = (type(exc).__name__, str(exc))
msg = {'result': None, 'failure': failure, 'ending': False}
producer.publish(msg)
msg = {'result': None, 'failure': None, 'ending': True}
producer.publish(msg)
g = eventlet.spawn_n(response_greenthread)
eventlet.sleep(0)
with get_connection() as conn:
ctx = context.get_admin_context()
with pytest.raises(RemoteError):
queue_declared.wait()
nova.send_rpc(
conn,
context=ctx,
exchange='test_rpc',
topic='test',
method='test_method',
args={'foo': 'bar', },
timeout=3)
def check_greenthread_dead():
assert not g
assert_stops_raising(check_greenthread_dead)