call callback at the end of all required responses or if there is an error

Author: Fabrice Ongenae

lib/engine_socketio.js

@@ -228,11 +228,15 @@ SocketIoEngine.prototype.step = function (requestSpec, ee) {
             if (!err) {
               markEndTime(ee, context, startedAt);
             }
-            // Stop listening on the response channel only if all messages were received
-            if (requiredResponsesCountsByChannel[channel] - 1 === index) {
+            // Stop listening on the response channel only if all messages were received or if there is an error
+            if (requiredResponsesCountsByChannel[channel] - 1 === index || err) {
               socketio.off(channel);
             }
-            return callback(err, context);
+
+            // Call the callback function if all channel messages were received or if there is an error
+            if (requiredResponsesCount === currentResponsesCount || err) {
+              callback(err, context);
+            }
           });
         });
       });