def test_subscribe(monkeypatch):
# Test subscription and message receipt
mock_connection(monkeypatch, connection_success=True)
# connection, exchange, queue, key, msg
def on_message(self, exchange, queue, key, body):
assert (key == "mykey")
assert (body == "mybody")
connection = AsyncConnection(connection_params)
connection.connect()
queue = {"exchange": "exchange", "name": "name", "key": "key"}
connection.subscribe([queue, queue], on_message)
connection.publish(queue, "key", "mybody")
assert (consumed_message == "mybody")
# connection, exchange, queue, key, msg
def on_message_fail(self, exchange, queue, key, body):
raise Exception
connection.subscribe([queue, queue], on_message_fail)
connection.publish(queue, "key", "mybody")
assert (consumed_message == "mybody")
connection.disconnect()
def test_publish_failure(monkeypatch):
# Test publish failure
mock_connection(monkeypatch, connection_success=True)
connection = AsyncConnection(connection_params)
queue = {"name": "name", "key": "key"}
with pytest.raises(Exception):
connection.publish(queue, "key", "message")
def test_publish(monkeypatch):
# Test publish message
mock_connection(monkeypatch, connection_success=True)
connection = AsyncConnection(connection_params)
connection.connect()
queue = {"exchange": "exchange", "name": "name", "key": "key"}
connection.publish(queue, "key", "message")
assert (published_message == json.dumps("message"))
connection.disconnect()