diff --git a/include/websock.js b/include/websock.js
index 782a1194..9d6a3060 100644
--- a/include/websock.js
+++ b/include/websock.js
@@ -371,8 +371,9 @@ function close() {
// Override internal functions for testing
// Takes a send function, returns reference to recv function
-function testMode(override_send) {
+function testMode(override_send, data_mode) {
test_mode = true;
+ mode = data_mode;
api.send = override_send;
api.close = function () {};
return recv_message;
diff --git a/tests/vnc_perf.html b/tests/vnc_perf.html
index d75394ec..10c89717 100644
--- a/tests/vnc_perf.html
+++ b/tests/vnc_perf.html
@@ -2,7 +2,6 @@
VNC Performance Benchmark
-
@@ -38,6 +37,8 @@
@@ -201,7 +202,7 @@
}
rfb = new RFB({'target': $D('VNC_canvas'),
'updateState': updateState});
- rfb.testMode(send_array);
+ rfb.testMode(send_array, VNC_frame_encoding);
}
diff --git a/tests/vnc_playback.html b/tests/vnc_playback.html
index 5ef0e449..18741f74 100644
--- a/tests/vnc_playback.html
+++ b/tests/vnc_playback.html
@@ -2,7 +2,6 @@
VNC Playback
-
@@ -40,6 +39,8 @@
@@ -98,7 +99,7 @@
mode = 'realtime';
}
- recv_message = rfb.testMode(send_array);
+ recv_message = rfb.testMode(send_array, VNC_frame_encoding);
next_iteration();
}