Merge pull request #22 from fermyon/mqtt

Add mqtt tests

Author: rylev

Cargo.lock

@@ -0,0 +1,12 @@
+[package]
+name = "outbound-mqtt-test-component"
+version = "0.1.0"
+edition = "2021"
+
+[lib]
+crate-type = ["cdylib"]
+
+[dependencies]
+anyhow = { workspace = true }
+helper = { workspace = true }
+wit-bindgen = { workspace = true }

components/outbound-mqtt/Cargo.toml

@@ -0,0 +1,34 @@
+use anyhow::Context as _;
+use helper::bindings::{
+    fermyon::spin2_0_0::mqtt::{self, Qos},
+    wasi::http0_2_0::types::{IncomingRequest, OutgoingResponse, ResponseOutparam},
+};
+
+struct Component;
+helper::gen_http_trigger_bindings!(Component);
+
+impl bindings::Guest for Component {
+    fn handle(request: IncomingRequest, response_out: ResponseOutparam) {
+        helper::handle_result(handle(request), response_out)
+    }
+}
+
+const MQTT_ADDRESS: &str = "MQTT_ADDRESS";
+const MQTT_USERNAME: &str = "MQTT_USERNAME";
+const MQTT_PASSWORD: &str = "MQTT_PASSWORD";
+
+fn handle(request: IncomingRequest) -> anyhow::Result<OutgoingResponse> {
+    let address = get_header(&request, &MQTT_ADDRESS.to_owned())?;
+    let username = get_header(&request, &MQTT_USERNAME.to_owned())?;
+    let password = get_header(&request, &MQTT_PASSWORD.to_owned())?;
+
+    let connection = mqtt::Connection::open(&address, &username, &password, 30)?;
+
+    connection.publish("telemetry-topic", &b"Eureka!".to_vec(), Qos::AtLeastOnce)?;
+
+    Ok(helper::ok_response())
+}
+
+fn get_header(request: &IncomingRequest, header_key: &String) -> anyhow::Result<String> {
+    helper::get_header(request, header_key).with_context(|| format!("no {} header", header_key))
+}

components/outbound-mqtt/src/lib.rs

@@ -18,7 +18,7 @@ impl bindings::Guest for Component {
 const REDIS_ADDRESS_HEADER: &str = "REDIS_ADDRESS";
 
 fn handle(request: IncomingRequest) -> anyhow::Result<OutgoingResponse> {
-    let Some(address) = helper::get_header(request, &REDIS_ADDRESS_HEADER.to_owned()) else {
+    let Some(address) = helper::get_header(&request, &REDIS_ADDRESS_HEADER.to_owned()) else {
         // Otherwise, return a 400 Bad Request response.
         return Ok(helper::response(
             400,

components/outbound-redis/src/lib.rs

@@ -176,6 +176,8 @@ pub enum Precondition {
     Sqlite,
     /// The test expects a Redis service to be available.
     Redis,
+    /// The test expects a Redis service to be available.
+    Mqtt,
 }
 
 #[derive(Debug, Clone, serde::Deserialize)]

crates/conformance-tests/src/config.rs

@@ -65,7 +65,7 @@ pub fn handle_result(result: anyhow::Result<OutgoingResponse>, response_out: Res
 }
 
 /// Get the value of a header from a request.
-pub fn get_header(request: IncomingRequest, header_key: &String) -> Option<String> {
+pub fn get_header(request: &IncomingRequest, header_key: &String) -> Option<String> {
     request
         .headers()
         .get(header_key)

crates/helper/src/lib.rs

@@ -0,0 +1,15 @@
+spin_manifest_version = 2
+
+[application]
+name = "outbound-mqtt"
+authors = ["Fermyon Engineering <engineering@fermyon.com>"]
+version = "0.1.0"
+
+[[trigger.http]]
+route = "/"
+component = "test"
+
+[component.test]
+source = "%{source=outbound-mqtt}"
+allowed_outbound_hosts = ["mqtt://localhost:%{port=1883}"]
+environment = { MQTT_ADDRESS = "mqtt://localhost:%{port=1883}?client_id=spintest" }

tests/outbound-mqtt/spin.toml

@@ -0,0 +1,40 @@
+{
+    "invocations": [
+        {
+            "request": {
+                "path": "/",
+                "headers": [
+                    {
+                        "name": "Host",
+                        "value": "example.com"
+                    },
+                    {
+                        "name": "MQTT_ADDRESS",
+                        "value": "mqtt://localhost:%{port=1883}?client_id=spintest"
+                    },
+                    {
+                        "name": "MQTT_USERNAME",
+                        "value": "user"
+                    },
+                    {
+                        "name": "MQTT_PASSWORD",
+                        "value": "password"
+                    },
+                ]
+            },
+            "response": {
+                "headers": [
+                    {
+                        "name": "Content-Length",
+                        "value": "0"
+                    },
+                    {
+                        "name": "Date",
+                        "optional": true
+                    }
+                ]
+            }
+        }
+    ],
+    "preconditions": [ { "kind": "mqtt" } ]
+}