Fix traffic management in playback tests

1 files changed, 5 insertions, 4 deletions

diff --git a/tests/playback.js b/tests/playback.js
index b39c981..745e1f5 100644
--- a/tests/playback.js
+++ b/tests/playback.js
@@ -152,11 +152,12 @@ RecordingPlayer.prototype = {
 
     _doPacket: function () {
         // Avoid having excessive queue buildup in non-realtime mode
-        if (!this._trafficManagement && this._rfb._flushing) {
+        if (this._trafficManagement && this._rfb._flushing) {
             let player = this;
-            this._rfb.display.set_onFlush(function () {
-                this._rfb._display.set_onFlush(this._rfb._onFlush.bind(this._rfb));
-                this._rfb._onFlush();
+            let orig = this._rfb._display.get_onFlush();
+            this._rfb._display.set_onFlush(function () {
+                player._rfb._display.set_onFlush(orig);
+                player._rfb._onFlush();
                 player._doPacket();
             });
             return;