test_rabbit.py

"""The most basic example of message passing via RabbitMQ

RabbitMQ message path
===
Producer -> Exchange -> Routing (rules) -> Queue(s) -> Consumer
Protocol used (other are also available): AMQP

Run broker (in docker)
===
`docker run -it --rm --name rabbitmq -p 5672:5672 -p 15672:15672 rabbitmq:3-management`

Access admin panel via: `http://localhost:15672` creds: guest/guest
"""

# import threading
import pika
import pytest


BROKER = "localhost"
QUEUE = "hello"
BODY = "Hello World!"


@pytest.fixture
def broker():
    pass


@pytest.fixture
def pub():
    connection = pika.BlockingConnection(pika.ConnectionParameters(BROKER))
    channel = connection.channel()
    channel.queue_declare(queue=QUEUE)
    yield channel
    connection.close()


@pytest.fixture
def sub():
    conn = pika.BlockingConnection(pika.ConnectionParameters(BROKER))
    channel = conn.channel()
    channel.queue_declare(queue=QUEUE)
    yield channel
    conn.close()


def test_basic_get(pub, sub):
    # messages = []

    # sub.basic_consume(
    #     queue=QUEUE,
    #     auto_ack=True,
    #     on_message_callback=lambda c, m, p, b: messages.append(b))

    # consume_thread = threading.Thread(target=channel.start_consuming)
    # consume_thread.start()

    pub.basic_publish(exchange="", routing_key=QUEUE, body=BODY)

    # XXX: thread safety
    # sub.stop_consuming()
    # thread.join(0)
    # assert messages.pop() == BODY

    msg = sub.basic_get(QUEUE)
    assert msg[-1].decode() == BODY