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()