feat: Add simple message publisher and client test utilities

Author: Alan Christie

simple_message_publisher.py

@@ -0,0 +1,38 @@
+#!/usr/bin/env python
+"""A simple RabbitMQ message publisher for local testing.
+Takes a 'routing key' and sends a message to the 'expected' AS exchange
+on a localhost RabbitMQ server.
+
+Usage: simple_message_publisher.py <routing_key>
+"""
+import asyncio
+import sys
+
+import aio_pika
+
+_ROUTING_KEY: str = sys.argv[1]
+
+_AMPQ_EXCHANGE: str = "event-streams"
+
+
+async def main() -> None:
+    """Publish a message."""
+    connection = await aio_pika.connect_robust(
+        "amqp://es:cheddar1963@localhost:5672",
+    )
+
+    async with connection:
+        channel = await connection.channel()
+
+        es_exchange = await channel.declare_exchange(
+            _AMPQ_EXCHANGE,
+            aio_pika.ExchangeType.DIRECT,
+        )
+        await es_exchange.publish(
+            aio_pika.Message(body=f"Hello {_ROUTING_KEY}".encode()),
+            routing_key=_ROUTING_KEY,
+        )
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

simple_ws_client.py

@@ -0,0 +1,26 @@
+#!/usr/bin/env python
+"""A simple WebSocket client for local testing.
+
+Usage: simple_ws_client.py <location>
+"""
+import asyncio
+import sys
+
+from simple_websocket import AioClient, ConnectionClosed
+
+_LOCATION: str = sys.argv[1]
+
+
+async def main():
+    """Connect to the WebSocket and just read messages."""
+    ws = await AioClient.connect(_LOCATION)
+    try:
+        while True:
+            data = await ws.receive()
+            print(f"< {data}")
+    except (KeyboardInterrupt, EOFError, ConnectionClosed):
+        await ws.close()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())