diff --git a/test/README.md b/test/README.md new file mode 100644 index 0000000..990ff84 --- /dev/null +++ b/test/README.md @@ -0,0 +1,23 @@ +# Test compose + +This folder contains a simple Docker Compose stack to run the Go server together with an MQTT broker and a Python subscriber client. + +Quickstart: + +```bash +cd test +docker compose up --build +``` + +- The Go server is built from the repository `Dockerfile` and is configured via environment variables in `docker-compose.yml`. +- The MQTT broker is `eclipse-mosquitto` and is exposed on port `1883`. +- The `subscriber` service runs `subscribe.py` and prints any messages published to the `lambdaiot` topic. + +To run the Python subscriber locally instead of in Docker: + +```bash +python3 -m venv .venv +source .venv/bin/activate +pip install -r requirements.txt +MQTT_BROKER=localhost:1883 MQTT_TOPIC=lambdaiot python subscribe.py +``` diff --git a/test/docker-compose.yml b/test/docker-compose.yml new file mode 100644 index 0000000..76c951c --- /dev/null +++ b/test/docker-compose.yml @@ -0,0 +1,42 @@ +version: "3.8" + +services: + server: + build: + context: .. + dockerfile: Dockerfile + ports: + - "8080:8080" + environment: + - SERVER_ADDRESS=0.0.0.0 + - SERVER_PORT=8080 + - JWT_SECRET=secret + - MQTT_BROKER=tcp://mosquitto:1883 + - MQTT_CLIENT_ID=lambdaiot-server + - MQTT_TOPIC=lambdaiot + depends_on: + - mosquitto + + mosquitto: + image: eclipse-mosquitto:2.0 + ports: + - "1883:1883" + volumes: + - mosquitto_data:/mosquitto/data + - mosquitto_log:/mosquitto/log + + subscriber: + image: python:3.11-slim + volumes: + - ./subscribe.py:/app/subscribe.py + working_dir: /app + environment: + - MQTT_BROKER=mosquitto:1883 + - MQTT_TOPIC=lambdaiot + command: sh -c "pip install paho-mqtt && python subscribe.py" + depends_on: + - mosquitto + +volumes: + mosquitto_data: + mosquitto_log: diff --git a/test/requirements.txt b/test/requirements.txt new file mode 100644 index 0000000..5f9c1b2 --- /dev/null +++ b/test/requirements.txt @@ -0,0 +1 @@ +paho-mqtt==1.6.1 diff --git a/test/subscribe.py b/test/subscribe.py new file mode 100644 index 0000000..6f8d2f6 --- /dev/null +++ b/test/subscribe.py @@ -0,0 +1,38 @@ +#!/usr/bin/env python3 +import os +import sys +import time +import paho.mqtt.client as mqtt + +broker = os.getenv('MQTT_BROKER', 'localhost:1883') +topic = os.getenv('MQTT_TOPIC', 'lambdaiot') + +if broker.startswith('tcp://'): + broker = broker[len('tcp://'):] + +host, sep, port = broker.partition(':') +port = int(port) if sep else 1883 + +def on_connect(client, userdata, flags, rc): + print(f"Connected with result code {rc}") + client.subscribe(topic) + print(f"Subscribed to {topic}") + +def on_message(client, userdata, msg): + try: + payload = msg.payload.decode() + except Exception: + payload = str(msg.payload) + print(f"[{msg.topic}] {payload}") + +client = mqtt.Client() +client.on_connect = on_connect +client.on_message = on_message + +try: + client.connect(host, port, 60) +except Exception as e: + print("Could not connect to broker:", e) + sys.exit(1) + +client.loop_forever()