HiveMQ

Author: cristianrgreco

docs/modules/arangodb.md

@@ -6,9 +6,9 @@
 npm install @testcontainers/arangodb --save-dev
 ```
 
-## Example
+## Examples
 
-This example uses the following libraries:
+These examples use the following libraries:
 
 - [arangojs](https://www.npmjs.com/package/arangojs/v/6.0.0-alpha.0)
 

docs/modules/hivemq.md

@@ -1,30 +1,23 @@
-# HiveMQ MQTT Module
-
-This module allows automatic start up of [HiveMQ's](https://www.hivemq.com/) docker container within
-test suites, to enable programmatic testing of JavaScript based MQTT client applications.
+# HiveMQ
 
 ## Install
 
 ```bash
 npm install @testcontainers/hivemq --save-dev
 ```
 
-## Resources
-
-- [Community forum](https://community.hivemq.com/)
-- [HiveMQ website](https://www.hivemq.com/)
-- [MQTT Essentials](https://www.hivemq.com/mqtt-essentials/)
-- [MQTT 5 Essentials](https://www.hivemq.com/mqtt-5/)
+## Examples
 
-Please make sure to check out the hivemq-docs for the [Community Edition](https://github.com/hivemq/hivemq-community-edition/wiki/).
+These examples use the following libraries:
 
-!!! Info
-We are working to support the HiveMQ Enterprise Edition as outlined in the [Java Test Containers Module](https://java.testcontainers.org/modules/hivemq/).
+- [mqtt](https://www.npmjs.com/package/mqtt)
 
-## Examples
+        npm install mqtt
 
-<!--codeinclude-->
+Choose an image from the [container registry](https://hub.docker.com/r/hivemq/hivemq-ce) and substitute `IMAGE`.
 
-[Connect with a mqtt.js client to HiveMQ](../../packages/modules/hivemq/src/hivemq-container.test.ts) inside_block:connect
+### Produce/consume a message
 
+<!--codeinclude-->
+[](../../packages/modules/hivemq/src/hivemq-container.test.ts) inside_block:hivemqConnect
 <!--/codeinclude-->

packages/modules/hivemq/src/hivemq-container.test.ts

@@ -2,32 +2,28 @@ import mqtt from "mqtt";
 import { expect } from "vitest";
 import { getImage } from "../../../testcontainers/src/utils/test-helper";
 import { HiveMQContainer } from "./hivemq-container";
+
 const IMAGE = getImage(__dirname);
 
 describe("HiveMQContainer", { timeout: 240_000 }, () => {
-  // connect {
   it("should connect to HiveMQ Community Edition via MQTT.js", async () => {
+    // hivemqConnect {
     await using container = await new HiveMQContainer(IMAGE).start();
 
-    const testMqttClient = mqtt.connect(container.getConnectionString());
+    const mqttClient = await mqtt.connectAsync(container.getConnectionString());
 
-    const promise = new Promise<void>((resolve) => {
-      testMqttClient.on("message", (topic, message) => {
-        expect(message.toString()).toEqual("Test Message");
-        testMqttClient.end();
-        resolve();
-      });
+    const firstMessagePromise = new Promise<{ topic: string; message: Buffer }>((resolve, reject) => {
+      mqttClient.once("message", (topic, message) => resolve({ topic, message }));
+      mqttClient.once("error", (err) => reject(err));
     });
 
-    testMqttClient.on("connect", () => {
-      testMqttClient.subscribe("test", (error) => {
-        if (!error) {
-          testMqttClient.publish("test", "Test Message");
-        }
-      });
-    });
+    await mqttClient.subscribeAsync("test");
+    await mqttClient.publishAsync("test", "Test Message");
+
+    const { message } = await firstMessagePromise;
+    expect(message.toString()).toEqual("Test Message");
 
-    await expect(promise).resolves.toBeUndefined();
+    mqttClient.end();
+    // }
   });
-  // }
 });