tests/fake.websocket.js

   1 import Base64 from '../core/base64.js';
   2
   3 // PhantomJS can't create Event objects directly, so we need to use this
   4 function make_event(name, props) {
   5     const evt = document.createEvent('Event');
   6     evt.initEvent(name, true, true);
   7     if (props) {
   8         for (let prop in props) {
   9             evt[prop] = props[prop];
  10         }
  11     }
  12     return evt;
  13 }
  14
  15 export default class FakeWebSocket {
  16     constructor(uri, protocols) {
  17         this.url = uri;
  18         this.binaryType = "arraybuffer";
  19         this.extensions = "";
  20
  21         if (!protocols || typeof protocols === 'string') {
  22             this.protocol = protocols;
  23         } else {
  24             this.protocol = protocols[0];
  25         }
  26
  27         this._send_queue = new Uint8Array(20000);
  28
  29         this.readyState = FakeWebSocket.CONNECTING;
  30         this.bufferedAmount = 0;
  31
  32         this.__is_fake = true;
  33     }
  34
  35     close(code, reason) {
  36         this.readyState = FakeWebSocket.CLOSED;
  37         if (this.onclose) {
  38             this.onclose(make_event("close", { 'code': code, 'reason': reason, 'wasClean': true }));
  39         }
  40     }
  41
  42     send(data) {
  43         if (this.protocol == 'base64') {
  44             data = Base64.decode(data);
  45         } else {
  46             data = new Uint8Array(data);
  47         }
  48         this._send_queue.set(data, this.bufferedAmount);
  49         this.bufferedAmount += data.length;
  50     }
  51
  52     _get_sent_data() {
  53         const res = new Uint8Array(this._send_queue.buffer, 0, this.bufferedAmount);
  54         this.bufferedAmount = 0;
  55         return res;
  56     }
  57
  58     _open() {
  59         this.readyState = FakeWebSocket.OPEN;
  60         if (this.onopen) {
  61             this.onopen(make_event('open'));
  62         }
  63     }
  64
  65     _receive_data(data) {
  66         this.onmessage(make_event("message", { 'data': data }));
  67     }
  68 }
  69
  70 FakeWebSocket.OPEN = WebSocket.OPEN;
  71 FakeWebSocket.CONNECTING = WebSocket.CONNECTING;
  72 FakeWebSocket.CLOSING = WebSocket.CLOSING;
  73 FakeWebSocket.CLOSED = WebSocket.CLOSED;
  74
  75 FakeWebSocket.__is_fake = true;
  76
  77 FakeWebSocket.replace = () => {
  78     if (!WebSocket.__is_fake) {
  79         const real_version = WebSocket;
  80         // eslint-disable-next-line no-global-assign
  81         WebSocket = FakeWebSocket;
  82         FakeWebSocket.__real_version = real_version;
  83     }
  84 };
  85
  86 FakeWebSocket.restore = () => {
  87     if (WebSocket.__is_fake) {
  88         // eslint-disable-next-line no-global-assign
  89         WebSocket = WebSocket.__real_version;
  90     }
  91 };