More reliable test in 'subscribe and publish QoS 1 in parallel'

moscajs · Jul 13, 2019 · 1d3bc57 · 1d3bc57
1 parent 10c3eb6
commit 1d3bc57
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/test/qos1.js b/test/qos1.js
@@ -686,18 +686,18 @@ test('subscribe and publish QoS 1 in parallel', function (t) {
   s.outStream.once('data', function (packet) {
     t.equal(packet.cmd, 'puback')
     t.equal(packet.messageId, 42, 'messageId must match')
-    s.outStream.once('data', function (packet) {
-      s.inStream.write({
-        cmd: 'puback',
-        messageId: packet.messageId
-      })
-      t.deepEqual(packet, expected, 'packet must match')
-      s.outStream.once('data', function (packet) {
-        t.equal(packet.cmd, 'suback')
+    s.outStream.on('data', function (packet) {
+      if (packet.cmd === 'suback') {
         t.deepEqual(packet.granted, [1])
         t.equal(packet.messageId, 24)
-        t.end()
-      })
+      }
+      if (packet.cmd === 'publish') {
+        s.inStream.write({
+          cmd: 'puback',
+          messageId: packet.messageId
+        })
+        t.deepEqual(packet, expected, 'packet must match')
+      }
     })
   })
 
@@ -717,4 +717,8 @@ test('subscribe and publish QoS 1 in parallel', function (t) {
     qos: 1,
     messageId: 42
   })
+
+  setTimeout(function () {
+    t.end()
+  }, 50)
 })