1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
// @ts-nocheck
import type { Server } from "socket.io";
import request from "supertest";
import { io as ioc, ManagerOptions, Socket as ClientSocket, SocketOptions } from "socket.io-client";
export function createClient(
io: Server,
nsp: string = "/",
opts?: Partial<ManagerOptions & SocketOptions>,
): ClientSocket {
// @ts-ignore
const port = io.httpServer.address().port;
return ioc(`http://localhost:${port}${nsp}`, opts);
}
export function success(done: Function, io: Server, ...clients: ClientSocket[]) {
io.close();
clients.forEach(client => client.disconnect());
done();
}
export function fail(done: Function, io: Server, err: any | unknown, ...clients: ClientSocket[]) {
io.close();
clients.forEach(client => client.disconnect());
done(err);
}
export function getPort(io: Server): number {
// @ts-ignore
return io.httpServer.address().port;
}
export function createPartialDone(count: number, done: (err?: Error) => void) {
let i = 0;
return () => {
if (++i === count) {
done();
} else if (i > count) {
done(new Error(`partialDone() called too many times: ${i} > ${count}`));
}
};
}
// TODO: update superagent as latest release now supports promises
export function eioHandshake(httpServer): Promise<string> {
return new Promise(resolve => {
request(httpServer)
.get("/socket.io/")
.query({ transport: "polling", EIO: 4 })
.end((err, res) => {
const sid = JSON.parse(res.text.substring(1)).sid;
resolve(sid);
});
});
}
export function eioPush(httpServer, sid: string, body: string): Promise<void> {
return new Promise(resolve => {
request(httpServer)
.post("/socket.io/")
.send(body)
.query({ transport: "polling", EIO: 4, sid })
.expect(200)
.end(() => {
resolve();
});
});
}
export function eioPoll(httpServer, sid): Promise<string> {
return new Promise(resolve => {
request(httpServer)
.get("/socket.io/")
.query({ transport: "polling", EIO: 4, sid })
.expect(200)
.end((err, res) => {
resolve(res.text);
});
});
}
export function waitFor<T = unknown>(emitter, event) {
return new Promise<T>(resolve => {
emitter.once(event, resolve);
});
}
|
