@mswjs/interceptors 0.26.4 → 0.26.5
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/lib/browser/interceptors/WebSocket/index.d.ts +40 -25
- package/lib/browser/interceptors/WebSocket/index.js +30 -31
- package/lib/browser/interceptors/WebSocket/index.js.map +1 -1
- package/lib/browser/interceptors/WebSocket/index.mjs +30 -31
- package/lib/browser/interceptors/WebSocket/index.mjs.map +1 -1
- package/package.json +1 -1
- package/src/interceptors/WebSocket/WebSocketClassTransport.ts +13 -9
- package/src/interceptors/WebSocket/WebSocketClientConnection.ts +18 -9
- package/src/interceptors/WebSocket/WebSocketOverride.ts +4 -4
- package/src/interceptors/WebSocket/WebSocketServerConnection.ts +10 -12
- package/src/interceptors/WebSocket/WebSocketTransport.ts +17 -12
- package/src/interceptors/WebSocket/index.ts +18 -11
|
@@ -2,26 +2,31 @@ import { I as Interceptor } from '../../Interceptor-b7c08a9f.js';
|
|
|
2
2
|
import '@open-draft/logger';
|
|
3
3
|
import 'strict-event-emitter';
|
|
4
4
|
|
|
5
|
-
type
|
|
6
|
-
type WebSocketTransportOnIncomingCallback = (event: MessageEvent<
|
|
7
|
-
type WebSocketTransportOnOutgoingCallback = (data:
|
|
5
|
+
type WebSocketData = string | ArrayBufferLike | Blob | ArrayBufferView;
|
|
6
|
+
type WebSocketTransportOnIncomingCallback = (event: MessageEvent<WebSocketData>) => void;
|
|
7
|
+
type WebSocketTransportOnOutgoingCallback = (data: WebSocketData) => void;
|
|
8
8
|
type WebSocketTransportOnCloseCallback = (event: CloseEvent) => void;
|
|
9
9
|
declare abstract class WebSocketTransport {
|
|
10
10
|
/**
|
|
11
|
-
*
|
|
12
|
-
* This is called when the client receives
|
|
13
|
-
*
|
|
14
|
-
*
|
|
15
|
-
* This way, we can trigger the "message" event
|
|
16
|
-
* on the mocked connection to let the user know.
|
|
11
|
+
* A callback for the incoming server events.
|
|
12
|
+
* This is called when the WebSocket client receives
|
|
13
|
+
* a message from the server.
|
|
17
14
|
*/
|
|
18
15
|
abstract onIncoming: WebSocketTransportOnIncomingCallback;
|
|
16
|
+
/**
|
|
17
|
+
* A callback for outgoing client events.
|
|
18
|
+
* This is called when the WebSocket client sends data.
|
|
19
|
+
*/
|
|
19
20
|
abstract onOutgoing: WebSocketTransportOnOutgoingCallback;
|
|
21
|
+
/**
|
|
22
|
+
* A callback for the close client event.
|
|
23
|
+
* This is called when the WebSocket client is closed.
|
|
24
|
+
*/
|
|
20
25
|
abstract onClose: WebSocketTransportOnCloseCallback;
|
|
21
26
|
/**
|
|
22
27
|
* Send the data from the server to this client.
|
|
23
28
|
*/
|
|
24
|
-
abstract send(data:
|
|
29
|
+
abstract send(data: WebSocketData): void;
|
|
25
30
|
/**
|
|
26
31
|
* Close the client connection.
|
|
27
32
|
*/
|
|
@@ -77,7 +82,7 @@ declare class WebSocketOverride extends EventTarget implements WebSocket {
|
|
|
77
82
|
/**
|
|
78
83
|
* @see https://websockets.spec.whatwg.org/#ref-for-dom-websocket-send%E2%91%A0
|
|
79
84
|
*/
|
|
80
|
-
send(data:
|
|
85
|
+
send(data: WebSocketData): void;
|
|
81
86
|
close(code?: number, reason?: string): void;
|
|
82
87
|
private [kClose];
|
|
83
88
|
addEventListener<K extends keyof WebSocketEventMap>(type: K, listener: (this: WebSocket, event: WebSocketEventMap[K]) => void, options?: boolean | AddEventListenerOptions): void;
|
|
@@ -94,18 +99,24 @@ declare class WebSocketOverride extends EventTarget implements WebSocket {
|
|
|
94
99
|
*/
|
|
95
100
|
|
|
96
101
|
declare const kEmitter$1: unique symbol;
|
|
102
|
+
interface WebSocketClientConnectionProtocol {
|
|
103
|
+
id: string;
|
|
104
|
+
url: URL;
|
|
105
|
+
send(data: WebSocketData): void;
|
|
106
|
+
close(code?: number, reason?: string): void;
|
|
107
|
+
}
|
|
97
108
|
/**
|
|
98
109
|
* The WebSocket client instance represents an incoming
|
|
99
110
|
* client connection. The user can control the connection,
|
|
100
111
|
* send and receive events.
|
|
101
112
|
*/
|
|
102
|
-
declare class WebSocketClientConnection {
|
|
103
|
-
|
|
104
|
-
|
|
113
|
+
declare class WebSocketClientConnection implements WebSocketClientConnectionProtocol {
|
|
114
|
+
private readonly socket;
|
|
115
|
+
private readonly transport;
|
|
105
116
|
readonly id: string;
|
|
106
117
|
readonly url: URL;
|
|
107
|
-
|
|
108
|
-
constructor(
|
|
118
|
+
private [kEmitter$1];
|
|
119
|
+
constructor(socket: WebSocket, transport: WebSocketTransport);
|
|
109
120
|
/**
|
|
110
121
|
* Listen for the outgoing events from the connected WebSocket client.
|
|
111
122
|
*/
|
|
@@ -117,7 +128,7 @@ declare class WebSocketClientConnection {
|
|
|
117
128
|
/**
|
|
118
129
|
* Send data to the connected client.
|
|
119
130
|
*/
|
|
120
|
-
send(data:
|
|
131
|
+
send(data: WebSocketData): void;
|
|
121
132
|
/**
|
|
122
133
|
* Close the WebSocket connection.
|
|
123
134
|
* @param {number} code A status code (see https://www.rfc-editor.org/rfc/rfc6455#section-7.4.1).
|
|
@@ -126,13 +137,17 @@ declare class WebSocketClientConnection {
|
|
|
126
137
|
close(code?: number, reason?: string): void;
|
|
127
138
|
}
|
|
128
139
|
|
|
140
|
+
/**
|
|
141
|
+
* Abstraction over the given mock `WebSocket` instance that allows
|
|
142
|
+
* for controlling that instance (e.g. sending and receiving messages).
|
|
143
|
+
*/
|
|
129
144
|
declare class WebSocketClassTransport extends WebSocketTransport {
|
|
130
|
-
protected readonly
|
|
145
|
+
protected readonly socket: WebSocketOverride;
|
|
131
146
|
onOutgoing: WebSocketTransportOnOutgoingCallback;
|
|
132
147
|
onIncoming: WebSocketTransportOnIncomingCallback;
|
|
133
148
|
onClose: WebSocketTransportOnCloseCallback;
|
|
134
|
-
constructor(
|
|
135
|
-
send(data:
|
|
149
|
+
constructor(socket: WebSocketOverride);
|
|
150
|
+
send(data: WebSocketData): void;
|
|
136
151
|
close(code: number, reason?: string): void;
|
|
137
152
|
}
|
|
138
153
|
|
|
@@ -143,15 +158,15 @@ declare const kEmitter: unique symbol;
|
|
|
143
158
|
* establish it by calling `server.connect()`.
|
|
144
159
|
*/
|
|
145
160
|
declare class WebSocketServerConnection {
|
|
146
|
-
private readonly
|
|
147
|
-
private readonly createConnection;
|
|
161
|
+
private readonly socket;
|
|
148
162
|
private readonly transport;
|
|
163
|
+
private readonly createConnection;
|
|
149
164
|
/**
|
|
150
165
|
* A WebSocket instance connected to the original server.
|
|
151
166
|
*/
|
|
152
167
|
private realWebSocket?;
|
|
153
168
|
private [kEmitter];
|
|
154
|
-
constructor(
|
|
169
|
+
constructor(socket: WebSocketOverride, transport: WebSocketClassTransport, createConnection: () => WebSocket);
|
|
155
170
|
/**
|
|
156
171
|
* Server ready state.
|
|
157
172
|
* Proxies the ready state of the original WebSocket instance,
|
|
@@ -178,7 +193,7 @@ declare class WebSocketServerConnection {
|
|
|
178
193
|
* server.send(new Blob(['hello']))
|
|
179
194
|
* server.send(new TextEncoder().encode('hello'))
|
|
180
195
|
*/
|
|
181
|
-
send(data:
|
|
196
|
+
send(data: WebSocketData): void;
|
|
182
197
|
}
|
|
183
198
|
|
|
184
199
|
type WebSocketEventMap$1 = {
|
|
@@ -206,4 +221,4 @@ declare class WebSocketInterceptor extends Interceptor<WebSocketEventMap$1> {
|
|
|
206
221
|
protected setup(): void;
|
|
207
222
|
}
|
|
208
223
|
|
|
209
|
-
export { WebSocketClientConnection, WebSocketEventMap$1 as WebSocketEventMap, WebSocketInterceptor,
|
|
224
|
+
export { WebSocketClientConnection, WebSocketClientConnectionProtocol, WebSocketData, WebSocketEventMap$1 as WebSocketEventMap, WebSocketInterceptor, WebSocketServerConnection, WebSocketTransport };
|
|
@@ -49,20 +49,20 @@ var CloseEvent = class extends Event {
|
|
|
49
49
|
// src/interceptors/WebSocket/WebSocketClientConnection.ts
|
|
50
50
|
var kEmitter = Symbol("kEmitter");
|
|
51
51
|
var WebSocketClientConnection = class {
|
|
52
|
-
constructor(
|
|
53
|
-
this.
|
|
52
|
+
constructor(socket, transport) {
|
|
53
|
+
this.socket = socket;
|
|
54
54
|
this.transport = transport;
|
|
55
55
|
this.id = _chunkLQSFVQBOjs.uuidv4.call(void 0, );
|
|
56
|
-
this.url = new URL(
|
|
56
|
+
this.url = new URL(socket.url);
|
|
57
57
|
this[kEmitter] = new EventTarget();
|
|
58
58
|
this.transport.onOutgoing = (data) => {
|
|
59
59
|
this[kEmitter].dispatchEvent(
|
|
60
|
-
bindEvent(this.
|
|
60
|
+
bindEvent(this.socket, new MessageEvent("message", { data }))
|
|
61
61
|
);
|
|
62
62
|
};
|
|
63
63
|
this.transport.onClose = (event) => {
|
|
64
64
|
this[kEmitter].dispatchEvent(
|
|
65
|
-
bindEvent(this.
|
|
65
|
+
bindEvent(this.socket, new CloseEvent("close", event))
|
|
66
66
|
);
|
|
67
67
|
};
|
|
68
68
|
}
|
|
@@ -103,10 +103,10 @@ kEmitter;
|
|
|
103
103
|
var _outvariant = require('outvariant');
|
|
104
104
|
var kEmitter2 = Symbol("kEmitter");
|
|
105
105
|
var WebSocketServerConnection = class {
|
|
106
|
-
constructor(
|
|
107
|
-
this.
|
|
108
|
-
this.createConnection = createConnection;
|
|
106
|
+
constructor(socket, transport, createConnection) {
|
|
107
|
+
this.socket = socket;
|
|
109
108
|
this.transport = transport;
|
|
109
|
+
this.createConnection = createConnection;
|
|
110
110
|
this[kEmitter2] = new EventTarget();
|
|
111
111
|
this.transport.onIncoming = (event) => {
|
|
112
112
|
const messageEvent = bindEvent(
|
|
@@ -119,14 +119,14 @@ var WebSocketServerConnection = class {
|
|
|
119
119
|
);
|
|
120
120
|
this[kEmitter2].dispatchEvent(messageEvent);
|
|
121
121
|
if (!messageEvent.defaultPrevented) {
|
|
122
|
-
this.
|
|
122
|
+
this.socket.dispatchEvent(
|
|
123
123
|
bindEvent(
|
|
124
124
|
/**
|
|
125
125
|
* @note Bind the forwarded original server events
|
|
126
126
|
* to the mock WebSocket instance so it would
|
|
127
127
|
* dispatch them straight away.
|
|
128
128
|
*/
|
|
129
|
-
this.
|
|
129
|
+
this.socket,
|
|
130
130
|
// Clone the message event again to prevent
|
|
131
131
|
// the "already being dispatched" exception.
|
|
132
132
|
new MessageEvent("message", {
|
|
@@ -159,7 +159,7 @@ var WebSocketServerConnection = class {
|
|
|
159
159
|
'Failed to call "connect()" on the original WebSocket instance: the connection already open'
|
|
160
160
|
);
|
|
161
161
|
const ws = this.createConnection();
|
|
162
|
-
this.
|
|
162
|
+
this.socket.addEventListener(
|
|
163
163
|
"close",
|
|
164
164
|
(event) => {
|
|
165
165
|
ws.close(event.code, event.reason);
|
|
@@ -170,9 +170,7 @@ var WebSocketServerConnection = class {
|
|
|
170
170
|
this.transport.onIncoming(event);
|
|
171
171
|
});
|
|
172
172
|
ws.addEventListener("error", () => {
|
|
173
|
-
this.
|
|
174
|
-
bindEvent(this.mockWebSocket, new Event("error"))
|
|
175
|
-
);
|
|
173
|
+
this.socket.dispatchEvent(bindEvent(this.socket, new Event("error")));
|
|
176
174
|
});
|
|
177
175
|
this.realWebSocket = ws;
|
|
178
176
|
}
|
|
@@ -208,7 +206,7 @@ var WebSocketServerConnection = class {
|
|
|
208
206
|
_outvariant.invariant.call(void 0,
|
|
209
207
|
realWebSocket,
|
|
210
208
|
'Failed to call "server.send()" for "%s": the connection is not open. Did you forget to call "await server.connect()"?',
|
|
211
|
-
this.
|
|
209
|
+
this.socket.url
|
|
212
210
|
);
|
|
213
211
|
if (realWebSocket.readyState === realWebSocket.CONNECTING) {
|
|
214
212
|
realWebSocket.addEventListener(
|
|
@@ -380,19 +378,19 @@ function getDataSize(data) {
|
|
|
380
378
|
|
|
381
379
|
// src/interceptors/WebSocket/WebSocketClassTransport.ts
|
|
382
380
|
var WebSocketClassTransport = class extends WebSocketTransport {
|
|
383
|
-
constructor(
|
|
381
|
+
constructor(socket) {
|
|
384
382
|
super();
|
|
385
|
-
this.
|
|
383
|
+
this.socket = socket;
|
|
386
384
|
this.onOutgoing = () => {
|
|
387
385
|
};
|
|
388
386
|
this.onIncoming = () => {
|
|
389
387
|
};
|
|
390
388
|
this.onClose = () => {
|
|
391
389
|
};
|
|
392
|
-
this.
|
|
390
|
+
this.socket.addEventListener("close", (event) => this.onClose(event), {
|
|
393
391
|
once: true
|
|
394
392
|
});
|
|
395
|
-
this.
|
|
393
|
+
this.socket[kOnSend] = (...args) => this.onOutgoing(...args);
|
|
396
394
|
}
|
|
397
395
|
send(data) {
|
|
398
396
|
queueMicrotask(() => {
|
|
@@ -405,17 +403,17 @@ var WebSocketClassTransport = class extends WebSocketTransport {
|
|
|
405
403
|
* mocked message events like the one below
|
|
406
404
|
* (must be dispatched on the client instance).
|
|
407
405
|
*/
|
|
408
|
-
this.
|
|
406
|
+
this.socket,
|
|
409
407
|
new MessageEvent("message", {
|
|
410
408
|
data,
|
|
411
|
-
origin: this.
|
|
409
|
+
origin: this.socket.url
|
|
412
410
|
})
|
|
413
411
|
);
|
|
414
|
-
this.
|
|
412
|
+
this.socket.dispatchEvent(message);
|
|
415
413
|
});
|
|
416
414
|
}
|
|
417
415
|
close(code, reason) {
|
|
418
|
-
this.
|
|
416
|
+
this.socket[kClose](code, reason);
|
|
419
417
|
}
|
|
420
418
|
};
|
|
421
419
|
|
|
@@ -434,17 +432,17 @@ var _WebSocketInterceptor = class extends _chunkEM6NYHQVjs.Interceptor {
|
|
|
434
432
|
const createConnection = () => {
|
|
435
433
|
return Reflect.construct(target, args, newTarget);
|
|
436
434
|
};
|
|
437
|
-
const
|
|
438
|
-
const transport = new WebSocketClassTransport(
|
|
435
|
+
const socket = new WebSocketOverride(url, protocols);
|
|
436
|
+
const transport = new WebSocketClassTransport(socket);
|
|
439
437
|
this.emitter.emit("connection", {
|
|
440
|
-
client: new WebSocketClientConnection(
|
|
438
|
+
client: new WebSocketClientConnection(socket, transport),
|
|
441
439
|
server: new WebSocketServerConnection(
|
|
442
|
-
|
|
443
|
-
|
|
444
|
-
|
|
440
|
+
socket,
|
|
441
|
+
transport,
|
|
442
|
+
createConnection
|
|
445
443
|
)
|
|
446
444
|
});
|
|
447
|
-
return
|
|
445
|
+
return socket;
|
|
448
446
|
}
|
|
449
447
|
});
|
|
450
448
|
globalThis.WebSocket = webSocketProxy.proxy;
|
|
@@ -459,5 +457,6 @@ WebSocketInterceptor.symbol = Symbol("websocket");
|
|
|
459
457
|
|
|
460
458
|
|
|
461
459
|
|
|
462
|
-
|
|
460
|
+
|
|
461
|
+
exports.WebSocketClientConnection = WebSocketClientConnection; exports.WebSocketInterceptor = WebSocketInterceptor; exports.WebSocketServerConnection = WebSocketServerConnection; exports.WebSocketTransport = WebSocketTransport;
|
|
463
462
|
//# sourceMappingURL=index.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../../../../src/interceptors/WebSocket/utils/bindEvent.ts","../../../../src/interceptors/WebSocket/utils/events.ts","../../../../src/interceptors/WebSocket/WebSocketClientConnection.ts","../../../../src/interceptors/WebSocket/WebSocketServerConnection.ts","../../../../src/interceptors/WebSocket/WebSocketTransport.ts","../../../../src/interceptors/WebSocket/WebSocketOverride.ts","../../../../src/interceptors/WebSocket/WebSocketClassTransport.ts","../../../../src/interceptors/WebSocket/index.ts"],"names":["kEmitter","invariant"],"mappings":";;;;;;;;AAEO,SAAS,UACd,QACA,OACuB;AACvB,SAAO,eAAe,OAAO,UAAU;AAAA,IACrC,YAAY;AAAA,IACZ,UAAU;AAAA,IACV,OAAO;AAAA,EACT,CAAC;AACD,SAAO;AACT;;;ACZA,IAAM,cAAc,OAAO,aAAa;AACxC,IAAM,oBAAoB,OAAO,mBAAmB;AAS7C,IAAM,yBAAN,cAA8C,aAAgB;AAAA,EAInE,YAAY,MAAc,MAA2B;AACnD,UAAM,MAAM,IAAI;AAChB,SAAK,WAAW,IAAI,CAAC,CAAC,KAAK;AAC3B,SAAK,iBAAiB,IAAI;AAAA,EAC5B;AAAA,EAEA,IAAI,aAAa;AACf,WAAO,KAAK,WAAW;AAAA,EACzB;AAAA,EAEA,IAAI,mBAAmB;AACrB,WAAO,KAAK,iBAAiB;AAAA,EAC/B;AAAA,EAEO,iBAAuB;AAC5B,QAAI,KAAK,cAAc,CAAC,KAAK,iBAAiB,GAAG;AAC/C,WAAK,iBAAiB,IAAI;AAAA,IAC5B;AAAA,EACF;AACF;AAtBG,aACA;AA6BI,IAAM,aAAN,cAAyB,MAAM;AAAA,EAKpC,YAAY,MAAc,OAAuB,CAAC,GAAG;AACnD,UAAM,MAAM,IAAI;AAChB,SAAK,OAAO,KAAK,SAAS,SAAY,IAAI,KAAK;AAC/C,SAAK,SAAS,KAAK,WAAW,SAAY,KAAK,KAAK;AACpD,SAAK,WAAW,KAAK,aAAa,SAAY,QAAQ,KAAK;AAAA,EAC7D;AACF;;;ACvCA,IAAM,WAAW,OAAO,UAAU;AAO3B,IAAM,4BAAN,MAAgC;AAAA,EAMrC,YACqB,IACA,WACnB;AAFmB;AACA;AAEnB,SAAK,KAAK,OAAO;AACjB,SAAK,MAAM,IAAI,IAAI,GAAG,GAAG;AACzB,SAAK,QAAQ,IAAI,IAAI,YAAY;AAIjC,SAAK,UAAU,aAAa,CAAC,SAAS;AACpC,WAAK,QAAQ,EAAE;AAAA,QACb,UAAU,KAAK,IAAI,IAAI,aAAa,WAAW,EAAE,KAAK,CAAC,CAAC;AAAA,MAC1D;AAAA,IACF;AAEA,SAAK,UAAU,UAAU,CAAC,UAAU;AAClC,WAAK,QAAQ,EAAE;AAAA,QACb,UAAU,KAAK,IAAI,IAAI,WAAW,SAAS,KAAK,CAAC;AAAA,MACnD;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKO,iBACL,OACA,UACA,SACM;AACN,SAAK,QAAQ,EAAE,iBAAiB,OAAO,UAA2B,OAAO;AAAA,EAC3E;AAAA;AAAA;AAAA;AAAA,EAKO,oBACL,OACA,UACA,SACM;AACN,SAAK,QAAQ,EAAE;AAAA,MACb;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKO,KAAK,MAA8B;AACxC,SAAK,UAAU,KAAK,IAAI;AAAA,EAC1B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOO,MAAM,MAAe,QAAuB;AACjD,SAAK,UAAU,MAAM,MAAM,MAAM;AAAA,EACnC;AACF;AAlEa;;;ACxBb,SAAS,iBAAiB;AAO1B,IAAMA,YAAW,OAAO,UAAU;AAO3B,IAAM,4BAAN,MAAgC;AAAA,EAOrC,YACmB,eACA,kBACA,WACjB;AAHiB;AACA;AACA;AAEjB,SAAKA,SAAQ,IAAI,IAAI,YAAY;AAMjC,SAAK,UAAU,aAAa,CAAC,UAAU;AAKrC,YAAM,eAAe;AAAA,QACnB,KAAK;AAAA,QACL,IAAI,uBAAuB,WAAW;AAAA,UACpC,MAAM,MAAM;AAAA,UACZ,QAAQ,MAAM;AAAA,UACd,YAAY;AAAA,QACd,CAAC;AAAA,MACH;AASA,WAAKA,SAAQ,EAAE,cAAc,YAAY;AAMzC,UAAI,CAAC,aAAa,kBAAkB;AAClC,aAAK,cAAc;AAAA,UACjB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,YAME,KAAK;AAAA;AAAA;AAAA,YAGL,IAAI,aAAa,WAAW;AAAA,cAC1B,MAAM,MAAM;AAAA,cACZ,QAAQ,MAAM;AAAA,YAChB,CAAC;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,IAAW,aAAqB;AAC9B,QAAI,KAAK,eAAe;AACtB,aAAO,KAAK,cAAc;AAAA,IAC5B;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKO,UAAgB;AACrB;AAAA,MACE,KAAK,eAAe;AAAA,MACpB;AAAA,IACF;AAEA,UAAM,KAAK,KAAK,iBAAiB;AAIjC,SAAK,cAAc;AAAA,MACjB;AAAA,MACA,CAAC,UAAU;AACT,WAAG,MAAM,MAAM,MAAM,MAAM,MAAM;AAAA,MACnC;AAAA,MACA,EAAE,MAAM,KAAK;AAAA,IACf;AAEA,OAAG,iBAAiB,WAAW,CAAC,UAAU;AACxC,WAAK,UAAU,WAAW,KAAK;AAAA,IACjC,CAAC;AAID,OAAG,iBAAiB,SAAS,MAAM;AACjC,WAAK,cAAc;AAAA,QACjB,UAAU,KAAK,eAAe,IAAI,MAAM,OAAO,CAAC;AAAA,MAClD;AAAA,IACF,CAAC;AAED,SAAK,gBAAgB;AAAA,EACvB;AAAA;AAAA;AAAA;AAAA,EAKO,iBACL,OACA,UACA,SACM;AACN,SAAKA,SAAQ,EAAE;AAAA,MACb;AAAA,MACA,SAAS,KAAK,KAAK,aAAc;AAAA,MACjC;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKO,oBACL,OACA,UACA,SACM;AACN,SAAKA,SAAQ,EAAE;AAAA,MACb;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASO,KAAK,MAA8B;AACxC,UAAM,EAAE,cAAc,IAAI;AAC1B;AAAA,MACE;AAAA,MACA;AAAA,MACA,KAAK,cAAc;AAAA,IACrB;AAKA,QAAI,cAAc,eAAe,cAAc,YAAY;AACzD,oBAAc;AAAA,QACZ;AAAA,QACA,MAAM;AACJ,wBAAc,KAAK,IAAI;AAAA,QACzB;AAAA,QACA,EAAE,MAAM,KAAK;AAAA,MACf;AACA;AAAA,IACF;AAGA,kBAAc,KAAK,IAAI;AAAA,EACzB;AACF;AA7KWA;;;ACPJ,IAAe,qBAAf,MAAkC;AAsBzC;;;AClCA,SAAS,aAAAC,kBAAiB;AAc1B,IAAM,mCACJ;AAEK,IAAM,UAAU,OAAO,SAAS;AAChC,IAAM,SAAS,OAAO,QAAQ;AAE9B,IAAM,oBAAN,cAAgC,YAAiC;AAAA,EAwBtE,YAAY,KAAmB,WAAoC;AACjE,UAAM;AApBR,SAAS,aAAa,UAAU;AAChC,SAAS,OAAO,UAAU;AAC1B,SAAS,UAAU,UAAU;AAC7B,SAAS,SAAS,UAAU;AAS5B,SAAQ,UAAyC;AACjD,SAAQ,aAA8C;AACtD,SAAQ,WAA0C;AAClD,SAAQ,WAA0C;AAMhD,SAAK,MAAM,IAAI,SAAS;AACxB,SAAK,WAAW;AAChB,SAAK,aAAa;AAClB,SAAK,aAAa;AAClB,SAAK,aAAa,KAAK;AACvB,SAAK,iBAAiB;AAEtB,YAAQ,IAAI,MAAM,cAAc,KAAK,UAAU;AAC/C,mBAAe,MAAM;AACnB,cAAQ,IAAI,MAAM,cAAc,KAAK,IAAI;AACzC,WAAK,WAAW,YAAY,UAAU,CAAC,IAAI;AAE3C,WAAK,cAAc,UAAU,MAAM,IAAI,MAAM,MAAM,CAAC,CAAC;AAAA,IACvD,CAAC;AAAA,EACH;AAAA,EAEA,IAAI,OAAO,UAAyC;AAClD,SAAK,oBAAoB,QAAQ,KAAK,OAAO;AAC7C,SAAK,UAAU;AACf,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,QAAQ,QAAQ;AAAA,IACxC;AAAA,EACF;AAAA,EACA,IAAI,SAAwC;AAC1C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,UAAU,UAA2C;AACvD,SAAK;AAAA,MACH;AAAA,MACA,KAAK;AAAA,IACP;AACA,SAAK,aAAa;AAClB,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,WAAW,QAAQ;AAAA,IAC3C;AAAA,EACF;AAAA,EACA,IAAI,YAA6C;AAC/C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,QAAQ,UAAyC;AACnD,SAAK,oBAAoB,SAAS,KAAK,QAAQ;AAC/C,SAAK,WAAW;AAChB,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,SAAS,QAAQ;AAAA,IACzC;AAAA,EACF;AAAA,EACA,IAAI,UAAyC;AAC3C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,QAAQ,UAAyC;AACnD,SAAK,oBAAoB,SAAS,KAAK,QAAkC;AACzE,SAAK,WAAW;AAChB,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,SAAS,QAAQ;AAAA,IACzC;AAAA,EACF;AAAA,EACA,IAAI,UAAyC;AAC3C,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKO,KAAK,MAA8B;AACxC,QAAI,KAAK,eAAe,KAAK,YAAY;AACvC,WAAK,MAAM;AACX,YAAM,IAAI,aAAa,mBAAmB;AAAA,IAC5C;AAIA,QAAI,KAAK,eAAe,KAAK,WAAW,KAAK,eAAe,KAAK,QAAQ;AACvE;AAAA,IACF;AAIA,SAAK,kBAAkB,YAAY,IAAI;AAEvC,mBAAe,MAAM;AAhIzB;AAmIM,WAAK,iBAAiB;AAOtB,iBAAK,aAAL,8BAAgB;AAAA,IAClB,CAAC;AAAA,EACH;AAAA,EAEO,MAAM,OAAe,KAAM,QAAuB;AACvD,IAAAA,WAAU,MAAM,gCAAgC;AAChD,IAAAA;AAAA,MACE,SAAS,OAAS,QAAQ,OAAQ,QAAQ;AAAA,MAC1C;AAAA,IACF;AAEA,QAAI,KAAK,eAAe,KAAK,WAAW,KAAK,eAAe,KAAK,QAAQ;AACvE;AAAA,IACF;AAEA,SAAK,MAAM,EAAE,MAAM,MAAM;AAAA,EAC3B;AAAA,EAEA,EAlHS,SAkHA,OAAM,EAAE,OAAe,KAAM,QAAuB;AAC3D,SAAK,aAAa,KAAK;AAEvB,mBAAe,MAAM;AACnB,WAAK,aAAa,KAAK;AAKvB,UAAI,OAAO,OAAQ,QAAQ,MAAM;AAC/B,aAAK,cAAc,UAAU,MAAM,IAAI,MAAM,OAAO,CAAC,CAAC;AAAA,MACxD;AAEA,WAAK;AAAA,QACH;AAAA,UACE;AAAA,UACA,IAAI,WAAW,SAAS;AAAA,YACtB;AAAA,YACA;AAAA,YACA,UAAU,SAAS;AAAA,UACrB,CAAC;AAAA,QACH;AAAA,MACF;AAGA,WAAK,UAAU;AACf,WAAK,aAAa;AAClB,WAAK,WAAW;AAChB,WAAK,WAAW;AAAA,IAClB,CAAC;AAAA,EACH;AAAA,EAYO,iBACL,MACA,UACA,SACM;AACN,WAAO,MAAM;AAAA,MACX;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,oBACE,MACA,UACA,SACM;AACN,WAAO,MAAM,oBAAoB,MAAM,UAAU,OAAO;AAAA,EAC1D;AACF;AArMa,kBACK,aAAa,UAAU;AAD5B,kBAEK,OAAO,UAAU;AAFtB,kBAGK,UAAU,UAAU;AAHzB,kBAIK,SAAS,UAAU;AAmMrC,SAAS,YAAY,MAAgC;AACnD,MAAI,OAAO,SAAS,UAAU;AAC5B,WAAO,KAAK;AAAA,EACd;AAEA,MAAI,gBAAgB,MAAM;AACxB,WAAO,KAAK;AAAA,EACd;AAEA,SAAO,KAAK;AACd;;;AC3NO,IAAM,0BAAN,cAAsC,mBAAmB;AAAA,EAK9D,YAA+B,IAAuB;AACpD,UAAM;AADuB;AAJ/B,SAAO,aAAmD,MAAM;AAAA,IAAC;AACjE,SAAO,aAAmD,MAAM;AAAA,IAAC;AACjE,SAAO,UAA6C,MAAM;AAAA,IAAC;AAKzD,SAAK,GAAG,iBAAiB,SAAS,CAAC,UAAU,KAAK,QAAQ,KAAK,GAAG;AAAA,MAChE,MAAM;AAAA,IACR,CAAC;AACD,SAAK,GAAG,OAAO,IAAI,IAAI,SAAS,KAAK,WAAW,GAAG,IAAI;AAAA,EACzD;AAAA,EAEO,KAAK,MAA8B;AACxC,mBAAe,MAAM;AACnB,YAAM,UAAU;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,QASd,KAAK;AAAA,QACL,IAAI,aAAa,WAAW;AAAA,UAC1B;AAAA,UACA,QAAQ,KAAK,GAAG;AAAA,QAClB,CAAC;AAAA,MACH;AAEA,WAAK,GAAG,cAAc,OAAO;AAAA,IAC/B,CAAC;AAAA,EACH;AAAA,EAEO,MAAM,MAAc,QAAuB;AAMhD,SAAK,GAAG,MAAM,EAAE,MAAM,MAAM;AAAA,EAC9B;AACF;;;ACzBO,IAAM,wBAAN,cAAmC,YAA+B;AAAA,EAGvE,cAAc;AACZ,UAAM,sBAAqB,MAAM;AAAA,EACnC;AAAA,EAEU,mBAA4B;AAGpC,WAAO,OAAO,WAAW,cAAc;AAAA,EACzC;AAAA,EAEU,QAAc;AACtB,UAAM,iBAAiB,MAAM,UAAU,WAAW,WAAW;AAAA,MAC3D,WAAW,CACT,QACA,MACA,cACG;AACH,cAAM,CAAC,KAAK,SAAS,IAAI;AAEzB,cAAM,mBAAmB,MAAiB;AACxC,iBAAO,QAAQ,UAAU,QAAQ,MAAM,SAAS;AAAA,QAClD;AAKA,cAAM,SAAS,IAAI,kBAAkB,KAAK,SAAS;AACnD,cAAM,YAAY,IAAI,wBAAwB,MAAM;AAKpD,aAAK,QAAQ,KAAK,cAAc;AAAA,UAC9B,QAAQ,IAAI,0BAA0B,QAAQ,SAAS;AAAA,UACvD,QAAQ,IAAI;AAAA,YACV;AAAA,YACA;AAAA,YACA;AAAA,UACF;AAAA,QACF,CAAC;AAED,eAAO;AAAA,MACT;AAAA,IACF,CAAC;AAED,eAAW,YAAY,eAAe;AAEtC,SAAK,cAAc,KAAK,MAAM;AAC5B,qBAAe,OAAO;AAAA,IACxB,CAAC;AAAA,EACH;AACF;AAtDO,IAAM,uBAAN;AAAM,qBACJ,SAAS,OAAO,WAAW","sourcesContent":["type EventWithTarget<E extends Event, T> = E & { target: T }\n\nexport function bindEvent<E extends Event, T>(\n target: T,\n event: E\n): EventWithTarget<E, T> {\n Object.defineProperty(event, 'target', {\n enumerable: true,\n writable: true,\n value: target,\n })\n return event as EventWithTarget<E, T>\n}\n","const kCancelable = Symbol('kCancelable')\nconst kDefaultPrevented = Symbol('kDefaultPrevented')\n\n/**\n * A `MessageEvent` superset that supports event cancellation\n * in Node.js. It's rather non-intrusive so it can be safely\n * used in the browser as well.\n *\n * @see https://github.com/nodejs/node/issues/51767\n */\nexport class CancelableMessageEvent<T = any> extends MessageEvent<T> {\n [kCancelable]: boolean;\n [kDefaultPrevented]: boolean\n\n constructor(type: string, init: MessageEventInit<T>) {\n super(type, init)\n this[kCancelable] = !!init.cancelable\n this[kDefaultPrevented] = false\n }\n\n get cancelable() {\n return this[kCancelable]\n }\n\n get defaultPrevented() {\n return this[kDefaultPrevented]\n }\n\n public preventDefault(): void {\n if (this.cancelable && !this[kDefaultPrevented]) {\n this[kDefaultPrevented] = true\n }\n }\n}\n\ninterface CloseEventInit extends EventInit {\n code?: number\n reason?: string\n wasClean?: boolean\n}\n\nexport class CloseEvent extends Event {\n public code: number\n public reason: string\n public wasClean: boolean\n\n constructor(type: string, init: CloseEventInit = {}) {\n super(type, init)\n this.code = init.code === undefined ? 0 : init.code\n this.reason = init.reason === undefined ? '' : init.reason\n this.wasClean = init.wasClean === undefined ? false : init.wasClean\n }\n}\n","/**\n * WebSocket client class.\n * This represents an incoming WebSocket client connection.\n * @note Keep this class implementation-agnostic because it's\n * meant to be used over any WebSocket implementation\n * (not all of them follow the one from WHATWG).\n */\nimport type { WebSocketRawData, WebSocketTransport } from './WebSocketTransport'\nimport { WebSocketMessageListener } from './WebSocketOverride'\nimport { bindEvent } from './utils/bindEvent'\nimport { CloseEvent } from './utils/events'\nimport { uuidv4 } from '../../utils/uuid'\n\nconst kEmitter = Symbol('kEmitter')\n\n/**\n * The WebSocket client instance represents an incoming\n * client connection. The user can control the connection,\n * send and receive events.\n */\nexport class WebSocketClientConnection {\n public readonly id: string\n public readonly url: URL\n\n protected [kEmitter]: EventTarget\n\n constructor(\n protected readonly ws: WebSocket,\n protected readonly transport: WebSocketTransport\n ) {\n this.id = uuidv4()\n this.url = new URL(ws.url)\n this[kEmitter] = new EventTarget()\n\n // Emit outgoing client data (\"ws.send()\") as \"message\"\n // events on the client connection.\n this.transport.onOutgoing = (data) => {\n this[kEmitter].dispatchEvent(\n bindEvent(this.ws, new MessageEvent('message', { data }))\n )\n }\n\n this.transport.onClose = (event) => {\n this[kEmitter].dispatchEvent(\n bindEvent(this.ws, new CloseEvent('close', event))\n )\n }\n }\n\n /**\n * Listen for the outgoing events from the connected WebSocket client.\n */\n public addEventListener(\n event: string,\n listener: WebSocketMessageListener,\n options?: AddEventListenerOptions | boolean\n ): void {\n this[kEmitter].addEventListener(event, listener as EventListener, options)\n }\n\n /**\n * Removes the listener for the given event.\n */\n public removeEventListener(\n event: string,\n listener: WebSocketMessageListener,\n options?: EventListenerOptions | boolean\n ): void {\n this[kEmitter].removeEventListener(\n event,\n listener as EventListener,\n options\n )\n }\n\n /**\n * Send data to the connected client.\n */\n public send(data: WebSocketRawData): void {\n this.transport.send(data)\n }\n\n /**\n * Close the WebSocket connection.\n * @param {number} code A status code (see https://www.rfc-editor.org/rfc/rfc6455#section-7.4.1).\n * @param {string} reason A custom connection close reason.\n */\n public close(code?: number, reason?: string): void {\n this.transport.close(code, reason)\n }\n}\n","import { invariant } from 'outvariant'\nimport type { WebSocketOverride } from './WebSocketOverride'\nimport type { WebSocketRawData } from './WebSocketTransport'\nimport type { WebSocketClassTransport } from './WebSocketClassTransport'\nimport { bindEvent } from './utils/bindEvent'\nimport { CancelableMessageEvent } from './utils/events'\n\nconst kEmitter = Symbol('kEmitter')\n\n/**\n * The WebSocket server instance represents the actual production\n * WebSocket server connection. It's idle by default but you can\n * establish it by calling `server.connect()`.\n */\nexport class WebSocketServerConnection {\n /**\n * A WebSocket instance connected to the original server.\n */\n private realWebSocket?: WebSocket\n private [kEmitter]: EventTarget\n\n constructor(\n private readonly mockWebSocket: WebSocketOverride,\n private readonly createConnection: () => WebSocket,\n private readonly transport: WebSocketClassTransport\n ) {\n this[kEmitter] = new EventTarget()\n\n // Handle incoming events from the actual server.\n // The (mock) WebSocket instance will call this\n // whenever a \"message\" event from the actual server\n // is dispatched on it (the dispatch will be skipped).\n this.transport.onIncoming = (event) => {\n // Clone the event to dispatch it on this class\n // once again and prevent the \"already being dispatched\"\n // exception. Clone it here so we can observe this event\n // being prevented in the \"server.on()\" listeners.\n const messageEvent = bindEvent(\n this.realWebSocket!,\n new CancelableMessageEvent('message', {\n data: event.data,\n origin: event.origin,\n cancelable: true,\n })\n )\n\n /**\n * @note Emit \"message\" event on the WebSocketClassServer\n * instance to let the interceptor know about these\n * incoming events from the original server. In that listener,\n * the interceptor can modify or skip the event forwarding\n * to the mock WebSocket instance.\n */\n this[kEmitter].dispatchEvent(messageEvent)\n\n /**\n * @note Forward the incoming server events to the client.\n * Preventing the default on the message event stops this.\n */\n if (!messageEvent.defaultPrevented) {\n this.mockWebSocket.dispatchEvent(\n bindEvent(\n /**\n * @note Bind the forwarded original server events\n * to the mock WebSocket instance so it would\n * dispatch them straight away.\n */\n this.mockWebSocket,\n // Clone the message event again to prevent\n // the \"already being dispatched\" exception.\n new MessageEvent('message', {\n data: event.data,\n origin: event.origin,\n })\n )\n )\n }\n }\n }\n\n /**\n * Server ready state.\n * Proxies the ready state of the original WebSocket instance,\n * if set. If the original connection hasn't been established,\n * defaults to `-1`.\n */\n public get readyState(): number {\n if (this.realWebSocket) {\n return this.realWebSocket.readyState\n }\n\n return -1\n }\n\n /**\n * Open connection to the original WebSocket server.\n */\n public connect(): void {\n invariant(\n this.readyState === -1,\n 'Failed to call \"connect()\" on the original WebSocket instance: the connection already open'\n )\n\n const ws = this.createConnection()\n\n // Close the original connection when the (mock)\n // client closes, regardless of the reason.\n this.mockWebSocket.addEventListener(\n 'close',\n (event) => {\n ws.close(event.code, event.reason)\n },\n { once: true }\n )\n\n ws.addEventListener('message', (event) => {\n this.transport.onIncoming(event)\n })\n\n // Forward server errors to the WebSocket client as-is.\n // We may consider exposing them to the interceptor in the future.\n ws.addEventListener('error', () => {\n this.mockWebSocket.dispatchEvent(\n bindEvent(this.mockWebSocket, new Event('error'))\n )\n })\n\n this.realWebSocket = ws\n }\n\n /**\n * Listen for the incoming events from the original WebSocket server.\n */\n public addEventListener<K extends keyof WebSocketEventMap>(\n event: K,\n listener: (this: WebSocket, event: WebSocketEventMap[K]) => void,\n options?: AddEventListenerOptions | boolean\n ): void {\n this[kEmitter].addEventListener(\n event,\n listener.bind(this.realWebSocket!) as EventListener,\n options\n )\n }\n\n /**\n * Removes the listener for the given event.\n */\n public removeEventListener<K extends keyof WebSocketEventMap>(\n event: K,\n listener: (this: WebSocket, event: WebSocketEventMap[K]) => void,\n options?: EventListenerOptions | boolean\n ): void {\n this[kEmitter].removeEventListener(\n event,\n listener as EventListener,\n options\n )\n }\n\n /**\n * Send data to the original WebSocket server.\n * @example\n * server.send('hello')\n * server.send(new Blob(['hello']))\n * server.send(new TextEncoder().encode('hello'))\n */\n public send(data: WebSocketRawData): void {\n const { realWebSocket } = this\n invariant(\n realWebSocket,\n 'Failed to call \"server.send()\" for \"%s\": the connection is not open. Did you forget to call \"await server.connect()\"?',\n this.mockWebSocket.url\n )\n\n // Delegate the send to when the original connection is open.\n // Unlike the mock, connecting to the original server may take time\n // so we cannot call this on the next tick.\n if (realWebSocket.readyState === realWebSocket.CONNECTING) {\n realWebSocket.addEventListener(\n 'open',\n () => {\n realWebSocket.send(data)\n },\n { once: true }\n )\n return\n }\n\n // Send the data to the original WebSocket server.\n realWebSocket.send(data)\n }\n}\n","export type WebSocketRawData = string | ArrayBufferLike | Blob | ArrayBufferView\n\nexport type WebSocketTransportOnIncomingCallback = (\n event: MessageEvent<WebSocketRawData>\n) => void\n\nexport type WebSocketTransportOnOutgoingCallback = (\n data: WebSocketRawData\n) => void\n\nexport type WebSocketTransportOnCloseCallback = (event: CloseEvent) => void\n\nexport abstract class WebSocketTransport {\n /**\n * Listener for the incoming server events.\n * This is called when the client receives the\n * event from the original server connection.\n *\n * This way, we can trigger the \"message\" event\n * on the mocked connection to let the user know.\n */\n abstract onIncoming: WebSocketTransportOnIncomingCallback\n abstract onOutgoing: WebSocketTransportOnOutgoingCallback\n abstract onClose: WebSocketTransportOnCloseCallback\n\n /**\n * Send the data from the server to this client.\n */\n abstract send(data: WebSocketRawData): void\n\n /**\n * Close the client connection.\n */\n abstract close(code?: number, reason?: string): void\n}\n","import { invariant } from 'outvariant'\nimport type { WebSocketRawData } from './WebSocketTransport'\nimport { bindEvent } from './utils/bindEvent'\nimport { CloseEvent } from './utils/events'\n\ntype WebSocketEventListener = (this: WebSocket, event: Event) => void\n\nexport type WebSocketMessageListener = (\n this: WebSocket,\n event: MessageEvent\n) => void\n\ntype WebSocketCloseListener = (this: WebSocket, event: CloseEvent) => void\n\nconst WEBSOCKET_CLOSE_CODE_RANGE_ERROR =\n 'InvalidAccessError: close code out of user configurable range'\n\nexport const kOnSend = Symbol('kOnSend')\nexport const kClose = Symbol('kClose')\n\nexport class WebSocketOverride extends EventTarget implements WebSocket {\n static readonly CONNECTING = WebSocket.CONNECTING\n static readonly OPEN = WebSocket.OPEN\n static readonly CLOSING = WebSocket.CLOSING\n static readonly CLOSED = WebSocket.CLOSED\n readonly CONNECTING = WebSocket.CONNECTING\n readonly OPEN = WebSocket.OPEN\n readonly CLOSING = WebSocket.CLOSING\n readonly CLOSED = WebSocket.CLOSED\n\n public url: string\n public protocol: string\n public extensions: string\n public binaryType: BinaryType\n public readyState: number\n public bufferedAmount: number\n\n private _onopen: WebSocketEventListener | null = null\n private _onmessage: WebSocketMessageListener | null = null\n private _onerror: WebSocketEventListener | null = null\n private _onclose: WebSocketCloseListener | null = null\n\n private [kOnSend]?: (data: WebSocketRawData) => void\n\n constructor(url: string | URL, protocols?: string | Array<string>) {\n super()\n this.url = url.toString()\n this.protocol = ''\n this.extensions = ''\n this.binaryType = 'blob'\n this.readyState = this.CONNECTING\n this.bufferedAmount = 0\n\n Reflect.set(this, 'readyState', this.CONNECTING)\n queueMicrotask(() => {\n Reflect.set(this, 'readyState', this.OPEN)\n this.protocol = protocols ? protocols[0] : 'ws'\n\n this.dispatchEvent(bindEvent(this, new Event('open')))\n })\n }\n\n set onopen(listener: WebSocketEventListener | null) {\n this.removeEventListener('open', this._onopen)\n this._onopen = listener\n if (listener !== null) {\n this.addEventListener('open', listener)\n }\n }\n get onopen(): WebSocketEventListener | null {\n return this._onopen\n }\n\n set onmessage(listener: WebSocketMessageListener | null) {\n this.removeEventListener(\n 'message',\n this._onmessage as WebSocketEventListener\n )\n this._onmessage = listener\n if (listener !== null) {\n this.addEventListener('message', listener)\n }\n }\n get onmessage(): WebSocketMessageListener | null {\n return this._onmessage\n }\n\n set onerror(listener: WebSocketEventListener | null) {\n this.removeEventListener('error', this._onerror)\n this._onerror = listener\n if (listener !== null) {\n this.addEventListener('error', listener)\n }\n }\n get onerror(): WebSocketEventListener | null {\n return this._onerror\n }\n\n set onclose(listener: WebSocketCloseListener | null) {\n this.removeEventListener('close', this._onclose as WebSocketEventListener)\n this._onclose = listener\n if (listener !== null) {\n this.addEventListener('close', listener)\n }\n }\n get onclose(): WebSocketCloseListener | null {\n return this._onclose\n }\n\n /**\n * @see https://websockets.spec.whatwg.org/#ref-for-dom-websocket-send%E2%91%A0\n */\n public send(data: WebSocketRawData): void {\n if (this.readyState === this.CONNECTING) {\n this.close()\n throw new DOMException('InvalidStateError')\n }\n\n // Sending when the socket is about to close\n // discards the sent data.\n if (this.readyState === this.CLOSING || this.readyState === this.CLOSED) {\n return\n }\n\n // Buffer the data to send in this even loop\n // but send it in the next.\n this.bufferedAmount += getDataSize(data)\n\n queueMicrotask(() => {\n // This is a bit optimistic but since no actual data transfer\n // is involved, all the data will be \"sent\" on the next tick.\n this.bufferedAmount = 0\n\n /**\n * @note Notify the parent about outgoing data.\n * This notifies the transport and the connection\n * listens to the outgoing data to emit the \"message\" event.\n */\n this[kOnSend]?.(data)\n })\n }\n\n public close(code: number = 1000, reason?: string): void {\n invariant(code, WEBSOCKET_CLOSE_CODE_RANGE_ERROR)\n invariant(\n code === 1000 || (code >= 3000 && code <= 4999),\n WEBSOCKET_CLOSE_CODE_RANGE_ERROR\n )\n\n if (this.readyState === this.CLOSING || this.readyState === this.CLOSED) {\n return\n }\n\n this[kClose](code, reason)\n }\n\n private [kClose](code: number = 1000, reason?: string): void {\n this.readyState = this.CLOSING\n\n queueMicrotask(() => {\n this.readyState = this.CLOSED\n\n // Non-user-configurable close status codes\n // represent connection termination and must\n // emit the \"error\" event before closing.\n if (code > 1000 && code <= 1015) {\n this.dispatchEvent(bindEvent(this, new Event('error')))\n }\n\n this.dispatchEvent(\n bindEvent(\n this,\n new CloseEvent('close', {\n code,\n reason,\n wasClean: code === 1000,\n })\n )\n )\n\n // Remove all event listeners once the socket is closed.\n this._onopen = null\n this._onmessage = null\n this._onerror = null\n this._onclose = null\n })\n }\n\n public addEventListener<K extends keyof WebSocketEventMap>(\n type: K,\n listener: (this: WebSocket, event: WebSocketEventMap[K]) => void,\n options?: boolean | AddEventListenerOptions\n ): void\n public addEventListener(\n type: string,\n listener: EventListenerOrEventListenerObject,\n options?: boolean | AddEventListenerOptions\n ): void\n public addEventListener(\n type: unknown,\n listener: unknown,\n options?: unknown\n ): void {\n return super.addEventListener(\n type as string,\n listener as EventListener,\n options as AddEventListenerOptions\n )\n }\n\n removeEventListener<K extends keyof WebSocketEventMap>(\n type: K,\n callback: EventListenerOrEventListenerObject | null,\n options?: boolean | EventListenerOptions\n ): void {\n return super.removeEventListener(type, callback, options)\n }\n}\n\nfunction getDataSize(data: WebSocketRawData): number {\n if (typeof data === 'string') {\n return data.length\n }\n\n if (data instanceof Blob) {\n return data.size\n }\n\n return data.byteLength\n}\n","import { bindEvent } from './utils/bindEvent'\nimport {\n WebSocketRawData,\n WebSocketTransport,\n WebSocketTransportOnCloseCallback,\n WebSocketTransportOnIncomingCallback,\n WebSocketTransportOnOutgoingCallback,\n} from './WebSocketTransport'\nimport { kOnSend, kClose, WebSocketOverride } from './WebSocketOverride'\n\nexport class WebSocketClassTransport extends WebSocketTransport {\n public onOutgoing: WebSocketTransportOnOutgoingCallback = () => {}\n public onIncoming: WebSocketTransportOnIncomingCallback = () => {}\n public onClose: WebSocketTransportOnCloseCallback = () => {}\n\n constructor(protected readonly ws: WebSocketOverride) {\n super()\n\n this.ws.addEventListener('close', (event) => this.onClose(event), {\n once: true,\n })\n this.ws[kOnSend] = (...args) => this.onOutgoing(...args)\n }\n\n public send(data: WebSocketRawData): void {\n queueMicrotask(() => {\n const message = bindEvent(\n /**\n * @note Setting this event's \"target\" to the\n * WebSocket override instance is important.\n * This way it can tell apart original incoming events\n * (must be forwarded to the transport) from the\n * mocked message events like the one below\n * (must be dispatched on the client instance).\n */\n this.ws,\n new MessageEvent('message', {\n data,\n origin: this.ws.url,\n })\n )\n\n this.ws.dispatchEvent(message)\n })\n }\n\n public close(code: number, reason?: string): void {\n /**\n * @note Call the internal close method directly\n * to allow closing the connection with the status codes\n * that are non-configurable by the user (> 1000 <= 1015).\n */\n this.ws[kClose](code, reason)\n }\n}\n","import { Interceptor } from '../../Interceptor'\nimport { WebSocketClientConnection } from './WebSocketClientConnection'\nimport { WebSocketServerConnection } from './WebSocketServerConnection'\nimport { WebSocketClassTransport } from './WebSocketClassTransport'\nimport { WebSocketOverride } from './WebSocketOverride'\n\nexport type { WebSocketRawData } from './WebSocketTransport'\nexport { WebSocketClientConnection, WebSocketServerConnection }\n\nexport type WebSocketEventMap = {\n connection: [\n args: {\n /**\n * The incoming WebSocket client connection.\n */\n client: WebSocketClientConnection\n\n /**\n * The original WebSocket server connection.\n */\n server: WebSocketServerConnection\n }\n ]\n}\n\n/**\n * Intercept the outgoing WebSocket connections created using\n * the global `WebSocket` class.\n */\nexport class WebSocketInterceptor extends Interceptor<WebSocketEventMap> {\n static symbol = Symbol('websocket')\n\n constructor() {\n super(WebSocketInterceptor.symbol)\n }\n\n protected checkEnvironment(): boolean {\n // Enable this interceptor in any environment\n // that has a global WebSocket API.\n return typeof globalThis.WebSocket !== 'undefined'\n }\n\n protected setup(): void {\n const webSocketProxy = Proxy.revocable(globalThis.WebSocket, {\n construct: (\n target,\n args: ConstructorParameters<typeof globalThis.WebSocket>,\n newTarget\n ) => {\n const [url, protocols] = args\n\n const createConnection = (): WebSocket => {\n return Reflect.construct(target, args, newTarget)\n }\n\n // All WebSocket instances are mocked and don't forward\n // any events to the original server (no connection established).\n // To forward the events, the user must use the \"server.send()\" API.\n const mockWs = new WebSocketOverride(url, protocols)\n const transport = new WebSocketClassTransport(mockWs)\n\n // The \"globalThis.WebSocket\" class stands for\n // the client-side connection. Assume it's established\n // as soon as the WebSocket instance is constructed.\n this.emitter.emit('connection', {\n client: new WebSocketClientConnection(mockWs, transport),\n server: new WebSocketServerConnection(\n mockWs,\n createConnection,\n transport\n ),\n })\n\n return mockWs\n },\n })\n\n globalThis.WebSocket = webSocketProxy.proxy\n\n this.subscriptions.push(() => {\n webSocketProxy.revoke()\n })\n }\n}\n"]}
|
|
1
|
+
{"version":3,"sources":["../../../../src/interceptors/WebSocket/utils/bindEvent.ts","../../../../src/interceptors/WebSocket/utils/events.ts","../../../../src/interceptors/WebSocket/WebSocketClientConnection.ts","../../../../src/interceptors/WebSocket/WebSocketServerConnection.ts","../../../../src/interceptors/WebSocket/WebSocketTransport.ts","../../../../src/interceptors/WebSocket/WebSocketOverride.ts","../../../../src/interceptors/WebSocket/WebSocketClassTransport.ts","../../../../src/interceptors/WebSocket/index.ts"],"names":["kEmitter","invariant"],"mappings":";;;;;;;;AAEO,SAAS,UACd,QACA,OACuB;AACvB,SAAO,eAAe,OAAO,UAAU;AAAA,IACrC,YAAY;AAAA,IACZ,UAAU;AAAA,IACV,OAAO;AAAA,EACT,CAAC;AACD,SAAO;AACT;;;ACZA,IAAM,cAAc,OAAO,aAAa;AACxC,IAAM,oBAAoB,OAAO,mBAAmB;AAS7C,IAAM,yBAAN,cAA8C,aAAgB;AAAA,EAInE,YAAY,MAAc,MAA2B;AACnD,UAAM,MAAM,IAAI;AAChB,SAAK,WAAW,IAAI,CAAC,CAAC,KAAK;AAC3B,SAAK,iBAAiB,IAAI;AAAA,EAC5B;AAAA,EAEA,IAAI,aAAa;AACf,WAAO,KAAK,WAAW;AAAA,EACzB;AAAA,EAEA,IAAI,mBAAmB;AACrB,WAAO,KAAK,iBAAiB;AAAA,EAC/B;AAAA,EAEO,iBAAuB;AAC5B,QAAI,KAAK,cAAc,CAAC,KAAK,iBAAiB,GAAG;AAC/C,WAAK,iBAAiB,IAAI;AAAA,IAC5B;AAAA,EACF;AACF;AAtBG,aACA;AA6BI,IAAM,aAAN,cAAyB,MAAM;AAAA,EAKpC,YAAY,MAAc,OAAuB,CAAC,GAAG;AACnD,UAAM,MAAM,IAAI;AAChB,SAAK,OAAO,KAAK,SAAS,SAAY,IAAI,KAAK;AAC/C,SAAK,SAAS,KAAK,WAAW,SAAY,KAAK,KAAK;AACpD,SAAK,WAAW,KAAK,aAAa,SAAY,QAAQ,KAAK;AAAA,EAC7D;AACF;;;ACvCA,IAAM,WAAW,OAAO,UAAU;AAc3B,IAAM,4BAAN,MAEP;AAAA,EAME,YACmB,QACA,WACjB;AAFiB;AACA;AAEjB,SAAK,KAAK,OAAO;AACjB,SAAK,MAAM,IAAI,IAAI,OAAO,GAAG;AAC7B,SAAK,QAAQ,IAAI,IAAI,YAAY;AAIjC,SAAK,UAAU,aAAa,CAAC,SAAS;AACpC,WAAK,QAAQ,EAAE;AAAA,QACb,UAAU,KAAK,QAAQ,IAAI,aAAa,WAAW,EAAE,KAAK,CAAC,CAAC;AAAA,MAC9D;AAAA,IACF;AAEA,SAAK,UAAU,UAAU,CAAC,UAAU;AAClC,WAAK,QAAQ,EAAE;AAAA,QACb,UAAU,KAAK,QAAQ,IAAI,WAAW,SAAS,KAAK,CAAC;AAAA,MACvD;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKO,iBACL,OACA,UACA,SACM;AACN,SAAK,QAAQ,EAAE,iBAAiB,OAAO,UAA2B,OAAO;AAAA,EAC3E;AAAA;AAAA;AAAA;AAAA,EAKO,oBACL,OACA,UACA,SACM;AACN,SAAK,QAAQ,EAAE;AAAA,MACb;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKO,KAAK,MAA2B;AACrC,SAAK,UAAU,KAAK,IAAI;AAAA,EAC1B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOO,MAAM,MAAe,QAAuB;AACjD,SAAK,UAAU,MAAM,MAAM,MAAM;AAAA,EACnC;AACF;AAlEW;;;ACjCX,SAAS,iBAAiB;AAO1B,IAAMA,YAAW,OAAO,UAAU;AAO3B,IAAM,4BAAN,MAAgC;AAAA,EAOrC,YACmB,QACA,WACA,kBACjB;AAHiB;AACA;AACA;AAEjB,SAAKA,SAAQ,IAAI,IAAI,YAAY;AAMjC,SAAK,UAAU,aAAa,CAAC,UAAU;AAKrC,YAAM,eAAe;AAAA,QACnB,KAAK;AAAA,QACL,IAAI,uBAAuB,WAAW;AAAA,UACpC,MAAM,MAAM;AAAA,UACZ,QAAQ,MAAM;AAAA,UACd,YAAY;AAAA,QACd,CAAC;AAAA,MACH;AASA,WAAKA,SAAQ,EAAE,cAAc,YAAY;AAMzC,UAAI,CAAC,aAAa,kBAAkB;AAClC,aAAK,OAAO;AAAA,UACV;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,YAME,KAAK;AAAA;AAAA;AAAA,YAGL,IAAI,aAAa,WAAW;AAAA,cAC1B,MAAM,MAAM;AAAA,cACZ,QAAQ,MAAM;AAAA,YAChB,CAAC;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,IAAW,aAAqB;AAC9B,QAAI,KAAK,eAAe;AACtB,aAAO,KAAK,cAAc;AAAA,IAC5B;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKO,UAAgB;AACrB;AAAA,MACE,KAAK,eAAe;AAAA,MACpB;AAAA,IACF;AAEA,UAAM,KAAK,KAAK,iBAAiB;AAIjC,SAAK,OAAO;AAAA,MACV;AAAA,MACA,CAAC,UAAU;AACT,WAAG,MAAM,MAAM,MAAM,MAAM,MAAM;AAAA,MACnC;AAAA,MACA,EAAE,MAAM,KAAK;AAAA,IACf;AAEA,OAAG,iBAAiB,WAAW,CAAC,UAAU;AACxC,WAAK,UAAU,WAAW,KAAK;AAAA,IACjC,CAAC;AAID,OAAG,iBAAiB,SAAS,MAAM;AACjC,WAAK,OAAO,cAAc,UAAU,KAAK,QAAQ,IAAI,MAAM,OAAO,CAAC,CAAC;AAAA,IACtE,CAAC;AAED,SAAK,gBAAgB;AAAA,EACvB;AAAA;AAAA;AAAA;AAAA,EAKO,iBACL,OACA,UACA,SACM;AACN,SAAKA,SAAQ,EAAE;AAAA,MACb;AAAA,MACA,SAAS,KAAK,KAAK,aAAc;AAAA,MACjC;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKO,oBACL,OACA,UACA,SACM;AACN,SAAKA,SAAQ,EAAE;AAAA,MACb;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASO,KAAK,MAA2B;AACrC,UAAM,EAAE,cAAc,IAAI;AAC1B;AAAA,MACE;AAAA,MACA;AAAA,MACA,KAAK,OAAO;AAAA,IACd;AAKA,QAAI,cAAc,eAAe,cAAc,YAAY;AACzD,oBAAc;AAAA,QACZ;AAAA,QACA,MAAM;AACJ,wBAAc,KAAK,IAAI;AAAA,QACzB;AAAA,QACA,EAAE,MAAM,KAAK;AAAA,MACf;AACA;AAAA,IACF;AAGA,kBAAc,KAAK,IAAI;AAAA,EACzB;AACF;AA3KWA;;;ACTJ,IAAe,qBAAf,MAAkC;AA6BzC;;;ACvCA,SAAS,aAAAC,kBAAiB;AAc1B,IAAM,mCACJ;AAEK,IAAM,UAAU,OAAO,SAAS;AAChC,IAAM,SAAS,OAAO,QAAQ;AAE9B,IAAM,oBAAN,cAAgC,YAAiC;AAAA,EAwBtE,YAAY,KAAmB,WAAoC;AACjE,UAAM;AApBR,SAAS,aAAa,UAAU;AAChC,SAAS,OAAO,UAAU;AAC1B,SAAS,UAAU,UAAU;AAC7B,SAAS,SAAS,UAAU;AAS5B,SAAQ,UAAyC;AACjD,SAAQ,aAA8C;AACtD,SAAQ,WAA0C;AAClD,SAAQ,WAA0C;AAMhD,SAAK,MAAM,IAAI,SAAS;AACxB,SAAK,WAAW;AAChB,SAAK,aAAa;AAClB,SAAK,aAAa;AAClB,SAAK,aAAa,KAAK;AACvB,SAAK,iBAAiB;AAEtB,YAAQ,IAAI,MAAM,cAAc,KAAK,UAAU;AAC/C,mBAAe,MAAM;AACnB,cAAQ,IAAI,MAAM,cAAc,KAAK,IAAI;AACzC,WAAK,WAAW,YAAY,UAAU,CAAC,IAAI;AAE3C,WAAK,cAAc,UAAU,MAAM,IAAI,MAAM,MAAM,CAAC,CAAC;AAAA,IACvD,CAAC;AAAA,EACH;AAAA,EAEA,IAAI,OAAO,UAAyC;AAClD,SAAK,oBAAoB,QAAQ,KAAK,OAAO;AAC7C,SAAK,UAAU;AACf,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,QAAQ,QAAQ;AAAA,IACxC;AAAA,EACF;AAAA,EACA,IAAI,SAAwC;AAC1C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,UAAU,UAA2C;AACvD,SAAK;AAAA,MACH;AAAA,MACA,KAAK;AAAA,IACP;AACA,SAAK,aAAa;AAClB,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,WAAW,QAAQ;AAAA,IAC3C;AAAA,EACF;AAAA,EACA,IAAI,YAA6C;AAC/C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,QAAQ,UAAyC;AACnD,SAAK,oBAAoB,SAAS,KAAK,QAAQ;AAC/C,SAAK,WAAW;AAChB,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,SAAS,QAAQ;AAAA,IACzC;AAAA,EACF;AAAA,EACA,IAAI,UAAyC;AAC3C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,QAAQ,UAAyC;AACnD,SAAK,oBAAoB,SAAS,KAAK,QAAkC;AACzE,SAAK,WAAW;AAChB,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,SAAS,QAAQ;AAAA,IACzC;AAAA,EACF;AAAA,EACA,IAAI,UAAyC;AAC3C,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKO,KAAK,MAA2B;AACrC,QAAI,KAAK,eAAe,KAAK,YAAY;AACvC,WAAK,MAAM;AACX,YAAM,IAAI,aAAa,mBAAmB;AAAA,IAC5C;AAIA,QAAI,KAAK,eAAe,KAAK,WAAW,KAAK,eAAe,KAAK,QAAQ;AACvE;AAAA,IACF;AAIA,SAAK,kBAAkB,YAAY,IAAI;AAEvC,mBAAe,MAAM;AAhIzB;AAmIM,WAAK,iBAAiB;AAOtB,iBAAK,aAAL,8BAAgB;AAAA,IAClB,CAAC;AAAA,EACH;AAAA,EAEO,MAAM,OAAe,KAAM,QAAuB;AACvD,IAAAA,WAAU,MAAM,gCAAgC;AAChD,IAAAA;AAAA,MACE,SAAS,OAAS,QAAQ,OAAQ,QAAQ;AAAA,MAC1C;AAAA,IACF;AAEA,QAAI,KAAK,eAAe,KAAK,WAAW,KAAK,eAAe,KAAK,QAAQ;AACvE;AAAA,IACF;AAEA,SAAK,MAAM,EAAE,MAAM,MAAM;AAAA,EAC3B;AAAA,EAEA,EAlHS,SAkHA,OAAM,EAAE,OAAe,KAAM,QAAuB;AAC3D,SAAK,aAAa,KAAK;AAEvB,mBAAe,MAAM;AACnB,WAAK,aAAa,KAAK;AAKvB,UAAI,OAAO,OAAQ,QAAQ,MAAM;AAC/B,aAAK,cAAc,UAAU,MAAM,IAAI,MAAM,OAAO,CAAC,CAAC;AAAA,MACxD;AAEA,WAAK;AAAA,QACH;AAAA,UACE;AAAA,UACA,IAAI,WAAW,SAAS;AAAA,YACtB;AAAA,YACA;AAAA,YACA,UAAU,SAAS;AAAA,UACrB,CAAC;AAAA,QACH;AAAA,MACF;AAGA,WAAK,UAAU;AACf,WAAK,aAAa;AAClB,WAAK,WAAW;AAChB,WAAK,WAAW;AAAA,IAClB,CAAC;AAAA,EACH;AAAA,EAYO,iBACL,MACA,UACA,SACM;AACN,WAAO,MAAM;AAAA,MACX;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,oBACE,MACA,UACA,SACM;AACN,WAAO,MAAM,oBAAoB,MAAM,UAAU,OAAO;AAAA,EAC1D;AACF;AArMa,kBACK,aAAa,UAAU;AAD5B,kBAEK,OAAO,UAAU;AAFtB,kBAGK,UAAU,UAAU;AAHzB,kBAIK,SAAS,UAAU;AAmMrC,SAAS,YAAY,MAA6B;AAChD,MAAI,OAAO,SAAS,UAAU;AAC5B,WAAO,KAAK;AAAA,EACd;AAEA,MAAI,gBAAgB,MAAM;AACxB,WAAO,KAAK;AAAA,EACd;AAEA,SAAO,KAAK;AACd;;;ACvNO,IAAM,0BAAN,cAAsC,mBAAmB;AAAA,EAK9D,YAA+B,QAA2B;AACxD,UAAM;AADuB;AAJ/B,SAAO,aAAmD,MAAM;AAAA,IAAC;AACjE,SAAO,aAAmD,MAAM;AAAA,IAAC;AACjE,SAAO,UAA6C,MAAM;AAAA,IAAC;AAKzD,SAAK,OAAO,iBAAiB,SAAS,CAAC,UAAU,KAAK,QAAQ,KAAK,GAAG;AAAA,MACpE,MAAM;AAAA,IACR,CAAC;AACD,SAAK,OAAO,OAAO,IAAI,IAAI,SAAS,KAAK,WAAW,GAAG,IAAI;AAAA,EAC7D;AAAA,EAEO,KAAK,MAA2B;AACrC,mBAAe,MAAM;AACnB,YAAM,UAAU;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,QASd,KAAK;AAAA,QACL,IAAI,aAAa,WAAW;AAAA,UAC1B;AAAA,UACA,QAAQ,KAAK,OAAO;AAAA,QACtB,CAAC;AAAA,MACH;AAEA,WAAK,OAAO,cAAc,OAAO;AAAA,IACnC,CAAC;AAAA,EACH;AAAA,EAEO,MAAM,MAAc,QAAuB;AAMhD,SAAK,OAAO,MAAM,EAAE,MAAM,MAAM;AAAA,EAClC;AACF;;;ACtBO,IAAM,wBAAN,cAAmC,YAA+B;AAAA,EAGvE,cAAc;AACZ,UAAM,sBAAqB,MAAM;AAAA,EACnC;AAAA,EAEU,mBAA4B;AAGpC,WAAO,OAAO,WAAW,cAAc;AAAA,EACzC;AAAA,EAEU,QAAc;AACtB,UAAM,iBAAiB,MAAM,UAAU,WAAW,WAAW;AAAA,MAC3D,WAAW,CACT,QACA,MACA,cACG;AACH,cAAM,CAAC,KAAK,SAAS,IAAI;AAEzB,cAAM,mBAAmB,MAAiB;AACxC,iBAAO,QAAQ,UAAU,QAAQ,MAAM,SAAS;AAAA,QAClD;AAKA,cAAM,SAAS,IAAI,kBAAkB,KAAK,SAAS;AACnD,cAAM,YAAY,IAAI,wBAAwB,MAAM;AAKpD,aAAK,QAAQ,KAAK,cAAc;AAAA,UAC9B,QAAQ,IAAI,0BAA0B,QAAQ,SAAS;AAAA,UACvD,QAAQ,IAAI;AAAA,YACV;AAAA,YACA;AAAA,YACA;AAAA,UACF;AAAA,QACF,CAAC;AAED,eAAO;AAAA,MACT;AAAA,IACF,CAAC;AAED,eAAW,YAAY,eAAe;AAEtC,SAAK,cAAc,KAAK,MAAM;AAC5B,qBAAe,OAAO;AAAA,IACxB,CAAC;AAAA,EACH;AACF;AAtDO,IAAM,uBAAN;AAAM,qBACJ,SAAS,OAAO,WAAW","sourcesContent":["type EventWithTarget<E extends Event, T> = E & { target: T }\n\nexport function bindEvent<E extends Event, T>(\n target: T,\n event: E\n): EventWithTarget<E, T> {\n Object.defineProperty(event, 'target', {\n enumerable: true,\n writable: true,\n value: target,\n })\n return event as EventWithTarget<E, T>\n}\n","const kCancelable = Symbol('kCancelable')\nconst kDefaultPrevented = Symbol('kDefaultPrevented')\n\n/**\n * A `MessageEvent` superset that supports event cancellation\n * in Node.js. It's rather non-intrusive so it can be safely\n * used in the browser as well.\n *\n * @see https://github.com/nodejs/node/issues/51767\n */\nexport class CancelableMessageEvent<T = any> extends MessageEvent<T> {\n [kCancelable]: boolean;\n [kDefaultPrevented]: boolean\n\n constructor(type: string, init: MessageEventInit<T>) {\n super(type, init)\n this[kCancelable] = !!init.cancelable\n this[kDefaultPrevented] = false\n }\n\n get cancelable() {\n return this[kCancelable]\n }\n\n get defaultPrevented() {\n return this[kDefaultPrevented]\n }\n\n public preventDefault(): void {\n if (this.cancelable && !this[kDefaultPrevented]) {\n this[kDefaultPrevented] = true\n }\n }\n}\n\ninterface CloseEventInit extends EventInit {\n code?: number\n reason?: string\n wasClean?: boolean\n}\n\nexport class CloseEvent extends Event {\n public code: number\n public reason: string\n public wasClean: boolean\n\n constructor(type: string, init: CloseEventInit = {}) {\n super(type, init)\n this.code = init.code === undefined ? 0 : init.code\n this.reason = init.reason === undefined ? '' : init.reason\n this.wasClean = init.wasClean === undefined ? false : init.wasClean\n }\n}\n","/**\n * WebSocket client class.\n * This represents an incoming WebSocket client connection.\n * @note Keep this class implementation-agnostic because it's\n * meant to be used over any WebSocket implementation\n * (not all of them follow the one from WHATWG).\n */\nimport type { WebSocketData, WebSocketTransport } from './WebSocketTransport'\nimport { WebSocketMessageListener } from './WebSocketOverride'\nimport { bindEvent } from './utils/bindEvent'\nimport { CloseEvent } from './utils/events'\nimport { uuidv4 } from '../../utils/uuid'\n\nconst kEmitter = Symbol('kEmitter')\n\nexport interface WebSocketClientConnectionProtocol {\n id: string\n url: URL\n send(data: WebSocketData): void\n close(code?: number, reason?: string): void\n}\n\n/**\n * The WebSocket client instance represents an incoming\n * client connection. The user can control the connection,\n * send and receive events.\n */\nexport class WebSocketClientConnection\n implements WebSocketClientConnectionProtocol\n{\n public readonly id: string\n public readonly url: URL\n\n private [kEmitter]: EventTarget\n\n constructor(\n private readonly socket: WebSocket,\n private readonly transport: WebSocketTransport\n ) {\n this.id = uuidv4()\n this.url = new URL(socket.url)\n this[kEmitter] = new EventTarget()\n\n // Emit outgoing client data (\"ws.send()\") as \"message\"\n // events on the client connection.\n this.transport.onOutgoing = (data) => {\n this[kEmitter].dispatchEvent(\n bindEvent(this.socket, new MessageEvent('message', { data }))\n )\n }\n\n this.transport.onClose = (event) => {\n this[kEmitter].dispatchEvent(\n bindEvent(this.socket, new CloseEvent('close', event))\n )\n }\n }\n\n /**\n * Listen for the outgoing events from the connected WebSocket client.\n */\n public addEventListener(\n event: string,\n listener: WebSocketMessageListener,\n options?: AddEventListenerOptions | boolean\n ): void {\n this[kEmitter].addEventListener(event, listener as EventListener, options)\n }\n\n /**\n * Removes the listener for the given event.\n */\n public removeEventListener(\n event: string,\n listener: WebSocketMessageListener,\n options?: EventListenerOptions | boolean\n ): void {\n this[kEmitter].removeEventListener(\n event,\n listener as EventListener,\n options\n )\n }\n\n /**\n * Send data to the connected client.\n */\n public send(data: WebSocketData): void {\n this.transport.send(data)\n }\n\n /**\n * Close the WebSocket connection.\n * @param {number} code A status code (see https://www.rfc-editor.org/rfc/rfc6455#section-7.4.1).\n * @param {string} reason A custom connection close reason.\n */\n public close(code?: number, reason?: string): void {\n this.transport.close(code, reason)\n }\n}\n","import { invariant } from 'outvariant'\nimport type { WebSocketOverride } from './WebSocketOverride'\nimport type { WebSocketData } from './WebSocketTransport'\nimport type { WebSocketClassTransport } from './WebSocketClassTransport'\nimport { bindEvent } from './utils/bindEvent'\nimport { CancelableMessageEvent } from './utils/events'\n\nconst kEmitter = Symbol('kEmitter')\n\n/**\n * The WebSocket server instance represents the actual production\n * WebSocket server connection. It's idle by default but you can\n * establish it by calling `server.connect()`.\n */\nexport class WebSocketServerConnection {\n /**\n * A WebSocket instance connected to the original server.\n */\n private realWebSocket?: WebSocket\n private [kEmitter]: EventTarget\n\n constructor(\n private readonly socket: WebSocketOverride,\n private readonly transport: WebSocketClassTransport,\n private readonly createConnection: () => WebSocket\n ) {\n this[kEmitter] = new EventTarget()\n\n // Handle incoming events from the actual server.\n // The (mock) WebSocket instance will call this\n // whenever a \"message\" event from the actual server\n // is dispatched on it (the dispatch will be skipped).\n this.transport.onIncoming = (event) => {\n // Clone the event to dispatch it on this class\n // once again and prevent the \"already being dispatched\"\n // exception. Clone it here so we can observe this event\n // being prevented in the \"server.on()\" listeners.\n const messageEvent = bindEvent(\n this.realWebSocket!,\n new CancelableMessageEvent('message', {\n data: event.data,\n origin: event.origin,\n cancelable: true,\n })\n )\n\n /**\n * @note Emit \"message\" event on the WebSocketClassServer\n * instance to let the interceptor know about these\n * incoming events from the original server. In that listener,\n * the interceptor can modify or skip the event forwarding\n * to the mock WebSocket instance.\n */\n this[kEmitter].dispatchEvent(messageEvent)\n\n /**\n * @note Forward the incoming server events to the client.\n * Preventing the default on the message event stops this.\n */\n if (!messageEvent.defaultPrevented) {\n this.socket.dispatchEvent(\n bindEvent(\n /**\n * @note Bind the forwarded original server events\n * to the mock WebSocket instance so it would\n * dispatch them straight away.\n */\n this.socket,\n // Clone the message event again to prevent\n // the \"already being dispatched\" exception.\n new MessageEvent('message', {\n data: event.data,\n origin: event.origin,\n })\n )\n )\n }\n }\n }\n\n /**\n * Server ready state.\n * Proxies the ready state of the original WebSocket instance,\n * if set. If the original connection hasn't been established,\n * defaults to `-1`.\n */\n public get readyState(): number {\n if (this.realWebSocket) {\n return this.realWebSocket.readyState\n }\n\n return -1\n }\n\n /**\n * Open connection to the original WebSocket server.\n */\n public connect(): void {\n invariant(\n this.readyState === -1,\n 'Failed to call \"connect()\" on the original WebSocket instance: the connection already open'\n )\n\n const ws = this.createConnection()\n\n // Close the original connection when the (mock)\n // client closes, regardless of the reason.\n this.socket.addEventListener(\n 'close',\n (event) => {\n ws.close(event.code, event.reason)\n },\n { once: true }\n )\n\n ws.addEventListener('message', (event) => {\n this.transport.onIncoming(event)\n })\n\n // Forward server errors to the WebSocket client as-is.\n // We may consider exposing them to the interceptor in the future.\n ws.addEventListener('error', () => {\n this.socket.dispatchEvent(bindEvent(this.socket, new Event('error')))\n })\n\n this.realWebSocket = ws\n }\n\n /**\n * Listen for the incoming events from the original WebSocket server.\n */\n public addEventListener<K extends keyof WebSocketEventMap>(\n event: K,\n listener: (this: WebSocket, event: WebSocketEventMap[K]) => void,\n options?: AddEventListenerOptions | boolean\n ): void {\n this[kEmitter].addEventListener(\n event,\n listener.bind(this.realWebSocket!) as EventListener,\n options\n )\n }\n\n /**\n * Removes the listener for the given event.\n */\n public removeEventListener<K extends keyof WebSocketEventMap>(\n event: K,\n listener: (this: WebSocket, event: WebSocketEventMap[K]) => void,\n options?: EventListenerOptions | boolean\n ): void {\n this[kEmitter].removeEventListener(\n event,\n listener as EventListener,\n options\n )\n }\n\n /**\n * Send data to the original WebSocket server.\n * @example\n * server.send('hello')\n * server.send(new Blob(['hello']))\n * server.send(new TextEncoder().encode('hello'))\n */\n public send(data: WebSocketData): void {\n const { realWebSocket } = this\n invariant(\n realWebSocket,\n 'Failed to call \"server.send()\" for \"%s\": the connection is not open. Did you forget to call \"await server.connect()\"?',\n this.socket.url\n )\n\n // Delegate the send to when the original connection is open.\n // Unlike the mock, connecting to the original server may take time\n // so we cannot call this on the next tick.\n if (realWebSocket.readyState === realWebSocket.CONNECTING) {\n realWebSocket.addEventListener(\n 'open',\n () => {\n realWebSocket.send(data)\n },\n { once: true }\n )\n return\n }\n\n // Send the data to the original WebSocket server.\n realWebSocket.send(data)\n }\n}\n","export type WebSocketData = string | ArrayBufferLike | Blob | ArrayBufferView\n\nexport type WebSocketTransportOnIncomingCallback = (\n event: MessageEvent<WebSocketData>\n) => void\n\nexport type WebSocketTransportOnOutgoingCallback = (data: WebSocketData) => void\n\nexport type WebSocketTransportOnCloseCallback = (event: CloseEvent) => void\n\nexport abstract class WebSocketTransport {\n /**\n * A callback for the incoming server events.\n * This is called when the WebSocket client receives\n * a message from the server.\n */\n abstract onIncoming: WebSocketTransportOnIncomingCallback\n\n /**\n * A callback for outgoing client events.\n * This is called when the WebSocket client sends data.\n */\n abstract onOutgoing: WebSocketTransportOnOutgoingCallback\n\n /**\n * A callback for the close client event.\n * This is called when the WebSocket client is closed.\n */\n abstract onClose: WebSocketTransportOnCloseCallback\n\n /**\n * Send the data from the server to this client.\n */\n abstract send(data: WebSocketData): void\n\n /**\n * Close the client connection.\n */\n abstract close(code?: number, reason?: string): void\n}\n","import { invariant } from 'outvariant'\nimport type { WebSocketData } from './WebSocketTransport'\nimport { bindEvent } from './utils/bindEvent'\nimport { CloseEvent } from './utils/events'\n\ntype WebSocketEventListener = (this: WebSocket, event: Event) => void\n\nexport type WebSocketMessageListener = (\n this: WebSocket,\n event: MessageEvent\n) => void\n\ntype WebSocketCloseListener = (this: WebSocket, event: CloseEvent) => void\n\nconst WEBSOCKET_CLOSE_CODE_RANGE_ERROR =\n 'InvalidAccessError: close code out of user configurable range'\n\nexport const kOnSend = Symbol('kOnSend')\nexport const kClose = Symbol('kClose')\n\nexport class WebSocketOverride extends EventTarget implements WebSocket {\n static readonly CONNECTING = WebSocket.CONNECTING\n static readonly OPEN = WebSocket.OPEN\n static readonly CLOSING = WebSocket.CLOSING\n static readonly CLOSED = WebSocket.CLOSED\n readonly CONNECTING = WebSocket.CONNECTING\n readonly OPEN = WebSocket.OPEN\n readonly CLOSING = WebSocket.CLOSING\n readonly CLOSED = WebSocket.CLOSED\n\n public url: string\n public protocol: string\n public extensions: string\n public binaryType: BinaryType\n public readyState: number\n public bufferedAmount: number\n\n private _onopen: WebSocketEventListener | null = null\n private _onmessage: WebSocketMessageListener | null = null\n private _onerror: WebSocketEventListener | null = null\n private _onclose: WebSocketCloseListener | null = null\n\n private [kOnSend]?: (data: WebSocketData) => void\n\n constructor(url: string | URL, protocols?: string | Array<string>) {\n super()\n this.url = url.toString()\n this.protocol = ''\n this.extensions = ''\n this.binaryType = 'blob'\n this.readyState = this.CONNECTING\n this.bufferedAmount = 0\n\n Reflect.set(this, 'readyState', this.CONNECTING)\n queueMicrotask(() => {\n Reflect.set(this, 'readyState', this.OPEN)\n this.protocol = protocols ? protocols[0] : 'ws'\n\n this.dispatchEvent(bindEvent(this, new Event('open')))\n })\n }\n\n set onopen(listener: WebSocketEventListener | null) {\n this.removeEventListener('open', this._onopen)\n this._onopen = listener\n if (listener !== null) {\n this.addEventListener('open', listener)\n }\n }\n get onopen(): WebSocketEventListener | null {\n return this._onopen\n }\n\n set onmessage(listener: WebSocketMessageListener | null) {\n this.removeEventListener(\n 'message',\n this._onmessage as WebSocketEventListener\n )\n this._onmessage = listener\n if (listener !== null) {\n this.addEventListener('message', listener)\n }\n }\n get onmessage(): WebSocketMessageListener | null {\n return this._onmessage\n }\n\n set onerror(listener: WebSocketEventListener | null) {\n this.removeEventListener('error', this._onerror)\n this._onerror = listener\n if (listener !== null) {\n this.addEventListener('error', listener)\n }\n }\n get onerror(): WebSocketEventListener | null {\n return this._onerror\n }\n\n set onclose(listener: WebSocketCloseListener | null) {\n this.removeEventListener('close', this._onclose as WebSocketEventListener)\n this._onclose = listener\n if (listener !== null) {\n this.addEventListener('close', listener)\n }\n }\n get onclose(): WebSocketCloseListener | null {\n return this._onclose\n }\n\n /**\n * @see https://websockets.spec.whatwg.org/#ref-for-dom-websocket-send%E2%91%A0\n */\n public send(data: WebSocketData): void {\n if (this.readyState === this.CONNECTING) {\n this.close()\n throw new DOMException('InvalidStateError')\n }\n\n // Sending when the socket is about to close\n // discards the sent data.\n if (this.readyState === this.CLOSING || this.readyState === this.CLOSED) {\n return\n }\n\n // Buffer the data to send in this even loop\n // but send it in the next.\n this.bufferedAmount += getDataSize(data)\n\n queueMicrotask(() => {\n // This is a bit optimistic but since no actual data transfer\n // is involved, all the data will be \"sent\" on the next tick.\n this.bufferedAmount = 0\n\n /**\n * @note Notify the parent about outgoing data.\n * This notifies the transport and the connection\n * listens to the outgoing data to emit the \"message\" event.\n */\n this[kOnSend]?.(data)\n })\n }\n\n public close(code: number = 1000, reason?: string): void {\n invariant(code, WEBSOCKET_CLOSE_CODE_RANGE_ERROR)\n invariant(\n code === 1000 || (code >= 3000 && code <= 4999),\n WEBSOCKET_CLOSE_CODE_RANGE_ERROR\n )\n\n if (this.readyState === this.CLOSING || this.readyState === this.CLOSED) {\n return\n }\n\n this[kClose](code, reason)\n }\n\n private [kClose](code: number = 1000, reason?: string): void {\n this.readyState = this.CLOSING\n\n queueMicrotask(() => {\n this.readyState = this.CLOSED\n\n // Non-user-configurable close status codes\n // represent connection termination and must\n // emit the \"error\" event before closing.\n if (code > 1000 && code <= 1015) {\n this.dispatchEvent(bindEvent(this, new Event('error')))\n }\n\n this.dispatchEvent(\n bindEvent(\n this,\n new CloseEvent('close', {\n code,\n reason,\n wasClean: code === 1000,\n })\n )\n )\n\n // Remove all event listeners once the socket is closed.\n this._onopen = null\n this._onmessage = null\n this._onerror = null\n this._onclose = null\n })\n }\n\n public addEventListener<K extends keyof WebSocketEventMap>(\n type: K,\n listener: (this: WebSocket, event: WebSocketEventMap[K]) => void,\n options?: boolean | AddEventListenerOptions\n ): void\n public addEventListener(\n type: string,\n listener: EventListenerOrEventListenerObject,\n options?: boolean | AddEventListenerOptions\n ): void\n public addEventListener(\n type: unknown,\n listener: unknown,\n options?: unknown\n ): void {\n return super.addEventListener(\n type as string,\n listener as EventListener,\n options as AddEventListenerOptions\n )\n }\n\n removeEventListener<K extends keyof WebSocketEventMap>(\n type: K,\n callback: EventListenerOrEventListenerObject | null,\n options?: boolean | EventListenerOptions\n ): void {\n return super.removeEventListener(type, callback, options)\n }\n}\n\nfunction getDataSize(data: WebSocketData): number {\n if (typeof data === 'string') {\n return data.length\n }\n\n if (data instanceof Blob) {\n return data.size\n }\n\n return data.byteLength\n}\n","import { bindEvent } from './utils/bindEvent'\nimport {\n WebSocketData,\n WebSocketTransport,\n WebSocketTransportOnCloseCallback,\n WebSocketTransportOnIncomingCallback,\n WebSocketTransportOnOutgoingCallback,\n} from './WebSocketTransport'\nimport { kOnSend, kClose, WebSocketOverride } from './WebSocketOverride'\n\n/**\n * Abstraction over the given mock `WebSocket` instance that allows\n * for controlling that instance (e.g. sending and receiving messages).\n */\nexport class WebSocketClassTransport extends WebSocketTransport {\n public onOutgoing: WebSocketTransportOnOutgoingCallback = () => {}\n public onIncoming: WebSocketTransportOnIncomingCallback = () => {}\n public onClose: WebSocketTransportOnCloseCallback = () => {}\n\n constructor(protected readonly socket: WebSocketOverride) {\n super()\n\n this.socket.addEventListener('close', (event) => this.onClose(event), {\n once: true,\n })\n this.socket[kOnSend] = (...args) => this.onOutgoing(...args)\n }\n\n public send(data: WebSocketData): void {\n queueMicrotask(() => {\n const message = bindEvent(\n /**\n * @note Setting this event's \"target\" to the\n * WebSocket override instance is important.\n * This way it can tell apart original incoming events\n * (must be forwarded to the transport) from the\n * mocked message events like the one below\n * (must be dispatched on the client instance).\n */\n this.socket,\n new MessageEvent('message', {\n data,\n origin: this.socket.url,\n })\n )\n\n this.socket.dispatchEvent(message)\n })\n }\n\n public close(code: number, reason?: string): void {\n /**\n * @note Call the internal close method directly\n * to allow closing the connection with the status codes\n * that are non-configurable by the user (> 1000 <= 1015).\n */\n this.socket[kClose](code, reason)\n }\n}\n","import { Interceptor } from '../../Interceptor'\nimport {\n type WebSocketClientConnectionProtocol,\n WebSocketClientConnection,\n} from './WebSocketClientConnection'\nimport { WebSocketServerConnection } from './WebSocketServerConnection'\nimport { WebSocketClassTransport } from './WebSocketClassTransport'\nimport { WebSocketOverride } from './WebSocketOverride'\n\nexport { type WebSocketData, WebSocketTransport } from './WebSocketTransport'\nexport {\n WebSocketClientConnection,\n WebSocketClientConnectionProtocol,\n WebSocketServerConnection,\n}\n\nexport type WebSocketEventMap = {\n connection: [\n args: {\n /**\n * The incoming WebSocket client connection.\n */\n client: WebSocketClientConnection\n\n /**\n * The original WebSocket server connection.\n */\n server: WebSocketServerConnection\n },\n ]\n}\n\n/**\n * Intercept the outgoing WebSocket connections created using\n * the global `WebSocket` class.\n */\nexport class WebSocketInterceptor extends Interceptor<WebSocketEventMap> {\n static symbol = Symbol('websocket')\n\n constructor() {\n super(WebSocketInterceptor.symbol)\n }\n\n protected checkEnvironment(): boolean {\n // Enable this interceptor in any environment\n // that has a global WebSocket API.\n return typeof globalThis.WebSocket !== 'undefined'\n }\n\n protected setup(): void {\n const webSocketProxy = Proxy.revocable(globalThis.WebSocket, {\n construct: (\n target,\n args: ConstructorParameters<typeof globalThis.WebSocket>,\n newTarget\n ) => {\n const [url, protocols] = args\n\n const createConnection = (): WebSocket => {\n return Reflect.construct(target, args, newTarget)\n }\n\n // All WebSocket instances are mocked and don't forward\n // any events to the original server (no connection established).\n // To forward the events, the user must use the \"server.send()\" API.\n const socket = new WebSocketOverride(url, protocols)\n const transport = new WebSocketClassTransport(socket)\n\n // The \"globalThis.WebSocket\" class stands for\n // the client-side connection. Assume it's established\n // as soon as the WebSocket instance is constructed.\n this.emitter.emit('connection', {\n client: new WebSocketClientConnection(socket, transport),\n server: new WebSocketServerConnection(\n socket,\n transport,\n createConnection\n ),\n })\n\n return socket\n },\n })\n\n globalThis.WebSocket = webSocketProxy.proxy\n\n this.subscriptions.push(() => {\n webSocketProxy.revoke()\n })\n }\n}\n"]}
|
|
@@ -49,20 +49,20 @@ var CloseEvent = class extends Event {
|
|
|
49
49
|
// src/interceptors/WebSocket/WebSocketClientConnection.ts
|
|
50
50
|
var kEmitter = Symbol("kEmitter");
|
|
51
51
|
var WebSocketClientConnection = class {
|
|
52
|
-
constructor(
|
|
53
|
-
this.
|
|
52
|
+
constructor(socket, transport) {
|
|
53
|
+
this.socket = socket;
|
|
54
54
|
this.transport = transport;
|
|
55
55
|
this.id = uuidv4();
|
|
56
|
-
this.url = new URL(
|
|
56
|
+
this.url = new URL(socket.url);
|
|
57
57
|
this[kEmitter] = new EventTarget();
|
|
58
58
|
this.transport.onOutgoing = (data) => {
|
|
59
59
|
this[kEmitter].dispatchEvent(
|
|
60
|
-
bindEvent(this.
|
|
60
|
+
bindEvent(this.socket, new MessageEvent("message", { data }))
|
|
61
61
|
);
|
|
62
62
|
};
|
|
63
63
|
this.transport.onClose = (event) => {
|
|
64
64
|
this[kEmitter].dispatchEvent(
|
|
65
|
-
bindEvent(this.
|
|
65
|
+
bindEvent(this.socket, new CloseEvent("close", event))
|
|
66
66
|
);
|
|
67
67
|
};
|
|
68
68
|
}
|
|
@@ -103,10 +103,10 @@ kEmitter;
|
|
|
103
103
|
import { invariant } from "outvariant";
|
|
104
104
|
var kEmitter2 = Symbol("kEmitter");
|
|
105
105
|
var WebSocketServerConnection = class {
|
|
106
|
-
constructor(
|
|
107
|
-
this.
|
|
108
|
-
this.createConnection = createConnection;
|
|
106
|
+
constructor(socket, transport, createConnection) {
|
|
107
|
+
this.socket = socket;
|
|
109
108
|
this.transport = transport;
|
|
109
|
+
this.createConnection = createConnection;
|
|
110
110
|
this[kEmitter2] = new EventTarget();
|
|
111
111
|
this.transport.onIncoming = (event) => {
|
|
112
112
|
const messageEvent = bindEvent(
|
|
@@ -119,14 +119,14 @@ var WebSocketServerConnection = class {
|
|
|
119
119
|
);
|
|
120
120
|
this[kEmitter2].dispatchEvent(messageEvent);
|
|
121
121
|
if (!messageEvent.defaultPrevented) {
|
|
122
|
-
this.
|
|
122
|
+
this.socket.dispatchEvent(
|
|
123
123
|
bindEvent(
|
|
124
124
|
/**
|
|
125
125
|
* @note Bind the forwarded original server events
|
|
126
126
|
* to the mock WebSocket instance so it would
|
|
127
127
|
* dispatch them straight away.
|
|
128
128
|
*/
|
|
129
|
-
this.
|
|
129
|
+
this.socket,
|
|
130
130
|
// Clone the message event again to prevent
|
|
131
131
|
// the "already being dispatched" exception.
|
|
132
132
|
new MessageEvent("message", {
|
|
@@ -159,7 +159,7 @@ var WebSocketServerConnection = class {
|
|
|
159
159
|
'Failed to call "connect()" on the original WebSocket instance: the connection already open'
|
|
160
160
|
);
|
|
161
161
|
const ws = this.createConnection();
|
|
162
|
-
this.
|
|
162
|
+
this.socket.addEventListener(
|
|
163
163
|
"close",
|
|
164
164
|
(event) => {
|
|
165
165
|
ws.close(event.code, event.reason);
|
|
@@ -170,9 +170,7 @@ var WebSocketServerConnection = class {
|
|
|
170
170
|
this.transport.onIncoming(event);
|
|
171
171
|
});
|
|
172
172
|
ws.addEventListener("error", () => {
|
|
173
|
-
this.
|
|
174
|
-
bindEvent(this.mockWebSocket, new Event("error"))
|
|
175
|
-
);
|
|
173
|
+
this.socket.dispatchEvent(bindEvent(this.socket, new Event("error")));
|
|
176
174
|
});
|
|
177
175
|
this.realWebSocket = ws;
|
|
178
176
|
}
|
|
@@ -208,7 +206,7 @@ var WebSocketServerConnection = class {
|
|
|
208
206
|
invariant(
|
|
209
207
|
realWebSocket,
|
|
210
208
|
'Failed to call "server.send()" for "%s": the connection is not open. Did you forget to call "await server.connect()"?',
|
|
211
|
-
this.
|
|
209
|
+
this.socket.url
|
|
212
210
|
);
|
|
213
211
|
if (realWebSocket.readyState === realWebSocket.CONNECTING) {
|
|
214
212
|
realWebSocket.addEventListener(
|
|
@@ -380,19 +378,19 @@ function getDataSize(data) {
|
|
|
380
378
|
|
|
381
379
|
// src/interceptors/WebSocket/WebSocketClassTransport.ts
|
|
382
380
|
var WebSocketClassTransport = class extends WebSocketTransport {
|
|
383
|
-
constructor(
|
|
381
|
+
constructor(socket) {
|
|
384
382
|
super();
|
|
385
|
-
this.
|
|
383
|
+
this.socket = socket;
|
|
386
384
|
this.onOutgoing = () => {
|
|
387
385
|
};
|
|
388
386
|
this.onIncoming = () => {
|
|
389
387
|
};
|
|
390
388
|
this.onClose = () => {
|
|
391
389
|
};
|
|
392
|
-
this.
|
|
390
|
+
this.socket.addEventListener("close", (event) => this.onClose(event), {
|
|
393
391
|
once: true
|
|
394
392
|
});
|
|
395
|
-
this.
|
|
393
|
+
this.socket[kOnSend] = (...args) => this.onOutgoing(...args);
|
|
396
394
|
}
|
|
397
395
|
send(data) {
|
|
398
396
|
queueMicrotask(() => {
|
|
@@ -405,17 +403,17 @@ var WebSocketClassTransport = class extends WebSocketTransport {
|
|
|
405
403
|
* mocked message events like the one below
|
|
406
404
|
* (must be dispatched on the client instance).
|
|
407
405
|
*/
|
|
408
|
-
this.
|
|
406
|
+
this.socket,
|
|
409
407
|
new MessageEvent("message", {
|
|
410
408
|
data,
|
|
411
|
-
origin: this.
|
|
409
|
+
origin: this.socket.url
|
|
412
410
|
})
|
|
413
411
|
);
|
|
414
|
-
this.
|
|
412
|
+
this.socket.dispatchEvent(message);
|
|
415
413
|
});
|
|
416
414
|
}
|
|
417
415
|
close(code, reason) {
|
|
418
|
-
this.
|
|
416
|
+
this.socket[kClose](code, reason);
|
|
419
417
|
}
|
|
420
418
|
};
|
|
421
419
|
|
|
@@ -434,17 +432,17 @@ var _WebSocketInterceptor = class extends Interceptor {
|
|
|
434
432
|
const createConnection = () => {
|
|
435
433
|
return Reflect.construct(target, args, newTarget);
|
|
436
434
|
};
|
|
437
|
-
const
|
|
438
|
-
const transport = new WebSocketClassTransport(
|
|
435
|
+
const socket = new WebSocketOverride(url, protocols);
|
|
436
|
+
const transport = new WebSocketClassTransport(socket);
|
|
439
437
|
this.emitter.emit("connection", {
|
|
440
|
-
client: new WebSocketClientConnection(
|
|
438
|
+
client: new WebSocketClientConnection(socket, transport),
|
|
441
439
|
server: new WebSocketServerConnection(
|
|
442
|
-
|
|
443
|
-
|
|
444
|
-
|
|
440
|
+
socket,
|
|
441
|
+
transport,
|
|
442
|
+
createConnection
|
|
445
443
|
)
|
|
446
444
|
});
|
|
447
|
-
return
|
|
445
|
+
return socket;
|
|
448
446
|
}
|
|
449
447
|
});
|
|
450
448
|
globalThis.WebSocket = webSocketProxy.proxy;
|
|
@@ -458,6 +456,7 @@ WebSocketInterceptor.symbol = Symbol("websocket");
|
|
|
458
456
|
export {
|
|
459
457
|
WebSocketClientConnection,
|
|
460
458
|
WebSocketInterceptor,
|
|
461
|
-
WebSocketServerConnection
|
|
459
|
+
WebSocketServerConnection,
|
|
460
|
+
WebSocketTransport
|
|
462
461
|
};
|
|
463
462
|
//# sourceMappingURL=index.mjs.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../../../../src/interceptors/WebSocket/utils/bindEvent.ts","../../../../src/interceptors/WebSocket/utils/events.ts","../../../../src/interceptors/WebSocket/WebSocketClientConnection.ts","../../../../src/interceptors/WebSocket/WebSocketServerConnection.ts","../../../../src/interceptors/WebSocket/WebSocketTransport.ts","../../../../src/interceptors/WebSocket/WebSocketOverride.ts","../../../../src/interceptors/WebSocket/WebSocketClassTransport.ts","../../../../src/interceptors/WebSocket/index.ts"],"sourcesContent":["type EventWithTarget<E extends Event, T> = E & { target: T }\n\nexport function bindEvent<E extends Event, T>(\n target: T,\n event: E\n): EventWithTarget<E, T> {\n Object.defineProperty(event, 'target', {\n enumerable: true,\n writable: true,\n value: target,\n })\n return event as EventWithTarget<E, T>\n}\n","const kCancelable = Symbol('kCancelable')\nconst kDefaultPrevented = Symbol('kDefaultPrevented')\n\n/**\n * A `MessageEvent` superset that supports event cancellation\n * in Node.js. It's rather non-intrusive so it can be safely\n * used in the browser as well.\n *\n * @see https://github.com/nodejs/node/issues/51767\n */\nexport class CancelableMessageEvent<T = any> extends MessageEvent<T> {\n [kCancelable]: boolean;\n [kDefaultPrevented]: boolean\n\n constructor(type: string, init: MessageEventInit<T>) {\n super(type, init)\n this[kCancelable] = !!init.cancelable\n this[kDefaultPrevented] = false\n }\n\n get cancelable() {\n return this[kCancelable]\n }\n\n get defaultPrevented() {\n return this[kDefaultPrevented]\n }\n\n public preventDefault(): void {\n if (this.cancelable && !this[kDefaultPrevented]) {\n this[kDefaultPrevented] = true\n }\n }\n}\n\ninterface CloseEventInit extends EventInit {\n code?: number\n reason?: string\n wasClean?: boolean\n}\n\nexport class CloseEvent extends Event {\n public code: number\n public reason: string\n public wasClean: boolean\n\n constructor(type: string, init: CloseEventInit = {}) {\n super(type, init)\n this.code = init.code === undefined ? 0 : init.code\n this.reason = init.reason === undefined ? '' : init.reason\n this.wasClean = init.wasClean === undefined ? false : init.wasClean\n }\n}\n","/**\n * WebSocket client class.\n * This represents an incoming WebSocket client connection.\n * @note Keep this class implementation-agnostic because it's\n * meant to be used over any WebSocket implementation\n * (not all of them follow the one from WHATWG).\n */\nimport type { WebSocketRawData, WebSocketTransport } from './WebSocketTransport'\nimport { WebSocketMessageListener } from './WebSocketOverride'\nimport { bindEvent } from './utils/bindEvent'\nimport { CloseEvent } from './utils/events'\nimport { uuidv4 } from '../../utils/uuid'\n\nconst kEmitter = Symbol('kEmitter')\n\n/**\n * The WebSocket client instance represents an incoming\n * client connection. The user can control the connection,\n * send and receive events.\n */\nexport class WebSocketClientConnection {\n public readonly id: string\n public readonly url: URL\n\n protected [kEmitter]: EventTarget\n\n constructor(\n protected readonly ws: WebSocket,\n protected readonly transport: WebSocketTransport\n ) {\n this.id = uuidv4()\n this.url = new URL(ws.url)\n this[kEmitter] = new EventTarget()\n\n // Emit outgoing client data (\"ws.send()\") as \"message\"\n // events on the client connection.\n this.transport.onOutgoing = (data) => {\n this[kEmitter].dispatchEvent(\n bindEvent(this.ws, new MessageEvent('message', { data }))\n )\n }\n\n this.transport.onClose = (event) => {\n this[kEmitter].dispatchEvent(\n bindEvent(this.ws, new CloseEvent('close', event))\n )\n }\n }\n\n /**\n * Listen for the outgoing events from the connected WebSocket client.\n */\n public addEventListener(\n event: string,\n listener: WebSocketMessageListener,\n options?: AddEventListenerOptions | boolean\n ): void {\n this[kEmitter].addEventListener(event, listener as EventListener, options)\n }\n\n /**\n * Removes the listener for the given event.\n */\n public removeEventListener(\n event: string,\n listener: WebSocketMessageListener,\n options?: EventListenerOptions | boolean\n ): void {\n this[kEmitter].removeEventListener(\n event,\n listener as EventListener,\n options\n )\n }\n\n /**\n * Send data to the connected client.\n */\n public send(data: WebSocketRawData): void {\n this.transport.send(data)\n }\n\n /**\n * Close the WebSocket connection.\n * @param {number} code A status code (see https://www.rfc-editor.org/rfc/rfc6455#section-7.4.1).\n * @param {string} reason A custom connection close reason.\n */\n public close(code?: number, reason?: string): void {\n this.transport.close(code, reason)\n }\n}\n","import { invariant } from 'outvariant'\nimport type { WebSocketOverride } from './WebSocketOverride'\nimport type { WebSocketRawData } from './WebSocketTransport'\nimport type { WebSocketClassTransport } from './WebSocketClassTransport'\nimport { bindEvent } from './utils/bindEvent'\nimport { CancelableMessageEvent } from './utils/events'\n\nconst kEmitter = Symbol('kEmitter')\n\n/**\n * The WebSocket server instance represents the actual production\n * WebSocket server connection. It's idle by default but you can\n * establish it by calling `server.connect()`.\n */\nexport class WebSocketServerConnection {\n /**\n * A WebSocket instance connected to the original server.\n */\n private realWebSocket?: WebSocket\n private [kEmitter]: EventTarget\n\n constructor(\n private readonly mockWebSocket: WebSocketOverride,\n private readonly createConnection: () => WebSocket,\n private readonly transport: WebSocketClassTransport\n ) {\n this[kEmitter] = new EventTarget()\n\n // Handle incoming events from the actual server.\n // The (mock) WebSocket instance will call this\n // whenever a \"message\" event from the actual server\n // is dispatched on it (the dispatch will be skipped).\n this.transport.onIncoming = (event) => {\n // Clone the event to dispatch it on this class\n // once again and prevent the \"already being dispatched\"\n // exception. Clone it here so we can observe this event\n // being prevented in the \"server.on()\" listeners.\n const messageEvent = bindEvent(\n this.realWebSocket!,\n new CancelableMessageEvent('message', {\n data: event.data,\n origin: event.origin,\n cancelable: true,\n })\n )\n\n /**\n * @note Emit \"message\" event on the WebSocketClassServer\n * instance to let the interceptor know about these\n * incoming events from the original server. In that listener,\n * the interceptor can modify or skip the event forwarding\n * to the mock WebSocket instance.\n */\n this[kEmitter].dispatchEvent(messageEvent)\n\n /**\n * @note Forward the incoming server events to the client.\n * Preventing the default on the message event stops this.\n */\n if (!messageEvent.defaultPrevented) {\n this.mockWebSocket.dispatchEvent(\n bindEvent(\n /**\n * @note Bind the forwarded original server events\n * to the mock WebSocket instance so it would\n * dispatch them straight away.\n */\n this.mockWebSocket,\n // Clone the message event again to prevent\n // the \"already being dispatched\" exception.\n new MessageEvent('message', {\n data: event.data,\n origin: event.origin,\n })\n )\n )\n }\n }\n }\n\n /**\n * Server ready state.\n * Proxies the ready state of the original WebSocket instance,\n * if set. If the original connection hasn't been established,\n * defaults to `-1`.\n */\n public get readyState(): number {\n if (this.realWebSocket) {\n return this.realWebSocket.readyState\n }\n\n return -1\n }\n\n /**\n * Open connection to the original WebSocket server.\n */\n public connect(): void {\n invariant(\n this.readyState === -1,\n 'Failed to call \"connect()\" on the original WebSocket instance: the connection already open'\n )\n\n const ws = this.createConnection()\n\n // Close the original connection when the (mock)\n // client closes, regardless of the reason.\n this.mockWebSocket.addEventListener(\n 'close',\n (event) => {\n ws.close(event.code, event.reason)\n },\n { once: true }\n )\n\n ws.addEventListener('message', (event) => {\n this.transport.onIncoming(event)\n })\n\n // Forward server errors to the WebSocket client as-is.\n // We may consider exposing them to the interceptor in the future.\n ws.addEventListener('error', () => {\n this.mockWebSocket.dispatchEvent(\n bindEvent(this.mockWebSocket, new Event('error'))\n )\n })\n\n this.realWebSocket = ws\n }\n\n /**\n * Listen for the incoming events from the original WebSocket server.\n */\n public addEventListener<K extends keyof WebSocketEventMap>(\n event: K,\n listener: (this: WebSocket, event: WebSocketEventMap[K]) => void,\n options?: AddEventListenerOptions | boolean\n ): void {\n this[kEmitter].addEventListener(\n event,\n listener.bind(this.realWebSocket!) as EventListener,\n options\n )\n }\n\n /**\n * Removes the listener for the given event.\n */\n public removeEventListener<K extends keyof WebSocketEventMap>(\n event: K,\n listener: (this: WebSocket, event: WebSocketEventMap[K]) => void,\n options?: EventListenerOptions | boolean\n ): void {\n this[kEmitter].removeEventListener(\n event,\n listener as EventListener,\n options\n )\n }\n\n /**\n * Send data to the original WebSocket server.\n * @example\n * server.send('hello')\n * server.send(new Blob(['hello']))\n * server.send(new TextEncoder().encode('hello'))\n */\n public send(data: WebSocketRawData): void {\n const { realWebSocket } = this\n invariant(\n realWebSocket,\n 'Failed to call \"server.send()\" for \"%s\": the connection is not open. Did you forget to call \"await server.connect()\"?',\n this.mockWebSocket.url\n )\n\n // Delegate the send to when the original connection is open.\n // Unlike the mock, connecting to the original server may take time\n // so we cannot call this on the next tick.\n if (realWebSocket.readyState === realWebSocket.CONNECTING) {\n realWebSocket.addEventListener(\n 'open',\n () => {\n realWebSocket.send(data)\n },\n { once: true }\n )\n return\n }\n\n // Send the data to the original WebSocket server.\n realWebSocket.send(data)\n }\n}\n","export type WebSocketRawData = string | ArrayBufferLike | Blob | ArrayBufferView\n\nexport type WebSocketTransportOnIncomingCallback = (\n event: MessageEvent<WebSocketRawData>\n) => void\n\nexport type WebSocketTransportOnOutgoingCallback = (\n data: WebSocketRawData\n) => void\n\nexport type WebSocketTransportOnCloseCallback = (event: CloseEvent) => void\n\nexport abstract class WebSocketTransport {\n /**\n * Listener for the incoming server events.\n * This is called when the client receives the\n * event from the original server connection.\n *\n * This way, we can trigger the \"message\" event\n * on the mocked connection to let the user know.\n */\n abstract onIncoming: WebSocketTransportOnIncomingCallback\n abstract onOutgoing: WebSocketTransportOnOutgoingCallback\n abstract onClose: WebSocketTransportOnCloseCallback\n\n /**\n * Send the data from the server to this client.\n */\n abstract send(data: WebSocketRawData): void\n\n /**\n * Close the client connection.\n */\n abstract close(code?: number, reason?: string): void\n}\n","import { invariant } from 'outvariant'\nimport type { WebSocketRawData } from './WebSocketTransport'\nimport { bindEvent } from './utils/bindEvent'\nimport { CloseEvent } from './utils/events'\n\ntype WebSocketEventListener = (this: WebSocket, event: Event) => void\n\nexport type WebSocketMessageListener = (\n this: WebSocket,\n event: MessageEvent\n) => void\n\ntype WebSocketCloseListener = (this: WebSocket, event: CloseEvent) => void\n\nconst WEBSOCKET_CLOSE_CODE_RANGE_ERROR =\n 'InvalidAccessError: close code out of user configurable range'\n\nexport const kOnSend = Symbol('kOnSend')\nexport const kClose = Symbol('kClose')\n\nexport class WebSocketOverride extends EventTarget implements WebSocket {\n static readonly CONNECTING = WebSocket.CONNECTING\n static readonly OPEN = WebSocket.OPEN\n static readonly CLOSING = WebSocket.CLOSING\n static readonly CLOSED = WebSocket.CLOSED\n readonly CONNECTING = WebSocket.CONNECTING\n readonly OPEN = WebSocket.OPEN\n readonly CLOSING = WebSocket.CLOSING\n readonly CLOSED = WebSocket.CLOSED\n\n public url: string\n public protocol: string\n public extensions: string\n public binaryType: BinaryType\n public readyState: number\n public bufferedAmount: number\n\n private _onopen: WebSocketEventListener | null = null\n private _onmessage: WebSocketMessageListener | null = null\n private _onerror: WebSocketEventListener | null = null\n private _onclose: WebSocketCloseListener | null = null\n\n private [kOnSend]?: (data: WebSocketRawData) => void\n\n constructor(url: string | URL, protocols?: string | Array<string>) {\n super()\n this.url = url.toString()\n this.protocol = ''\n this.extensions = ''\n this.binaryType = 'blob'\n this.readyState = this.CONNECTING\n this.bufferedAmount = 0\n\n Reflect.set(this, 'readyState', this.CONNECTING)\n queueMicrotask(() => {\n Reflect.set(this, 'readyState', this.OPEN)\n this.protocol = protocols ? protocols[0] : 'ws'\n\n this.dispatchEvent(bindEvent(this, new Event('open')))\n })\n }\n\n set onopen(listener: WebSocketEventListener | null) {\n this.removeEventListener('open', this._onopen)\n this._onopen = listener\n if (listener !== null) {\n this.addEventListener('open', listener)\n }\n }\n get onopen(): WebSocketEventListener | null {\n return this._onopen\n }\n\n set onmessage(listener: WebSocketMessageListener | null) {\n this.removeEventListener(\n 'message',\n this._onmessage as WebSocketEventListener\n )\n this._onmessage = listener\n if (listener !== null) {\n this.addEventListener('message', listener)\n }\n }\n get onmessage(): WebSocketMessageListener | null {\n return this._onmessage\n }\n\n set onerror(listener: WebSocketEventListener | null) {\n this.removeEventListener('error', this._onerror)\n this._onerror = listener\n if (listener !== null) {\n this.addEventListener('error', listener)\n }\n }\n get onerror(): WebSocketEventListener | null {\n return this._onerror\n }\n\n set onclose(listener: WebSocketCloseListener | null) {\n this.removeEventListener('close', this._onclose as WebSocketEventListener)\n this._onclose = listener\n if (listener !== null) {\n this.addEventListener('close', listener)\n }\n }\n get onclose(): WebSocketCloseListener | null {\n return this._onclose\n }\n\n /**\n * @see https://websockets.spec.whatwg.org/#ref-for-dom-websocket-send%E2%91%A0\n */\n public send(data: WebSocketRawData): void {\n if (this.readyState === this.CONNECTING) {\n this.close()\n throw new DOMException('InvalidStateError')\n }\n\n // Sending when the socket is about to close\n // discards the sent data.\n if (this.readyState === this.CLOSING || this.readyState === this.CLOSED) {\n return\n }\n\n // Buffer the data to send in this even loop\n // but send it in the next.\n this.bufferedAmount += getDataSize(data)\n\n queueMicrotask(() => {\n // This is a bit optimistic but since no actual data transfer\n // is involved, all the data will be \"sent\" on the next tick.\n this.bufferedAmount = 0\n\n /**\n * @note Notify the parent about outgoing data.\n * This notifies the transport and the connection\n * listens to the outgoing data to emit the \"message\" event.\n */\n this[kOnSend]?.(data)\n })\n }\n\n public close(code: number = 1000, reason?: string): void {\n invariant(code, WEBSOCKET_CLOSE_CODE_RANGE_ERROR)\n invariant(\n code === 1000 || (code >= 3000 && code <= 4999),\n WEBSOCKET_CLOSE_CODE_RANGE_ERROR\n )\n\n if (this.readyState === this.CLOSING || this.readyState === this.CLOSED) {\n return\n }\n\n this[kClose](code, reason)\n }\n\n private [kClose](code: number = 1000, reason?: string): void {\n this.readyState = this.CLOSING\n\n queueMicrotask(() => {\n this.readyState = this.CLOSED\n\n // Non-user-configurable close status codes\n // represent connection termination and must\n // emit the \"error\" event before closing.\n if (code > 1000 && code <= 1015) {\n this.dispatchEvent(bindEvent(this, new Event('error')))\n }\n\n this.dispatchEvent(\n bindEvent(\n this,\n new CloseEvent('close', {\n code,\n reason,\n wasClean: code === 1000,\n })\n )\n )\n\n // Remove all event listeners once the socket is closed.\n this._onopen = null\n this._onmessage = null\n this._onerror = null\n this._onclose = null\n })\n }\n\n public addEventListener<K extends keyof WebSocketEventMap>(\n type: K,\n listener: (this: WebSocket, event: WebSocketEventMap[K]) => void,\n options?: boolean | AddEventListenerOptions\n ): void\n public addEventListener(\n type: string,\n listener: EventListenerOrEventListenerObject,\n options?: boolean | AddEventListenerOptions\n ): void\n public addEventListener(\n type: unknown,\n listener: unknown,\n options?: unknown\n ): void {\n return super.addEventListener(\n type as string,\n listener as EventListener,\n options as AddEventListenerOptions\n )\n }\n\n removeEventListener<K extends keyof WebSocketEventMap>(\n type: K,\n callback: EventListenerOrEventListenerObject | null,\n options?: boolean | EventListenerOptions\n ): void {\n return super.removeEventListener(type, callback, options)\n }\n}\n\nfunction getDataSize(data: WebSocketRawData): number {\n if (typeof data === 'string') {\n return data.length\n }\n\n if (data instanceof Blob) {\n return data.size\n }\n\n return data.byteLength\n}\n","import { bindEvent } from './utils/bindEvent'\nimport {\n WebSocketRawData,\n WebSocketTransport,\n WebSocketTransportOnCloseCallback,\n WebSocketTransportOnIncomingCallback,\n WebSocketTransportOnOutgoingCallback,\n} from './WebSocketTransport'\nimport { kOnSend, kClose, WebSocketOverride } from './WebSocketOverride'\n\nexport class WebSocketClassTransport extends WebSocketTransport {\n public onOutgoing: WebSocketTransportOnOutgoingCallback = () => {}\n public onIncoming: WebSocketTransportOnIncomingCallback = () => {}\n public onClose: WebSocketTransportOnCloseCallback = () => {}\n\n constructor(protected readonly ws: WebSocketOverride) {\n super()\n\n this.ws.addEventListener('close', (event) => this.onClose(event), {\n once: true,\n })\n this.ws[kOnSend] = (...args) => this.onOutgoing(...args)\n }\n\n public send(data: WebSocketRawData): void {\n queueMicrotask(() => {\n const message = bindEvent(\n /**\n * @note Setting this event's \"target\" to the\n * WebSocket override instance is important.\n * This way it can tell apart original incoming events\n * (must be forwarded to the transport) from the\n * mocked message events like the one below\n * (must be dispatched on the client instance).\n */\n this.ws,\n new MessageEvent('message', {\n data,\n origin: this.ws.url,\n })\n )\n\n this.ws.dispatchEvent(message)\n })\n }\n\n public close(code: number, reason?: string): void {\n /**\n * @note Call the internal close method directly\n * to allow closing the connection with the status codes\n * that are non-configurable by the user (> 1000 <= 1015).\n */\n this.ws[kClose](code, reason)\n }\n}\n","import { Interceptor } from '../../Interceptor'\nimport { WebSocketClientConnection } from './WebSocketClientConnection'\nimport { WebSocketServerConnection } from './WebSocketServerConnection'\nimport { WebSocketClassTransport } from './WebSocketClassTransport'\nimport { WebSocketOverride } from './WebSocketOverride'\n\nexport type { WebSocketRawData } from './WebSocketTransport'\nexport { WebSocketClientConnection, WebSocketServerConnection }\n\nexport type WebSocketEventMap = {\n connection: [\n args: {\n /**\n * The incoming WebSocket client connection.\n */\n client: WebSocketClientConnection\n\n /**\n * The original WebSocket server connection.\n */\n server: WebSocketServerConnection\n }\n ]\n}\n\n/**\n * Intercept the outgoing WebSocket connections created using\n * the global `WebSocket` class.\n */\nexport class WebSocketInterceptor extends Interceptor<WebSocketEventMap> {\n static symbol = Symbol('websocket')\n\n constructor() {\n super(WebSocketInterceptor.symbol)\n }\n\n protected checkEnvironment(): boolean {\n // Enable this interceptor in any environment\n // that has a global WebSocket API.\n return typeof globalThis.WebSocket !== 'undefined'\n }\n\n protected setup(): void {\n const webSocketProxy = Proxy.revocable(globalThis.WebSocket, {\n construct: (\n target,\n args: ConstructorParameters<typeof globalThis.WebSocket>,\n newTarget\n ) => {\n const [url, protocols] = args\n\n const createConnection = (): WebSocket => {\n return Reflect.construct(target, args, newTarget)\n }\n\n // All WebSocket instances are mocked and don't forward\n // any events to the original server (no connection established).\n // To forward the events, the user must use the \"server.send()\" API.\n const mockWs = new WebSocketOverride(url, protocols)\n const transport = new WebSocketClassTransport(mockWs)\n\n // The \"globalThis.WebSocket\" class stands for\n // the client-side connection. Assume it's established\n // as soon as the WebSocket instance is constructed.\n this.emitter.emit('connection', {\n client: new WebSocketClientConnection(mockWs, transport),\n server: new WebSocketServerConnection(\n mockWs,\n createConnection,\n transport\n ),\n })\n\n return mockWs\n },\n })\n\n globalThis.WebSocket = webSocketProxy.proxy\n\n this.subscriptions.push(() => {\n webSocketProxy.revoke()\n })\n }\n}\n"],"mappings":";;;;;;;;AAEO,SAAS,UACd,QACA,OACuB;AACvB,SAAO,eAAe,OAAO,UAAU;AAAA,IACrC,YAAY;AAAA,IACZ,UAAU;AAAA,IACV,OAAO;AAAA,EACT,CAAC;AACD,SAAO;AACT;;;ACZA,IAAM,cAAc,OAAO,aAAa;AACxC,IAAM,oBAAoB,OAAO,mBAAmB;AAS7C,IAAM,yBAAN,cAA8C,aAAgB;AAAA,EAInE,YAAY,MAAc,MAA2B;AACnD,UAAM,MAAM,IAAI;AAChB,SAAK,WAAW,IAAI,CAAC,CAAC,KAAK;AAC3B,SAAK,iBAAiB,IAAI;AAAA,EAC5B;AAAA,EAEA,IAAI,aAAa;AACf,WAAO,KAAK,WAAW;AAAA,EACzB;AAAA,EAEA,IAAI,mBAAmB;AACrB,WAAO,KAAK,iBAAiB;AAAA,EAC/B;AAAA,EAEO,iBAAuB;AAC5B,QAAI,KAAK,cAAc,CAAC,KAAK,iBAAiB,GAAG;AAC/C,WAAK,iBAAiB,IAAI;AAAA,IAC5B;AAAA,EACF;AACF;AAtBG,aACA;AA6BI,IAAM,aAAN,cAAyB,MAAM;AAAA,EAKpC,YAAY,MAAc,OAAuB,CAAC,GAAG;AACnD,UAAM,MAAM,IAAI;AAChB,SAAK,OAAO,KAAK,SAAS,SAAY,IAAI,KAAK;AAC/C,SAAK,SAAS,KAAK,WAAW,SAAY,KAAK,KAAK;AACpD,SAAK,WAAW,KAAK,aAAa,SAAY,QAAQ,KAAK;AAAA,EAC7D;AACF;;;ACvCA,IAAM,WAAW,OAAO,UAAU;AAO3B,IAAM,4BAAN,MAAgC;AAAA,EAMrC,YACqB,IACA,WACnB;AAFmB;AACA;AAEnB,SAAK,KAAK,OAAO;AACjB,SAAK,MAAM,IAAI,IAAI,GAAG,GAAG;AACzB,SAAK,QAAQ,IAAI,IAAI,YAAY;AAIjC,SAAK,UAAU,aAAa,CAAC,SAAS;AACpC,WAAK,QAAQ,EAAE;AAAA,QACb,UAAU,KAAK,IAAI,IAAI,aAAa,WAAW,EAAE,KAAK,CAAC,CAAC;AAAA,MAC1D;AAAA,IACF;AAEA,SAAK,UAAU,UAAU,CAAC,UAAU;AAClC,WAAK,QAAQ,EAAE;AAAA,QACb,UAAU,KAAK,IAAI,IAAI,WAAW,SAAS,KAAK,CAAC;AAAA,MACnD;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKO,iBACL,OACA,UACA,SACM;AACN,SAAK,QAAQ,EAAE,iBAAiB,OAAO,UAA2B,OAAO;AAAA,EAC3E;AAAA;AAAA;AAAA;AAAA,EAKO,oBACL,OACA,UACA,SACM;AACN,SAAK,QAAQ,EAAE;AAAA,MACb;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKO,KAAK,MAA8B;AACxC,SAAK,UAAU,KAAK,IAAI;AAAA,EAC1B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOO,MAAM,MAAe,QAAuB;AACjD,SAAK,UAAU,MAAM,MAAM,MAAM;AAAA,EACnC;AACF;AAlEa;;;ACxBb,SAAS,iBAAiB;AAO1B,IAAMA,YAAW,OAAO,UAAU;AAO3B,IAAM,4BAAN,MAAgC;AAAA,EAOrC,YACmB,eACA,kBACA,WACjB;AAHiB;AACA;AACA;AAEjB,SAAKA,SAAQ,IAAI,IAAI,YAAY;AAMjC,SAAK,UAAU,aAAa,CAAC,UAAU;AAKrC,YAAM,eAAe;AAAA,QACnB,KAAK;AAAA,QACL,IAAI,uBAAuB,WAAW;AAAA,UACpC,MAAM,MAAM;AAAA,UACZ,QAAQ,MAAM;AAAA,UACd,YAAY;AAAA,QACd,CAAC;AAAA,MACH;AASA,WAAKA,SAAQ,EAAE,cAAc,YAAY;AAMzC,UAAI,CAAC,aAAa,kBAAkB;AAClC,aAAK,cAAc;AAAA,UACjB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,YAME,KAAK;AAAA;AAAA;AAAA,YAGL,IAAI,aAAa,WAAW;AAAA,cAC1B,MAAM,MAAM;AAAA,cACZ,QAAQ,MAAM;AAAA,YAChB,CAAC;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,IAAW,aAAqB;AAC9B,QAAI,KAAK,eAAe;AACtB,aAAO,KAAK,cAAc;AAAA,IAC5B;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKO,UAAgB;AACrB;AAAA,MACE,KAAK,eAAe;AAAA,MACpB;AAAA,IACF;AAEA,UAAM,KAAK,KAAK,iBAAiB;AAIjC,SAAK,cAAc;AAAA,MACjB;AAAA,MACA,CAAC,UAAU;AACT,WAAG,MAAM,MAAM,MAAM,MAAM,MAAM;AAAA,MACnC;AAAA,MACA,EAAE,MAAM,KAAK;AAAA,IACf;AAEA,OAAG,iBAAiB,WAAW,CAAC,UAAU;AACxC,WAAK,UAAU,WAAW,KAAK;AAAA,IACjC,CAAC;AAID,OAAG,iBAAiB,SAAS,MAAM;AACjC,WAAK,cAAc;AAAA,QACjB,UAAU,KAAK,eAAe,IAAI,MAAM,OAAO,CAAC;AAAA,MAClD;AAAA,IACF,CAAC;AAED,SAAK,gBAAgB;AAAA,EACvB;AAAA;AAAA;AAAA;AAAA,EAKO,iBACL,OACA,UACA,SACM;AACN,SAAKA,SAAQ,EAAE;AAAA,MACb;AAAA,MACA,SAAS,KAAK,KAAK,aAAc;AAAA,MACjC;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKO,oBACL,OACA,UACA,SACM;AACN,SAAKA,SAAQ,EAAE;AAAA,MACb;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASO,KAAK,MAA8B;AACxC,UAAM,EAAE,cAAc,IAAI;AAC1B;AAAA,MACE;AAAA,MACA;AAAA,MACA,KAAK,cAAc;AAAA,IACrB;AAKA,QAAI,cAAc,eAAe,cAAc,YAAY;AACzD,oBAAc;AAAA,QACZ;AAAA,QACA,MAAM;AACJ,wBAAc,KAAK,IAAI;AAAA,QACzB;AAAA,QACA,EAAE,MAAM,KAAK;AAAA,MACf;AACA;AAAA,IACF;AAGA,kBAAc,KAAK,IAAI;AAAA,EACzB;AACF;AA7KWA;;;ACPJ,IAAe,qBAAf,MAAkC;AAsBzC;;;AClCA,SAAS,aAAAC,kBAAiB;AAc1B,IAAM,mCACJ;AAEK,IAAM,UAAU,OAAO,SAAS;AAChC,IAAM,SAAS,OAAO,QAAQ;AAE9B,IAAM,oBAAN,cAAgC,YAAiC;AAAA,EAwBtE,YAAY,KAAmB,WAAoC;AACjE,UAAM;AApBR,SAAS,aAAa,UAAU;AAChC,SAAS,OAAO,UAAU;AAC1B,SAAS,UAAU,UAAU;AAC7B,SAAS,SAAS,UAAU;AAS5B,SAAQ,UAAyC;AACjD,SAAQ,aAA8C;AACtD,SAAQ,WAA0C;AAClD,SAAQ,WAA0C;AAMhD,SAAK,MAAM,IAAI,SAAS;AACxB,SAAK,WAAW;AAChB,SAAK,aAAa;AAClB,SAAK,aAAa;AAClB,SAAK,aAAa,KAAK;AACvB,SAAK,iBAAiB;AAEtB,YAAQ,IAAI,MAAM,cAAc,KAAK,UAAU;AAC/C,mBAAe,MAAM;AACnB,cAAQ,IAAI,MAAM,cAAc,KAAK,IAAI;AACzC,WAAK,WAAW,YAAY,UAAU,CAAC,IAAI;AAE3C,WAAK,cAAc,UAAU,MAAM,IAAI,MAAM,MAAM,CAAC,CAAC;AAAA,IACvD,CAAC;AAAA,EACH;AAAA,EAEA,IAAI,OAAO,UAAyC;AAClD,SAAK,oBAAoB,QAAQ,KAAK,OAAO;AAC7C,SAAK,UAAU;AACf,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,QAAQ,QAAQ;AAAA,IACxC;AAAA,EACF;AAAA,EACA,IAAI,SAAwC;AAC1C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,UAAU,UAA2C;AACvD,SAAK;AAAA,MACH;AAAA,MACA,KAAK;AAAA,IACP;AACA,SAAK,aAAa;AAClB,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,WAAW,QAAQ;AAAA,IAC3C;AAAA,EACF;AAAA,EACA,IAAI,YAA6C;AAC/C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,QAAQ,UAAyC;AACnD,SAAK,oBAAoB,SAAS,KAAK,QAAQ;AAC/C,SAAK,WAAW;AAChB,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,SAAS,QAAQ;AAAA,IACzC;AAAA,EACF;AAAA,EACA,IAAI,UAAyC;AAC3C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,QAAQ,UAAyC;AACnD,SAAK,oBAAoB,SAAS,KAAK,QAAkC;AACzE,SAAK,WAAW;AAChB,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,SAAS,QAAQ;AAAA,IACzC;AAAA,EACF;AAAA,EACA,IAAI,UAAyC;AAC3C,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKO,KAAK,MAA8B;AACxC,QAAI,KAAK,eAAe,KAAK,YAAY;AACvC,WAAK,MAAM;AACX,YAAM,IAAI,aAAa,mBAAmB;AAAA,IAC5C;AAIA,QAAI,KAAK,eAAe,KAAK,WAAW,KAAK,eAAe,KAAK,QAAQ;AACvE;AAAA,IACF;AAIA,SAAK,kBAAkB,YAAY,IAAI;AAEvC,mBAAe,MAAM;AAhIzB;AAmIM,WAAK,iBAAiB;AAOtB,iBAAK,aAAL,8BAAgB;AAAA,IAClB,CAAC;AAAA,EACH;AAAA,EAEO,MAAM,OAAe,KAAM,QAAuB;AACvD,IAAAC,WAAU,MAAM,gCAAgC;AAChD,IAAAA;AAAA,MACE,SAAS,OAAS,QAAQ,OAAQ,QAAQ;AAAA,MAC1C;AAAA,IACF;AAEA,QAAI,KAAK,eAAe,KAAK,WAAW,KAAK,eAAe,KAAK,QAAQ;AACvE;AAAA,IACF;AAEA,SAAK,MAAM,EAAE,MAAM,MAAM;AAAA,EAC3B;AAAA,EAEA,EAlHS,SAkHA,OAAM,EAAE,OAAe,KAAM,QAAuB;AAC3D,SAAK,aAAa,KAAK;AAEvB,mBAAe,MAAM;AACnB,WAAK,aAAa,KAAK;AAKvB,UAAI,OAAO,OAAQ,QAAQ,MAAM;AAC/B,aAAK,cAAc,UAAU,MAAM,IAAI,MAAM,OAAO,CAAC,CAAC;AAAA,MACxD;AAEA,WAAK;AAAA,QACH;AAAA,UACE;AAAA,UACA,IAAI,WAAW,SAAS;AAAA,YACtB;AAAA,YACA;AAAA,YACA,UAAU,SAAS;AAAA,UACrB,CAAC;AAAA,QACH;AAAA,MACF;AAGA,WAAK,UAAU;AACf,WAAK,aAAa;AAClB,WAAK,WAAW;AAChB,WAAK,WAAW;AAAA,IAClB,CAAC;AAAA,EACH;AAAA,EAYO,iBACL,MACA,UACA,SACM;AACN,WAAO,MAAM;AAAA,MACX;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,oBACE,MACA,UACA,SACM;AACN,WAAO,MAAM,oBAAoB,MAAM,UAAU,OAAO;AAAA,EAC1D;AACF;AArMa,kBACK,aAAa,UAAU;AAD5B,kBAEK,OAAO,UAAU;AAFtB,kBAGK,UAAU,UAAU;AAHzB,kBAIK,SAAS,UAAU;AAmMrC,SAAS,YAAY,MAAgC;AACnD,MAAI,OAAO,SAAS,UAAU;AAC5B,WAAO,KAAK;AAAA,EACd;AAEA,MAAI,gBAAgB,MAAM;AACxB,WAAO,KAAK;AAAA,EACd;AAEA,SAAO,KAAK;AACd;;;AC3NO,IAAM,0BAAN,cAAsC,mBAAmB;AAAA,EAK9D,YAA+B,IAAuB;AACpD,UAAM;AADuB;AAJ/B,SAAO,aAAmD,MAAM;AAAA,IAAC;AACjE,SAAO,aAAmD,MAAM;AAAA,IAAC;AACjE,SAAO,UAA6C,MAAM;AAAA,IAAC;AAKzD,SAAK,GAAG,iBAAiB,SAAS,CAAC,UAAU,KAAK,QAAQ,KAAK,GAAG;AAAA,MAChE,MAAM;AAAA,IACR,CAAC;AACD,SAAK,GAAG,OAAO,IAAI,IAAI,SAAS,KAAK,WAAW,GAAG,IAAI;AAAA,EACzD;AAAA,EAEO,KAAK,MAA8B;AACxC,mBAAe,MAAM;AACnB,YAAM,UAAU;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,QASd,KAAK;AAAA,QACL,IAAI,aAAa,WAAW;AAAA,UAC1B;AAAA,UACA,QAAQ,KAAK,GAAG;AAAA,QAClB,CAAC;AAAA,MACH;AAEA,WAAK,GAAG,cAAc,OAAO;AAAA,IAC/B,CAAC;AAAA,EACH;AAAA,EAEO,MAAM,MAAc,QAAuB;AAMhD,SAAK,GAAG,MAAM,EAAE,MAAM,MAAM;AAAA,EAC9B;AACF;;;ACzBO,IAAM,wBAAN,cAAmC,YAA+B;AAAA,EAGvE,cAAc;AACZ,UAAM,sBAAqB,MAAM;AAAA,EACnC;AAAA,EAEU,mBAA4B;AAGpC,WAAO,OAAO,WAAW,cAAc;AAAA,EACzC;AAAA,EAEU,QAAc;AACtB,UAAM,iBAAiB,MAAM,UAAU,WAAW,WAAW;AAAA,MAC3D,WAAW,CACT,QACA,MACA,cACG;AACH,cAAM,CAAC,KAAK,SAAS,IAAI;AAEzB,cAAM,mBAAmB,MAAiB;AACxC,iBAAO,QAAQ,UAAU,QAAQ,MAAM,SAAS;AAAA,QAClD;AAKA,cAAM,SAAS,IAAI,kBAAkB,KAAK,SAAS;AACnD,cAAM,YAAY,IAAI,wBAAwB,MAAM;AAKpD,aAAK,QAAQ,KAAK,cAAc;AAAA,UAC9B,QAAQ,IAAI,0BAA0B,QAAQ,SAAS;AAAA,UACvD,QAAQ,IAAI;AAAA,YACV;AAAA,YACA;AAAA,YACA;AAAA,UACF;AAAA,QACF,CAAC;AAED,eAAO;AAAA,MACT;AAAA,IACF,CAAC;AAED,eAAW,YAAY,eAAe;AAEtC,SAAK,cAAc,KAAK,MAAM;AAC5B,qBAAe,OAAO;AAAA,IACxB,CAAC;AAAA,EACH;AACF;AAtDO,IAAM,uBAAN;AAAM,qBACJ,SAAS,OAAO,WAAW;","names":["kEmitter","invariant","invariant"]}
|
|
1
|
+
{"version":3,"sources":["../../../../src/interceptors/WebSocket/utils/bindEvent.ts","../../../../src/interceptors/WebSocket/utils/events.ts","../../../../src/interceptors/WebSocket/WebSocketClientConnection.ts","../../../../src/interceptors/WebSocket/WebSocketServerConnection.ts","../../../../src/interceptors/WebSocket/WebSocketTransport.ts","../../../../src/interceptors/WebSocket/WebSocketOverride.ts","../../../../src/interceptors/WebSocket/WebSocketClassTransport.ts","../../../../src/interceptors/WebSocket/index.ts"],"sourcesContent":["type EventWithTarget<E extends Event, T> = E & { target: T }\n\nexport function bindEvent<E extends Event, T>(\n target: T,\n event: E\n): EventWithTarget<E, T> {\n Object.defineProperty(event, 'target', {\n enumerable: true,\n writable: true,\n value: target,\n })\n return event as EventWithTarget<E, T>\n}\n","const kCancelable = Symbol('kCancelable')\nconst kDefaultPrevented = Symbol('kDefaultPrevented')\n\n/**\n * A `MessageEvent` superset that supports event cancellation\n * in Node.js. It's rather non-intrusive so it can be safely\n * used in the browser as well.\n *\n * @see https://github.com/nodejs/node/issues/51767\n */\nexport class CancelableMessageEvent<T = any> extends MessageEvent<T> {\n [kCancelable]: boolean;\n [kDefaultPrevented]: boolean\n\n constructor(type: string, init: MessageEventInit<T>) {\n super(type, init)\n this[kCancelable] = !!init.cancelable\n this[kDefaultPrevented] = false\n }\n\n get cancelable() {\n return this[kCancelable]\n }\n\n get defaultPrevented() {\n return this[kDefaultPrevented]\n }\n\n public preventDefault(): void {\n if (this.cancelable && !this[kDefaultPrevented]) {\n this[kDefaultPrevented] = true\n }\n }\n}\n\ninterface CloseEventInit extends EventInit {\n code?: number\n reason?: string\n wasClean?: boolean\n}\n\nexport class CloseEvent extends Event {\n public code: number\n public reason: string\n public wasClean: boolean\n\n constructor(type: string, init: CloseEventInit = {}) {\n super(type, init)\n this.code = init.code === undefined ? 0 : init.code\n this.reason = init.reason === undefined ? '' : init.reason\n this.wasClean = init.wasClean === undefined ? false : init.wasClean\n }\n}\n","/**\n * WebSocket client class.\n * This represents an incoming WebSocket client connection.\n * @note Keep this class implementation-agnostic because it's\n * meant to be used over any WebSocket implementation\n * (not all of them follow the one from WHATWG).\n */\nimport type { WebSocketData, WebSocketTransport } from './WebSocketTransport'\nimport { WebSocketMessageListener } from './WebSocketOverride'\nimport { bindEvent } from './utils/bindEvent'\nimport { CloseEvent } from './utils/events'\nimport { uuidv4 } from '../../utils/uuid'\n\nconst kEmitter = Symbol('kEmitter')\n\nexport interface WebSocketClientConnectionProtocol {\n id: string\n url: URL\n send(data: WebSocketData): void\n close(code?: number, reason?: string): void\n}\n\n/**\n * The WebSocket client instance represents an incoming\n * client connection. The user can control the connection,\n * send and receive events.\n */\nexport class WebSocketClientConnection\n implements WebSocketClientConnectionProtocol\n{\n public readonly id: string\n public readonly url: URL\n\n private [kEmitter]: EventTarget\n\n constructor(\n private readonly socket: WebSocket,\n private readonly transport: WebSocketTransport\n ) {\n this.id = uuidv4()\n this.url = new URL(socket.url)\n this[kEmitter] = new EventTarget()\n\n // Emit outgoing client data (\"ws.send()\") as \"message\"\n // events on the client connection.\n this.transport.onOutgoing = (data) => {\n this[kEmitter].dispatchEvent(\n bindEvent(this.socket, new MessageEvent('message', { data }))\n )\n }\n\n this.transport.onClose = (event) => {\n this[kEmitter].dispatchEvent(\n bindEvent(this.socket, new CloseEvent('close', event))\n )\n }\n }\n\n /**\n * Listen for the outgoing events from the connected WebSocket client.\n */\n public addEventListener(\n event: string,\n listener: WebSocketMessageListener,\n options?: AddEventListenerOptions | boolean\n ): void {\n this[kEmitter].addEventListener(event, listener as EventListener, options)\n }\n\n /**\n * Removes the listener for the given event.\n */\n public removeEventListener(\n event: string,\n listener: WebSocketMessageListener,\n options?: EventListenerOptions | boolean\n ): void {\n this[kEmitter].removeEventListener(\n event,\n listener as EventListener,\n options\n )\n }\n\n /**\n * Send data to the connected client.\n */\n public send(data: WebSocketData): void {\n this.transport.send(data)\n }\n\n /**\n * Close the WebSocket connection.\n * @param {number} code A status code (see https://www.rfc-editor.org/rfc/rfc6455#section-7.4.1).\n * @param {string} reason A custom connection close reason.\n */\n public close(code?: number, reason?: string): void {\n this.transport.close(code, reason)\n }\n}\n","import { invariant } from 'outvariant'\nimport type { WebSocketOverride } from './WebSocketOverride'\nimport type { WebSocketData } from './WebSocketTransport'\nimport type { WebSocketClassTransport } from './WebSocketClassTransport'\nimport { bindEvent } from './utils/bindEvent'\nimport { CancelableMessageEvent } from './utils/events'\n\nconst kEmitter = Symbol('kEmitter')\n\n/**\n * The WebSocket server instance represents the actual production\n * WebSocket server connection. It's idle by default but you can\n * establish it by calling `server.connect()`.\n */\nexport class WebSocketServerConnection {\n /**\n * A WebSocket instance connected to the original server.\n */\n private realWebSocket?: WebSocket\n private [kEmitter]: EventTarget\n\n constructor(\n private readonly socket: WebSocketOverride,\n private readonly transport: WebSocketClassTransport,\n private readonly createConnection: () => WebSocket\n ) {\n this[kEmitter] = new EventTarget()\n\n // Handle incoming events from the actual server.\n // The (mock) WebSocket instance will call this\n // whenever a \"message\" event from the actual server\n // is dispatched on it (the dispatch will be skipped).\n this.transport.onIncoming = (event) => {\n // Clone the event to dispatch it on this class\n // once again and prevent the \"already being dispatched\"\n // exception. Clone it here so we can observe this event\n // being prevented in the \"server.on()\" listeners.\n const messageEvent = bindEvent(\n this.realWebSocket!,\n new CancelableMessageEvent('message', {\n data: event.data,\n origin: event.origin,\n cancelable: true,\n })\n )\n\n /**\n * @note Emit \"message\" event on the WebSocketClassServer\n * instance to let the interceptor know about these\n * incoming events from the original server. In that listener,\n * the interceptor can modify or skip the event forwarding\n * to the mock WebSocket instance.\n */\n this[kEmitter].dispatchEvent(messageEvent)\n\n /**\n * @note Forward the incoming server events to the client.\n * Preventing the default on the message event stops this.\n */\n if (!messageEvent.defaultPrevented) {\n this.socket.dispatchEvent(\n bindEvent(\n /**\n * @note Bind the forwarded original server events\n * to the mock WebSocket instance so it would\n * dispatch them straight away.\n */\n this.socket,\n // Clone the message event again to prevent\n // the \"already being dispatched\" exception.\n new MessageEvent('message', {\n data: event.data,\n origin: event.origin,\n })\n )\n )\n }\n }\n }\n\n /**\n * Server ready state.\n * Proxies the ready state of the original WebSocket instance,\n * if set. If the original connection hasn't been established,\n * defaults to `-1`.\n */\n public get readyState(): number {\n if (this.realWebSocket) {\n return this.realWebSocket.readyState\n }\n\n return -1\n }\n\n /**\n * Open connection to the original WebSocket server.\n */\n public connect(): void {\n invariant(\n this.readyState === -1,\n 'Failed to call \"connect()\" on the original WebSocket instance: the connection already open'\n )\n\n const ws = this.createConnection()\n\n // Close the original connection when the (mock)\n // client closes, regardless of the reason.\n this.socket.addEventListener(\n 'close',\n (event) => {\n ws.close(event.code, event.reason)\n },\n { once: true }\n )\n\n ws.addEventListener('message', (event) => {\n this.transport.onIncoming(event)\n })\n\n // Forward server errors to the WebSocket client as-is.\n // We may consider exposing them to the interceptor in the future.\n ws.addEventListener('error', () => {\n this.socket.dispatchEvent(bindEvent(this.socket, new Event('error')))\n })\n\n this.realWebSocket = ws\n }\n\n /**\n * Listen for the incoming events from the original WebSocket server.\n */\n public addEventListener<K extends keyof WebSocketEventMap>(\n event: K,\n listener: (this: WebSocket, event: WebSocketEventMap[K]) => void,\n options?: AddEventListenerOptions | boolean\n ): void {\n this[kEmitter].addEventListener(\n event,\n listener.bind(this.realWebSocket!) as EventListener,\n options\n )\n }\n\n /**\n * Removes the listener for the given event.\n */\n public removeEventListener<K extends keyof WebSocketEventMap>(\n event: K,\n listener: (this: WebSocket, event: WebSocketEventMap[K]) => void,\n options?: EventListenerOptions | boolean\n ): void {\n this[kEmitter].removeEventListener(\n event,\n listener as EventListener,\n options\n )\n }\n\n /**\n * Send data to the original WebSocket server.\n * @example\n * server.send('hello')\n * server.send(new Blob(['hello']))\n * server.send(new TextEncoder().encode('hello'))\n */\n public send(data: WebSocketData): void {\n const { realWebSocket } = this\n invariant(\n realWebSocket,\n 'Failed to call \"server.send()\" for \"%s\": the connection is not open. Did you forget to call \"await server.connect()\"?',\n this.socket.url\n )\n\n // Delegate the send to when the original connection is open.\n // Unlike the mock, connecting to the original server may take time\n // so we cannot call this on the next tick.\n if (realWebSocket.readyState === realWebSocket.CONNECTING) {\n realWebSocket.addEventListener(\n 'open',\n () => {\n realWebSocket.send(data)\n },\n { once: true }\n )\n return\n }\n\n // Send the data to the original WebSocket server.\n realWebSocket.send(data)\n }\n}\n","export type WebSocketData = string | ArrayBufferLike | Blob | ArrayBufferView\n\nexport type WebSocketTransportOnIncomingCallback = (\n event: MessageEvent<WebSocketData>\n) => void\n\nexport type WebSocketTransportOnOutgoingCallback = (data: WebSocketData) => void\n\nexport type WebSocketTransportOnCloseCallback = (event: CloseEvent) => void\n\nexport abstract class WebSocketTransport {\n /**\n * A callback for the incoming server events.\n * This is called when the WebSocket client receives\n * a message from the server.\n */\n abstract onIncoming: WebSocketTransportOnIncomingCallback\n\n /**\n * A callback for outgoing client events.\n * This is called when the WebSocket client sends data.\n */\n abstract onOutgoing: WebSocketTransportOnOutgoingCallback\n\n /**\n * A callback for the close client event.\n * This is called when the WebSocket client is closed.\n */\n abstract onClose: WebSocketTransportOnCloseCallback\n\n /**\n * Send the data from the server to this client.\n */\n abstract send(data: WebSocketData): void\n\n /**\n * Close the client connection.\n */\n abstract close(code?: number, reason?: string): void\n}\n","import { invariant } from 'outvariant'\nimport type { WebSocketData } from './WebSocketTransport'\nimport { bindEvent } from './utils/bindEvent'\nimport { CloseEvent } from './utils/events'\n\ntype WebSocketEventListener = (this: WebSocket, event: Event) => void\n\nexport type WebSocketMessageListener = (\n this: WebSocket,\n event: MessageEvent\n) => void\n\ntype WebSocketCloseListener = (this: WebSocket, event: CloseEvent) => void\n\nconst WEBSOCKET_CLOSE_CODE_RANGE_ERROR =\n 'InvalidAccessError: close code out of user configurable range'\n\nexport const kOnSend = Symbol('kOnSend')\nexport const kClose = Symbol('kClose')\n\nexport class WebSocketOverride extends EventTarget implements WebSocket {\n static readonly CONNECTING = WebSocket.CONNECTING\n static readonly OPEN = WebSocket.OPEN\n static readonly CLOSING = WebSocket.CLOSING\n static readonly CLOSED = WebSocket.CLOSED\n readonly CONNECTING = WebSocket.CONNECTING\n readonly OPEN = WebSocket.OPEN\n readonly CLOSING = WebSocket.CLOSING\n readonly CLOSED = WebSocket.CLOSED\n\n public url: string\n public protocol: string\n public extensions: string\n public binaryType: BinaryType\n public readyState: number\n public bufferedAmount: number\n\n private _onopen: WebSocketEventListener | null = null\n private _onmessage: WebSocketMessageListener | null = null\n private _onerror: WebSocketEventListener | null = null\n private _onclose: WebSocketCloseListener | null = null\n\n private [kOnSend]?: (data: WebSocketData) => void\n\n constructor(url: string | URL, protocols?: string | Array<string>) {\n super()\n this.url = url.toString()\n this.protocol = ''\n this.extensions = ''\n this.binaryType = 'blob'\n this.readyState = this.CONNECTING\n this.bufferedAmount = 0\n\n Reflect.set(this, 'readyState', this.CONNECTING)\n queueMicrotask(() => {\n Reflect.set(this, 'readyState', this.OPEN)\n this.protocol = protocols ? protocols[0] : 'ws'\n\n this.dispatchEvent(bindEvent(this, new Event('open')))\n })\n }\n\n set onopen(listener: WebSocketEventListener | null) {\n this.removeEventListener('open', this._onopen)\n this._onopen = listener\n if (listener !== null) {\n this.addEventListener('open', listener)\n }\n }\n get onopen(): WebSocketEventListener | null {\n return this._onopen\n }\n\n set onmessage(listener: WebSocketMessageListener | null) {\n this.removeEventListener(\n 'message',\n this._onmessage as WebSocketEventListener\n )\n this._onmessage = listener\n if (listener !== null) {\n this.addEventListener('message', listener)\n }\n }\n get onmessage(): WebSocketMessageListener | null {\n return this._onmessage\n }\n\n set onerror(listener: WebSocketEventListener | null) {\n this.removeEventListener('error', this._onerror)\n this._onerror = listener\n if (listener !== null) {\n this.addEventListener('error', listener)\n }\n }\n get onerror(): WebSocketEventListener | null {\n return this._onerror\n }\n\n set onclose(listener: WebSocketCloseListener | null) {\n this.removeEventListener('close', this._onclose as WebSocketEventListener)\n this._onclose = listener\n if (listener !== null) {\n this.addEventListener('close', listener)\n }\n }\n get onclose(): WebSocketCloseListener | null {\n return this._onclose\n }\n\n /**\n * @see https://websockets.spec.whatwg.org/#ref-for-dom-websocket-send%E2%91%A0\n */\n public send(data: WebSocketData): void {\n if (this.readyState === this.CONNECTING) {\n this.close()\n throw new DOMException('InvalidStateError')\n }\n\n // Sending when the socket is about to close\n // discards the sent data.\n if (this.readyState === this.CLOSING || this.readyState === this.CLOSED) {\n return\n }\n\n // Buffer the data to send in this even loop\n // but send it in the next.\n this.bufferedAmount += getDataSize(data)\n\n queueMicrotask(() => {\n // This is a bit optimistic but since no actual data transfer\n // is involved, all the data will be \"sent\" on the next tick.\n this.bufferedAmount = 0\n\n /**\n * @note Notify the parent about outgoing data.\n * This notifies the transport and the connection\n * listens to the outgoing data to emit the \"message\" event.\n */\n this[kOnSend]?.(data)\n })\n }\n\n public close(code: number = 1000, reason?: string): void {\n invariant(code, WEBSOCKET_CLOSE_CODE_RANGE_ERROR)\n invariant(\n code === 1000 || (code >= 3000 && code <= 4999),\n WEBSOCKET_CLOSE_CODE_RANGE_ERROR\n )\n\n if (this.readyState === this.CLOSING || this.readyState === this.CLOSED) {\n return\n }\n\n this[kClose](code, reason)\n }\n\n private [kClose](code: number = 1000, reason?: string): void {\n this.readyState = this.CLOSING\n\n queueMicrotask(() => {\n this.readyState = this.CLOSED\n\n // Non-user-configurable close status codes\n // represent connection termination and must\n // emit the \"error\" event before closing.\n if (code > 1000 && code <= 1015) {\n this.dispatchEvent(bindEvent(this, new Event('error')))\n }\n\n this.dispatchEvent(\n bindEvent(\n this,\n new CloseEvent('close', {\n code,\n reason,\n wasClean: code === 1000,\n })\n )\n )\n\n // Remove all event listeners once the socket is closed.\n this._onopen = null\n this._onmessage = null\n this._onerror = null\n this._onclose = null\n })\n }\n\n public addEventListener<K extends keyof WebSocketEventMap>(\n type: K,\n listener: (this: WebSocket, event: WebSocketEventMap[K]) => void,\n options?: boolean | AddEventListenerOptions\n ): void\n public addEventListener(\n type: string,\n listener: EventListenerOrEventListenerObject,\n options?: boolean | AddEventListenerOptions\n ): void\n public addEventListener(\n type: unknown,\n listener: unknown,\n options?: unknown\n ): void {\n return super.addEventListener(\n type as string,\n listener as EventListener,\n options as AddEventListenerOptions\n )\n }\n\n removeEventListener<K extends keyof WebSocketEventMap>(\n type: K,\n callback: EventListenerOrEventListenerObject | null,\n options?: boolean | EventListenerOptions\n ): void {\n return super.removeEventListener(type, callback, options)\n }\n}\n\nfunction getDataSize(data: WebSocketData): number {\n if (typeof data === 'string') {\n return data.length\n }\n\n if (data instanceof Blob) {\n return data.size\n }\n\n return data.byteLength\n}\n","import { bindEvent } from './utils/bindEvent'\nimport {\n WebSocketData,\n WebSocketTransport,\n WebSocketTransportOnCloseCallback,\n WebSocketTransportOnIncomingCallback,\n WebSocketTransportOnOutgoingCallback,\n} from './WebSocketTransport'\nimport { kOnSend, kClose, WebSocketOverride } from './WebSocketOverride'\n\n/**\n * Abstraction over the given mock `WebSocket` instance that allows\n * for controlling that instance (e.g. sending and receiving messages).\n */\nexport class WebSocketClassTransport extends WebSocketTransport {\n public onOutgoing: WebSocketTransportOnOutgoingCallback = () => {}\n public onIncoming: WebSocketTransportOnIncomingCallback = () => {}\n public onClose: WebSocketTransportOnCloseCallback = () => {}\n\n constructor(protected readonly socket: WebSocketOverride) {\n super()\n\n this.socket.addEventListener('close', (event) => this.onClose(event), {\n once: true,\n })\n this.socket[kOnSend] = (...args) => this.onOutgoing(...args)\n }\n\n public send(data: WebSocketData): void {\n queueMicrotask(() => {\n const message = bindEvent(\n /**\n * @note Setting this event's \"target\" to the\n * WebSocket override instance is important.\n * This way it can tell apart original incoming events\n * (must be forwarded to the transport) from the\n * mocked message events like the one below\n * (must be dispatched on the client instance).\n */\n this.socket,\n new MessageEvent('message', {\n data,\n origin: this.socket.url,\n })\n )\n\n this.socket.dispatchEvent(message)\n })\n }\n\n public close(code: number, reason?: string): void {\n /**\n * @note Call the internal close method directly\n * to allow closing the connection with the status codes\n * that are non-configurable by the user (> 1000 <= 1015).\n */\n this.socket[kClose](code, reason)\n }\n}\n","import { Interceptor } from '../../Interceptor'\nimport {\n type WebSocketClientConnectionProtocol,\n WebSocketClientConnection,\n} from './WebSocketClientConnection'\nimport { WebSocketServerConnection } from './WebSocketServerConnection'\nimport { WebSocketClassTransport } from './WebSocketClassTransport'\nimport { WebSocketOverride } from './WebSocketOverride'\n\nexport { type WebSocketData, WebSocketTransport } from './WebSocketTransport'\nexport {\n WebSocketClientConnection,\n WebSocketClientConnectionProtocol,\n WebSocketServerConnection,\n}\n\nexport type WebSocketEventMap = {\n connection: [\n args: {\n /**\n * The incoming WebSocket client connection.\n */\n client: WebSocketClientConnection\n\n /**\n * The original WebSocket server connection.\n */\n server: WebSocketServerConnection\n },\n ]\n}\n\n/**\n * Intercept the outgoing WebSocket connections created using\n * the global `WebSocket` class.\n */\nexport class WebSocketInterceptor extends Interceptor<WebSocketEventMap> {\n static symbol = Symbol('websocket')\n\n constructor() {\n super(WebSocketInterceptor.symbol)\n }\n\n protected checkEnvironment(): boolean {\n // Enable this interceptor in any environment\n // that has a global WebSocket API.\n return typeof globalThis.WebSocket !== 'undefined'\n }\n\n protected setup(): void {\n const webSocketProxy = Proxy.revocable(globalThis.WebSocket, {\n construct: (\n target,\n args: ConstructorParameters<typeof globalThis.WebSocket>,\n newTarget\n ) => {\n const [url, protocols] = args\n\n const createConnection = (): WebSocket => {\n return Reflect.construct(target, args, newTarget)\n }\n\n // All WebSocket instances are mocked and don't forward\n // any events to the original server (no connection established).\n // To forward the events, the user must use the \"server.send()\" API.\n const socket = new WebSocketOverride(url, protocols)\n const transport = new WebSocketClassTransport(socket)\n\n // The \"globalThis.WebSocket\" class stands for\n // the client-side connection. Assume it's established\n // as soon as the WebSocket instance is constructed.\n this.emitter.emit('connection', {\n client: new WebSocketClientConnection(socket, transport),\n server: new WebSocketServerConnection(\n socket,\n transport,\n createConnection\n ),\n })\n\n return socket\n },\n })\n\n globalThis.WebSocket = webSocketProxy.proxy\n\n this.subscriptions.push(() => {\n webSocketProxy.revoke()\n })\n }\n}\n"],"mappings":";;;;;;;;AAEO,SAAS,UACd,QACA,OACuB;AACvB,SAAO,eAAe,OAAO,UAAU;AAAA,IACrC,YAAY;AAAA,IACZ,UAAU;AAAA,IACV,OAAO;AAAA,EACT,CAAC;AACD,SAAO;AACT;;;ACZA,IAAM,cAAc,OAAO,aAAa;AACxC,IAAM,oBAAoB,OAAO,mBAAmB;AAS7C,IAAM,yBAAN,cAA8C,aAAgB;AAAA,EAInE,YAAY,MAAc,MAA2B;AACnD,UAAM,MAAM,IAAI;AAChB,SAAK,WAAW,IAAI,CAAC,CAAC,KAAK;AAC3B,SAAK,iBAAiB,IAAI;AAAA,EAC5B;AAAA,EAEA,IAAI,aAAa;AACf,WAAO,KAAK,WAAW;AAAA,EACzB;AAAA,EAEA,IAAI,mBAAmB;AACrB,WAAO,KAAK,iBAAiB;AAAA,EAC/B;AAAA,EAEO,iBAAuB;AAC5B,QAAI,KAAK,cAAc,CAAC,KAAK,iBAAiB,GAAG;AAC/C,WAAK,iBAAiB,IAAI;AAAA,IAC5B;AAAA,EACF;AACF;AAtBG,aACA;AA6BI,IAAM,aAAN,cAAyB,MAAM;AAAA,EAKpC,YAAY,MAAc,OAAuB,CAAC,GAAG;AACnD,UAAM,MAAM,IAAI;AAChB,SAAK,OAAO,KAAK,SAAS,SAAY,IAAI,KAAK;AAC/C,SAAK,SAAS,KAAK,WAAW,SAAY,KAAK,KAAK;AACpD,SAAK,WAAW,KAAK,aAAa,SAAY,QAAQ,KAAK;AAAA,EAC7D;AACF;;;ACvCA,IAAM,WAAW,OAAO,UAAU;AAc3B,IAAM,4BAAN,MAEP;AAAA,EAME,YACmB,QACA,WACjB;AAFiB;AACA;AAEjB,SAAK,KAAK,OAAO;AACjB,SAAK,MAAM,IAAI,IAAI,OAAO,GAAG;AAC7B,SAAK,QAAQ,IAAI,IAAI,YAAY;AAIjC,SAAK,UAAU,aAAa,CAAC,SAAS;AACpC,WAAK,QAAQ,EAAE;AAAA,QACb,UAAU,KAAK,QAAQ,IAAI,aAAa,WAAW,EAAE,KAAK,CAAC,CAAC;AAAA,MAC9D;AAAA,IACF;AAEA,SAAK,UAAU,UAAU,CAAC,UAAU;AAClC,WAAK,QAAQ,EAAE;AAAA,QACb,UAAU,KAAK,QAAQ,IAAI,WAAW,SAAS,KAAK,CAAC;AAAA,MACvD;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKO,iBACL,OACA,UACA,SACM;AACN,SAAK,QAAQ,EAAE,iBAAiB,OAAO,UAA2B,OAAO;AAAA,EAC3E;AAAA;AAAA;AAAA;AAAA,EAKO,oBACL,OACA,UACA,SACM;AACN,SAAK,QAAQ,EAAE;AAAA,MACb;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKO,KAAK,MAA2B;AACrC,SAAK,UAAU,KAAK,IAAI;AAAA,EAC1B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOO,MAAM,MAAe,QAAuB;AACjD,SAAK,UAAU,MAAM,MAAM,MAAM;AAAA,EACnC;AACF;AAlEW;;;ACjCX,SAAS,iBAAiB;AAO1B,IAAMA,YAAW,OAAO,UAAU;AAO3B,IAAM,4BAAN,MAAgC;AAAA,EAOrC,YACmB,QACA,WACA,kBACjB;AAHiB;AACA;AACA;AAEjB,SAAKA,SAAQ,IAAI,IAAI,YAAY;AAMjC,SAAK,UAAU,aAAa,CAAC,UAAU;AAKrC,YAAM,eAAe;AAAA,QACnB,KAAK;AAAA,QACL,IAAI,uBAAuB,WAAW;AAAA,UACpC,MAAM,MAAM;AAAA,UACZ,QAAQ,MAAM;AAAA,UACd,YAAY;AAAA,QACd,CAAC;AAAA,MACH;AASA,WAAKA,SAAQ,EAAE,cAAc,YAAY;AAMzC,UAAI,CAAC,aAAa,kBAAkB;AAClC,aAAK,OAAO;AAAA,UACV;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,YAME,KAAK;AAAA;AAAA;AAAA,YAGL,IAAI,aAAa,WAAW;AAAA,cAC1B,MAAM,MAAM;AAAA,cACZ,QAAQ,MAAM;AAAA,YAChB,CAAC;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,IAAW,aAAqB;AAC9B,QAAI,KAAK,eAAe;AACtB,aAAO,KAAK,cAAc;AAAA,IAC5B;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKO,UAAgB;AACrB;AAAA,MACE,KAAK,eAAe;AAAA,MACpB;AAAA,IACF;AAEA,UAAM,KAAK,KAAK,iBAAiB;AAIjC,SAAK,OAAO;AAAA,MACV;AAAA,MACA,CAAC,UAAU;AACT,WAAG,MAAM,MAAM,MAAM,MAAM,MAAM;AAAA,MACnC;AAAA,MACA,EAAE,MAAM,KAAK;AAAA,IACf;AAEA,OAAG,iBAAiB,WAAW,CAAC,UAAU;AACxC,WAAK,UAAU,WAAW,KAAK;AAAA,IACjC,CAAC;AAID,OAAG,iBAAiB,SAAS,MAAM;AACjC,WAAK,OAAO,cAAc,UAAU,KAAK,QAAQ,IAAI,MAAM,OAAO,CAAC,CAAC;AAAA,IACtE,CAAC;AAED,SAAK,gBAAgB;AAAA,EACvB;AAAA;AAAA;AAAA;AAAA,EAKO,iBACL,OACA,UACA,SACM;AACN,SAAKA,SAAQ,EAAE;AAAA,MACb;AAAA,MACA,SAAS,KAAK,KAAK,aAAc;AAAA,MACjC;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKO,oBACL,OACA,UACA,SACM;AACN,SAAKA,SAAQ,EAAE;AAAA,MACb;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASO,KAAK,MAA2B;AACrC,UAAM,EAAE,cAAc,IAAI;AAC1B;AAAA,MACE;AAAA,MACA;AAAA,MACA,KAAK,OAAO;AAAA,IACd;AAKA,QAAI,cAAc,eAAe,cAAc,YAAY;AACzD,oBAAc;AAAA,QACZ;AAAA,QACA,MAAM;AACJ,wBAAc,KAAK,IAAI;AAAA,QACzB;AAAA,QACA,EAAE,MAAM,KAAK;AAAA,MACf;AACA;AAAA,IACF;AAGA,kBAAc,KAAK,IAAI;AAAA,EACzB;AACF;AA3KWA;;;ACTJ,IAAe,qBAAf,MAAkC;AA6BzC;;;ACvCA,SAAS,aAAAC,kBAAiB;AAc1B,IAAM,mCACJ;AAEK,IAAM,UAAU,OAAO,SAAS;AAChC,IAAM,SAAS,OAAO,QAAQ;AAE9B,IAAM,oBAAN,cAAgC,YAAiC;AAAA,EAwBtE,YAAY,KAAmB,WAAoC;AACjE,UAAM;AApBR,SAAS,aAAa,UAAU;AAChC,SAAS,OAAO,UAAU;AAC1B,SAAS,UAAU,UAAU;AAC7B,SAAS,SAAS,UAAU;AAS5B,SAAQ,UAAyC;AACjD,SAAQ,aAA8C;AACtD,SAAQ,WAA0C;AAClD,SAAQ,WAA0C;AAMhD,SAAK,MAAM,IAAI,SAAS;AACxB,SAAK,WAAW;AAChB,SAAK,aAAa;AAClB,SAAK,aAAa;AAClB,SAAK,aAAa,KAAK;AACvB,SAAK,iBAAiB;AAEtB,YAAQ,IAAI,MAAM,cAAc,KAAK,UAAU;AAC/C,mBAAe,MAAM;AACnB,cAAQ,IAAI,MAAM,cAAc,KAAK,IAAI;AACzC,WAAK,WAAW,YAAY,UAAU,CAAC,IAAI;AAE3C,WAAK,cAAc,UAAU,MAAM,IAAI,MAAM,MAAM,CAAC,CAAC;AAAA,IACvD,CAAC;AAAA,EACH;AAAA,EAEA,IAAI,OAAO,UAAyC;AAClD,SAAK,oBAAoB,QAAQ,KAAK,OAAO;AAC7C,SAAK,UAAU;AACf,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,QAAQ,QAAQ;AAAA,IACxC;AAAA,EACF;AAAA,EACA,IAAI,SAAwC;AAC1C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,UAAU,UAA2C;AACvD,SAAK;AAAA,MACH;AAAA,MACA,KAAK;AAAA,IACP;AACA,SAAK,aAAa;AAClB,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,WAAW,QAAQ;AAAA,IAC3C;AAAA,EACF;AAAA,EACA,IAAI,YAA6C;AAC/C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,QAAQ,UAAyC;AACnD,SAAK,oBAAoB,SAAS,KAAK,QAAQ;AAC/C,SAAK,WAAW;AAChB,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,SAAS,QAAQ;AAAA,IACzC;AAAA,EACF;AAAA,EACA,IAAI,UAAyC;AAC3C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,QAAQ,UAAyC;AACnD,SAAK,oBAAoB,SAAS,KAAK,QAAkC;AACzE,SAAK,WAAW;AAChB,QAAI,aAAa,MAAM;AACrB,WAAK,iBAAiB,SAAS,QAAQ;AAAA,IACzC;AAAA,EACF;AAAA,EACA,IAAI,UAAyC;AAC3C,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKO,KAAK,MAA2B;AACrC,QAAI,KAAK,eAAe,KAAK,YAAY;AACvC,WAAK,MAAM;AACX,YAAM,IAAI,aAAa,mBAAmB;AAAA,IAC5C;AAIA,QAAI,KAAK,eAAe,KAAK,WAAW,KAAK,eAAe,KAAK,QAAQ;AACvE;AAAA,IACF;AAIA,SAAK,kBAAkB,YAAY,IAAI;AAEvC,mBAAe,MAAM;AAhIzB;AAmIM,WAAK,iBAAiB;AAOtB,iBAAK,aAAL,8BAAgB;AAAA,IAClB,CAAC;AAAA,EACH;AAAA,EAEO,MAAM,OAAe,KAAM,QAAuB;AACvD,IAAAC,WAAU,MAAM,gCAAgC;AAChD,IAAAA;AAAA,MACE,SAAS,OAAS,QAAQ,OAAQ,QAAQ;AAAA,MAC1C;AAAA,IACF;AAEA,QAAI,KAAK,eAAe,KAAK,WAAW,KAAK,eAAe,KAAK,QAAQ;AACvE;AAAA,IACF;AAEA,SAAK,MAAM,EAAE,MAAM,MAAM;AAAA,EAC3B;AAAA,EAEA,EAlHS,SAkHA,OAAM,EAAE,OAAe,KAAM,QAAuB;AAC3D,SAAK,aAAa,KAAK;AAEvB,mBAAe,MAAM;AACnB,WAAK,aAAa,KAAK;AAKvB,UAAI,OAAO,OAAQ,QAAQ,MAAM;AAC/B,aAAK,cAAc,UAAU,MAAM,IAAI,MAAM,OAAO,CAAC,CAAC;AAAA,MACxD;AAEA,WAAK;AAAA,QACH;AAAA,UACE;AAAA,UACA,IAAI,WAAW,SAAS;AAAA,YACtB;AAAA,YACA;AAAA,YACA,UAAU,SAAS;AAAA,UACrB,CAAC;AAAA,QACH;AAAA,MACF;AAGA,WAAK,UAAU;AACf,WAAK,aAAa;AAClB,WAAK,WAAW;AAChB,WAAK,WAAW;AAAA,IAClB,CAAC;AAAA,EACH;AAAA,EAYO,iBACL,MACA,UACA,SACM;AACN,WAAO,MAAM;AAAA,MACX;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,oBACE,MACA,UACA,SACM;AACN,WAAO,MAAM,oBAAoB,MAAM,UAAU,OAAO;AAAA,EAC1D;AACF;AArMa,kBACK,aAAa,UAAU;AAD5B,kBAEK,OAAO,UAAU;AAFtB,kBAGK,UAAU,UAAU;AAHzB,kBAIK,SAAS,UAAU;AAmMrC,SAAS,YAAY,MAA6B;AAChD,MAAI,OAAO,SAAS,UAAU;AAC5B,WAAO,KAAK;AAAA,EACd;AAEA,MAAI,gBAAgB,MAAM;AACxB,WAAO,KAAK;AAAA,EACd;AAEA,SAAO,KAAK;AACd;;;ACvNO,IAAM,0BAAN,cAAsC,mBAAmB;AAAA,EAK9D,YAA+B,QAA2B;AACxD,UAAM;AADuB;AAJ/B,SAAO,aAAmD,MAAM;AAAA,IAAC;AACjE,SAAO,aAAmD,MAAM;AAAA,IAAC;AACjE,SAAO,UAA6C,MAAM;AAAA,IAAC;AAKzD,SAAK,OAAO,iBAAiB,SAAS,CAAC,UAAU,KAAK,QAAQ,KAAK,GAAG;AAAA,MACpE,MAAM;AAAA,IACR,CAAC;AACD,SAAK,OAAO,OAAO,IAAI,IAAI,SAAS,KAAK,WAAW,GAAG,IAAI;AAAA,EAC7D;AAAA,EAEO,KAAK,MAA2B;AACrC,mBAAe,MAAM;AACnB,YAAM,UAAU;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,QASd,KAAK;AAAA,QACL,IAAI,aAAa,WAAW;AAAA,UAC1B;AAAA,UACA,QAAQ,KAAK,OAAO;AAAA,QACtB,CAAC;AAAA,MACH;AAEA,WAAK,OAAO,cAAc,OAAO;AAAA,IACnC,CAAC;AAAA,EACH;AAAA,EAEO,MAAM,MAAc,QAAuB;AAMhD,SAAK,OAAO,MAAM,EAAE,MAAM,MAAM;AAAA,EAClC;AACF;;;ACtBO,IAAM,wBAAN,cAAmC,YAA+B;AAAA,EAGvE,cAAc;AACZ,UAAM,sBAAqB,MAAM;AAAA,EACnC;AAAA,EAEU,mBAA4B;AAGpC,WAAO,OAAO,WAAW,cAAc;AAAA,EACzC;AAAA,EAEU,QAAc;AACtB,UAAM,iBAAiB,MAAM,UAAU,WAAW,WAAW;AAAA,MAC3D,WAAW,CACT,QACA,MACA,cACG;AACH,cAAM,CAAC,KAAK,SAAS,IAAI;AAEzB,cAAM,mBAAmB,MAAiB;AACxC,iBAAO,QAAQ,UAAU,QAAQ,MAAM,SAAS;AAAA,QAClD;AAKA,cAAM,SAAS,IAAI,kBAAkB,KAAK,SAAS;AACnD,cAAM,YAAY,IAAI,wBAAwB,MAAM;AAKpD,aAAK,QAAQ,KAAK,cAAc;AAAA,UAC9B,QAAQ,IAAI,0BAA0B,QAAQ,SAAS;AAAA,UACvD,QAAQ,IAAI;AAAA,YACV;AAAA,YACA;AAAA,YACA;AAAA,UACF;AAAA,QACF,CAAC;AAED,eAAO;AAAA,MACT;AAAA,IACF,CAAC;AAED,eAAW,YAAY,eAAe;AAEtC,SAAK,cAAc,KAAK,MAAM;AAC5B,qBAAe,OAAO;AAAA,IACxB,CAAC;AAAA,EACH;AACF;AAtDO,IAAM,uBAAN;AAAM,qBACJ,SAAS,OAAO,WAAW;","names":["kEmitter","invariant","invariant"]}
|
package/package.json
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@mswjs/interceptors",
|
|
3
3
|
"description": "Low-level HTTP/HTTPS/XHR/fetch request interception library.",
|
|
4
|
-
"version": "0.26.
|
|
4
|
+
"version": "0.26.5",
|
|
5
5
|
"main": "./lib/node/index.js",
|
|
6
6
|
"module": "./lib/node/index.mjs",
|
|
7
7
|
"types": "./lib/node/index.d.ts",
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { bindEvent } from './utils/bindEvent'
|
|
2
2
|
import {
|
|
3
|
-
|
|
3
|
+
WebSocketData,
|
|
4
4
|
WebSocketTransport,
|
|
5
5
|
WebSocketTransportOnCloseCallback,
|
|
6
6
|
WebSocketTransportOnIncomingCallback,
|
|
@@ -8,21 +8,25 @@ import {
|
|
|
8
8
|
} from './WebSocketTransport'
|
|
9
9
|
import { kOnSend, kClose, WebSocketOverride } from './WebSocketOverride'
|
|
10
10
|
|
|
11
|
+
/**
|
|
12
|
+
* Abstraction over the given mock `WebSocket` instance that allows
|
|
13
|
+
* for controlling that instance (e.g. sending and receiving messages).
|
|
14
|
+
*/
|
|
11
15
|
export class WebSocketClassTransport extends WebSocketTransport {
|
|
12
16
|
public onOutgoing: WebSocketTransportOnOutgoingCallback = () => {}
|
|
13
17
|
public onIncoming: WebSocketTransportOnIncomingCallback = () => {}
|
|
14
18
|
public onClose: WebSocketTransportOnCloseCallback = () => {}
|
|
15
19
|
|
|
16
|
-
constructor(protected readonly
|
|
20
|
+
constructor(protected readonly socket: WebSocketOverride) {
|
|
17
21
|
super()
|
|
18
22
|
|
|
19
|
-
this.
|
|
23
|
+
this.socket.addEventListener('close', (event) => this.onClose(event), {
|
|
20
24
|
once: true,
|
|
21
25
|
})
|
|
22
|
-
this.
|
|
26
|
+
this.socket[kOnSend] = (...args) => this.onOutgoing(...args)
|
|
23
27
|
}
|
|
24
28
|
|
|
25
|
-
public send(data:
|
|
29
|
+
public send(data: WebSocketData): void {
|
|
26
30
|
queueMicrotask(() => {
|
|
27
31
|
const message = bindEvent(
|
|
28
32
|
/**
|
|
@@ -33,14 +37,14 @@ export class WebSocketClassTransport extends WebSocketTransport {
|
|
|
33
37
|
* mocked message events like the one below
|
|
34
38
|
* (must be dispatched on the client instance).
|
|
35
39
|
*/
|
|
36
|
-
this.
|
|
40
|
+
this.socket,
|
|
37
41
|
new MessageEvent('message', {
|
|
38
42
|
data,
|
|
39
|
-
origin: this.
|
|
43
|
+
origin: this.socket.url,
|
|
40
44
|
})
|
|
41
45
|
)
|
|
42
46
|
|
|
43
|
-
this.
|
|
47
|
+
this.socket.dispatchEvent(message)
|
|
44
48
|
})
|
|
45
49
|
}
|
|
46
50
|
|
|
@@ -50,6 +54,6 @@ export class WebSocketClassTransport extends WebSocketTransport {
|
|
|
50
54
|
* to allow closing the connection with the status codes
|
|
51
55
|
* that are non-configurable by the user (> 1000 <= 1015).
|
|
52
56
|
*/
|
|
53
|
-
this.
|
|
57
|
+
this.socket[kClose](code, reason)
|
|
54
58
|
}
|
|
55
59
|
}
|
|
@@ -5,7 +5,7 @@
|
|
|
5
5
|
* meant to be used over any WebSocket implementation
|
|
6
6
|
* (not all of them follow the one from WHATWG).
|
|
7
7
|
*/
|
|
8
|
-
import type {
|
|
8
|
+
import type { WebSocketData, WebSocketTransport } from './WebSocketTransport'
|
|
9
9
|
import { WebSocketMessageListener } from './WebSocketOverride'
|
|
10
10
|
import { bindEvent } from './utils/bindEvent'
|
|
11
11
|
import { CloseEvent } from './utils/events'
|
|
@@ -13,36 +13,45 @@ import { uuidv4 } from '../../utils/uuid'
|
|
|
13
13
|
|
|
14
14
|
const kEmitter = Symbol('kEmitter')
|
|
15
15
|
|
|
16
|
+
export interface WebSocketClientConnectionProtocol {
|
|
17
|
+
id: string
|
|
18
|
+
url: URL
|
|
19
|
+
send(data: WebSocketData): void
|
|
20
|
+
close(code?: number, reason?: string): void
|
|
21
|
+
}
|
|
22
|
+
|
|
16
23
|
/**
|
|
17
24
|
* The WebSocket client instance represents an incoming
|
|
18
25
|
* client connection. The user can control the connection,
|
|
19
26
|
* send and receive events.
|
|
20
27
|
*/
|
|
21
|
-
export class WebSocketClientConnection
|
|
28
|
+
export class WebSocketClientConnection
|
|
29
|
+
implements WebSocketClientConnectionProtocol
|
|
30
|
+
{
|
|
22
31
|
public readonly id: string
|
|
23
32
|
public readonly url: URL
|
|
24
33
|
|
|
25
|
-
|
|
34
|
+
private [kEmitter]: EventTarget
|
|
26
35
|
|
|
27
36
|
constructor(
|
|
28
|
-
|
|
29
|
-
|
|
37
|
+
private readonly socket: WebSocket,
|
|
38
|
+
private readonly transport: WebSocketTransport
|
|
30
39
|
) {
|
|
31
40
|
this.id = uuidv4()
|
|
32
|
-
this.url = new URL(
|
|
41
|
+
this.url = new URL(socket.url)
|
|
33
42
|
this[kEmitter] = new EventTarget()
|
|
34
43
|
|
|
35
44
|
// Emit outgoing client data ("ws.send()") as "message"
|
|
36
45
|
// events on the client connection.
|
|
37
46
|
this.transport.onOutgoing = (data) => {
|
|
38
47
|
this[kEmitter].dispatchEvent(
|
|
39
|
-
bindEvent(this.
|
|
48
|
+
bindEvent(this.socket, new MessageEvent('message', { data }))
|
|
40
49
|
)
|
|
41
50
|
}
|
|
42
51
|
|
|
43
52
|
this.transport.onClose = (event) => {
|
|
44
53
|
this[kEmitter].dispatchEvent(
|
|
45
|
-
bindEvent(this.
|
|
54
|
+
bindEvent(this.socket, new CloseEvent('close', event))
|
|
46
55
|
)
|
|
47
56
|
}
|
|
48
57
|
}
|
|
@@ -76,7 +85,7 @@ export class WebSocketClientConnection {
|
|
|
76
85
|
/**
|
|
77
86
|
* Send data to the connected client.
|
|
78
87
|
*/
|
|
79
|
-
public send(data:
|
|
88
|
+
public send(data: WebSocketData): void {
|
|
80
89
|
this.transport.send(data)
|
|
81
90
|
}
|
|
82
91
|
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { invariant } from 'outvariant'
|
|
2
|
-
import type {
|
|
2
|
+
import type { WebSocketData } from './WebSocketTransport'
|
|
3
3
|
import { bindEvent } from './utils/bindEvent'
|
|
4
4
|
import { CloseEvent } from './utils/events'
|
|
5
5
|
|
|
@@ -40,7 +40,7 @@ export class WebSocketOverride extends EventTarget implements WebSocket {
|
|
|
40
40
|
private _onerror: WebSocketEventListener | null = null
|
|
41
41
|
private _onclose: WebSocketCloseListener | null = null
|
|
42
42
|
|
|
43
|
-
private [kOnSend]?: (data:
|
|
43
|
+
private [kOnSend]?: (data: WebSocketData) => void
|
|
44
44
|
|
|
45
45
|
constructor(url: string | URL, protocols?: string | Array<string>) {
|
|
46
46
|
super()
|
|
@@ -110,7 +110,7 @@ export class WebSocketOverride extends EventTarget implements WebSocket {
|
|
|
110
110
|
/**
|
|
111
111
|
* @see https://websockets.spec.whatwg.org/#ref-for-dom-websocket-send%E2%91%A0
|
|
112
112
|
*/
|
|
113
|
-
public send(data:
|
|
113
|
+
public send(data: WebSocketData): void {
|
|
114
114
|
if (this.readyState === this.CONNECTING) {
|
|
115
115
|
this.close()
|
|
116
116
|
throw new DOMException('InvalidStateError')
|
|
@@ -217,7 +217,7 @@ export class WebSocketOverride extends EventTarget implements WebSocket {
|
|
|
217
217
|
}
|
|
218
218
|
}
|
|
219
219
|
|
|
220
|
-
function getDataSize(data:
|
|
220
|
+
function getDataSize(data: WebSocketData): number {
|
|
221
221
|
if (typeof data === 'string') {
|
|
222
222
|
return data.length
|
|
223
223
|
}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { invariant } from 'outvariant'
|
|
2
2
|
import type { WebSocketOverride } from './WebSocketOverride'
|
|
3
|
-
import type {
|
|
3
|
+
import type { WebSocketData } from './WebSocketTransport'
|
|
4
4
|
import type { WebSocketClassTransport } from './WebSocketClassTransport'
|
|
5
5
|
import { bindEvent } from './utils/bindEvent'
|
|
6
6
|
import { CancelableMessageEvent } from './utils/events'
|
|
@@ -20,9 +20,9 @@ export class WebSocketServerConnection {
|
|
|
20
20
|
private [kEmitter]: EventTarget
|
|
21
21
|
|
|
22
22
|
constructor(
|
|
23
|
-
private readonly
|
|
24
|
-
private readonly
|
|
25
|
-
private readonly
|
|
23
|
+
private readonly socket: WebSocketOverride,
|
|
24
|
+
private readonly transport: WebSocketClassTransport,
|
|
25
|
+
private readonly createConnection: () => WebSocket
|
|
26
26
|
) {
|
|
27
27
|
this[kEmitter] = new EventTarget()
|
|
28
28
|
|
|
@@ -58,14 +58,14 @@ export class WebSocketServerConnection {
|
|
|
58
58
|
* Preventing the default on the message event stops this.
|
|
59
59
|
*/
|
|
60
60
|
if (!messageEvent.defaultPrevented) {
|
|
61
|
-
this.
|
|
61
|
+
this.socket.dispatchEvent(
|
|
62
62
|
bindEvent(
|
|
63
63
|
/**
|
|
64
64
|
* @note Bind the forwarded original server events
|
|
65
65
|
* to the mock WebSocket instance so it would
|
|
66
66
|
* dispatch them straight away.
|
|
67
67
|
*/
|
|
68
|
-
this.
|
|
68
|
+
this.socket,
|
|
69
69
|
// Clone the message event again to prevent
|
|
70
70
|
// the "already being dispatched" exception.
|
|
71
71
|
new MessageEvent('message', {
|
|
@@ -105,7 +105,7 @@ export class WebSocketServerConnection {
|
|
|
105
105
|
|
|
106
106
|
// Close the original connection when the (mock)
|
|
107
107
|
// client closes, regardless of the reason.
|
|
108
|
-
this.
|
|
108
|
+
this.socket.addEventListener(
|
|
109
109
|
'close',
|
|
110
110
|
(event) => {
|
|
111
111
|
ws.close(event.code, event.reason)
|
|
@@ -120,9 +120,7 @@ export class WebSocketServerConnection {
|
|
|
120
120
|
// Forward server errors to the WebSocket client as-is.
|
|
121
121
|
// We may consider exposing them to the interceptor in the future.
|
|
122
122
|
ws.addEventListener('error', () => {
|
|
123
|
-
this.
|
|
124
|
-
bindEvent(this.mockWebSocket, new Event('error'))
|
|
125
|
-
)
|
|
123
|
+
this.socket.dispatchEvent(bindEvent(this.socket, new Event('error')))
|
|
126
124
|
})
|
|
127
125
|
|
|
128
126
|
this.realWebSocket = ws
|
|
@@ -165,12 +163,12 @@ export class WebSocketServerConnection {
|
|
|
165
163
|
* server.send(new Blob(['hello']))
|
|
166
164
|
* server.send(new TextEncoder().encode('hello'))
|
|
167
165
|
*/
|
|
168
|
-
public send(data:
|
|
166
|
+
public send(data: WebSocketData): void {
|
|
169
167
|
const { realWebSocket } = this
|
|
170
168
|
invariant(
|
|
171
169
|
realWebSocket,
|
|
172
170
|
'Failed to call "server.send()" for "%s": the connection is not open. Did you forget to call "await server.connect()"?',
|
|
173
|
-
this.
|
|
171
|
+
this.socket.url
|
|
174
172
|
)
|
|
175
173
|
|
|
176
174
|
// Delegate the send to when the original connection is open.
|
|
@@ -1,32 +1,37 @@
|
|
|
1
|
-
export type
|
|
1
|
+
export type WebSocketData = string | ArrayBufferLike | Blob | ArrayBufferView
|
|
2
2
|
|
|
3
3
|
export type WebSocketTransportOnIncomingCallback = (
|
|
4
|
-
event: MessageEvent<
|
|
4
|
+
event: MessageEvent<WebSocketData>
|
|
5
5
|
) => void
|
|
6
6
|
|
|
7
|
-
export type WebSocketTransportOnOutgoingCallback = (
|
|
8
|
-
data: WebSocketRawData
|
|
9
|
-
) => void
|
|
7
|
+
export type WebSocketTransportOnOutgoingCallback = (data: WebSocketData) => void
|
|
10
8
|
|
|
11
9
|
export type WebSocketTransportOnCloseCallback = (event: CloseEvent) => void
|
|
12
10
|
|
|
13
11
|
export abstract class WebSocketTransport {
|
|
14
12
|
/**
|
|
15
|
-
*
|
|
16
|
-
* This is called when the client receives
|
|
17
|
-
*
|
|
18
|
-
*
|
|
19
|
-
* This way, we can trigger the "message" event
|
|
20
|
-
* on the mocked connection to let the user know.
|
|
13
|
+
* A callback for the incoming server events.
|
|
14
|
+
* This is called when the WebSocket client receives
|
|
15
|
+
* a message from the server.
|
|
21
16
|
*/
|
|
22
17
|
abstract onIncoming: WebSocketTransportOnIncomingCallback
|
|
18
|
+
|
|
19
|
+
/**
|
|
20
|
+
* A callback for outgoing client events.
|
|
21
|
+
* This is called when the WebSocket client sends data.
|
|
22
|
+
*/
|
|
23
23
|
abstract onOutgoing: WebSocketTransportOnOutgoingCallback
|
|
24
|
+
|
|
25
|
+
/**
|
|
26
|
+
* A callback for the close client event.
|
|
27
|
+
* This is called when the WebSocket client is closed.
|
|
28
|
+
*/
|
|
24
29
|
abstract onClose: WebSocketTransportOnCloseCallback
|
|
25
30
|
|
|
26
31
|
/**
|
|
27
32
|
* Send the data from the server to this client.
|
|
28
33
|
*/
|
|
29
|
-
abstract send(data:
|
|
34
|
+
abstract send(data: WebSocketData): void
|
|
30
35
|
|
|
31
36
|
/**
|
|
32
37
|
* Close the client connection.
|
|
@@ -1,11 +1,18 @@
|
|
|
1
1
|
import { Interceptor } from '../../Interceptor'
|
|
2
|
-
import {
|
|
2
|
+
import {
|
|
3
|
+
type WebSocketClientConnectionProtocol,
|
|
4
|
+
WebSocketClientConnection,
|
|
5
|
+
} from './WebSocketClientConnection'
|
|
3
6
|
import { WebSocketServerConnection } from './WebSocketServerConnection'
|
|
4
7
|
import { WebSocketClassTransport } from './WebSocketClassTransport'
|
|
5
8
|
import { WebSocketOverride } from './WebSocketOverride'
|
|
6
9
|
|
|
7
|
-
export type
|
|
8
|
-
export {
|
|
10
|
+
export { type WebSocketData, WebSocketTransport } from './WebSocketTransport'
|
|
11
|
+
export {
|
|
12
|
+
WebSocketClientConnection,
|
|
13
|
+
WebSocketClientConnectionProtocol,
|
|
14
|
+
WebSocketServerConnection,
|
|
15
|
+
}
|
|
9
16
|
|
|
10
17
|
export type WebSocketEventMap = {
|
|
11
18
|
connection: [
|
|
@@ -19,7 +26,7 @@ export type WebSocketEventMap = {
|
|
|
19
26
|
* The original WebSocket server connection.
|
|
20
27
|
*/
|
|
21
28
|
server: WebSocketServerConnection
|
|
22
|
-
}
|
|
29
|
+
},
|
|
23
30
|
]
|
|
24
31
|
}
|
|
25
32
|
|
|
@@ -56,22 +63,22 @@ export class WebSocketInterceptor extends Interceptor<WebSocketEventMap> {
|
|
|
56
63
|
// All WebSocket instances are mocked and don't forward
|
|
57
64
|
// any events to the original server (no connection established).
|
|
58
65
|
// To forward the events, the user must use the "server.send()" API.
|
|
59
|
-
const
|
|
60
|
-
const transport = new WebSocketClassTransport(
|
|
66
|
+
const socket = new WebSocketOverride(url, protocols)
|
|
67
|
+
const transport = new WebSocketClassTransport(socket)
|
|
61
68
|
|
|
62
69
|
// The "globalThis.WebSocket" class stands for
|
|
63
70
|
// the client-side connection. Assume it's established
|
|
64
71
|
// as soon as the WebSocket instance is constructed.
|
|
65
72
|
this.emitter.emit('connection', {
|
|
66
|
-
client: new WebSocketClientConnection(
|
|
73
|
+
client: new WebSocketClientConnection(socket, transport),
|
|
67
74
|
server: new WebSocketServerConnection(
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
75
|
+
socket,
|
|
76
|
+
transport,
|
|
77
|
+
createConnection
|
|
71
78
|
),
|
|
72
79
|
})
|
|
73
80
|
|
|
74
|
-
return
|
|
81
|
+
return socket
|
|
75
82
|
},
|
|
76
83
|
})
|
|
77
84
|
|