Test Case updates

Pera-Swarm · Sep 1, 2023 · 35cea75 · 35cea75
1 parent 8624ac7
commit 35cea75
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 5 deletions.
diff --git a/src/router.ts b/src/router.ts
@@ -140,7 +140,7 @@ export class MQTTRouter {
     /**
      * Method for starting the MQTT handler
      */
-    start = () => {
+    start = (callback: Function) => {
         this._mqttClient.on('connect', () => {
             console.log('MQTT_Connecting...\n');
             this.setup();
@@ -153,6 +153,10 @@ export class MQTTRouter {
             }
             this._publishQueue.start();
             console.log(`MQTT_Router: Connected to the channel '${channel}'\n`);
+
+            if (callback) {
+                callback(`MQTT_Router: Connected to the channel ${channel}`);
+            }
         });
 
         this._mqttClient.on('error', (err) => {

diff --git a/test/index.js b/test/index.js
@@ -48,6 +48,23 @@ const SAMPLE_ROUTES = [
         fallbackRetainHandler: () => {
             console.log('Fallback method');
         }
+    },
+    {
+        topic: 'quite',
+        allowRetained: true,
+        subscribe: true,
+        publish: false,
+        type: 'String',
+        handler: (msg) => {
+            try {
+                process.exit(0);
+            } catch (err) {
+                console.log('Handler error:', err);
+            }
+        },
+        fallbackRetainHandler: () => {
+            console.log('Fallback method');
+        }
     }
 ];
 
@@ -73,8 +90,6 @@ router = new MQTTRouter(
     SAMPLE_ON_ERROR_FN
 );
 
-router.start();
-
 const samplePublishMessages = [
     {
         topic: 'sample-2',
@@ -99,9 +114,18 @@ const samplePublishMessages = [
     {
         topic: 'sample-2',
         data: 'Sample Data 6'
+    },
+    {
+        topic: 'quite',
+        data: 'End the test'
     }
 ];
 
-samplePublishMessages.forEach((element) => {
-    router.pushToPublishQueue(element.topic, element.data);
+router.start((response) => {
+    console.log(response);
+    samplePublishMessages.forEach((element) => {
+        console.log('Publishing', element.topic);
+        console.log(element.data);
+        router.pushToPublishQueue(element.topic, element.data);
+    });
 });