def test_wait_exception_before_wait(self, queue_consumer):
correlation_id = 1
event = ConsumeEvent(queue_consumer, correlation_id)
exc = ExampleError()
event.send_exception(exc)
with pytest.raises(ExampleError):
event.wait()
assert not queue_consumer.get_message.called
def test_wait_exception_before_wait(self, queue_consumer):
correlation_id = 1
event = ConsumeEvent(queue_consumer, correlation_id)
exc = ExampleError()
event.send_exception(exc)
with pytest.raises(ExampleError):
event.wait()
assert not queue_consumer.get_message.called
def test_wait_already_disconnected(self, queue_consumer):
correlation_id = 1
event = ConsumeEvent(queue_consumer, correlation_id)
exc = RpcConnectionError()
event.send_exception(exc)
with pytest.raises(RpcConnectionError):
event.wait()
assert not queue_consumer.get_message.called
