Ejemplo n.º 1
0
def test_event_queue():
    queue = EventQueue("name")
    sent_event = {"event": "test"}
    queue.send_event(sent_event)
    received_event = queue.receive_event()

    assert received_event.pop("queue_name") == queue.queue_name
    assert received_event == sent_event
Ejemplo n.º 2
0
def test_event_queue_group():
    queue1 = EventQueue("name1")
    queue2 = EventQueue("name2")

    sent_event_1 = {"event": "test1"}
    queue1.send_event(sent_event_1)

    sent_event_2 = {"event": "test2"}
    queue2.send_event(sent_event_2)

    queue_group = EventQueueGroup([queue1, queue2])

    received_event_1 = queue_group.receive_event()
    assert received_event_1.pop("queue_name") == queue1.queue_name
    assert received_event_1 == sent_event_1

    received_event_2 = queue_group.receive_event()
    assert received_event_2.pop("queue_name") == queue2.queue_name
    assert received_event_2 == sent_event_2