fix: websocket connection of LiveQuery interrupts frequently (#8048)

This commit is contained in:
Layne Bernardo
2022-06-17 06:20:48 -05:00
committed by GitHub
parent 72fac8a5fc
commit 03caae1e61
2 changed files with 32 additions and 1 deletions

View File

@@ -76,6 +76,35 @@ describe('ParseWebSocketServer', function () {
expect(wssError).toBe('Invalid Packet');
});
it('can handle ping/pong', async () => {
const onConnectCallback = jasmine.createSpy('onConnectCallback');
const http = require('http');
const server = http.createServer();
const parseWebSocketServer = new ParseWebSocketServer(server, onConnectCallback, {
websocketTimeout: 10,
}).server;
const ws = new EventEmitter();
ws.readyState = 0;
ws.OPEN = 0;
ws.ping = jasmine.createSpy('ping');
ws.terminate = jasmine.createSpy('terminate');
parseWebSocketServer.onConnection(ws);
expect(onConnectCallback).toHaveBeenCalled();
expect(ws.waitingForPong).toBe(false);
await new Promise(resolve => setTimeout(resolve, 10));
expect(ws.ping).toHaveBeenCalled();
expect(ws.waitingForPong).toBe(true);
ws.emit('pong');
expect(ws.waitingForPong).toBe(false);
await new Promise(resolve => setTimeout(resolve, 10));
expect(ws.waitingForPong).toBe(true);
expect(ws.terminate).not.toHaveBeenCalled();
server.close();
});
it('closes interrupted connection', async () => {
const onConnectCallback = jasmine.createSpy('onConnectCallback');
const http = require('http');
@@ -93,8 +122,10 @@ describe('ParseWebSocketServer', function () {
// Make sure callback is called
expect(onConnectCallback).toHaveBeenCalled();
expect(ws.waitingForPong).toBe(false);
await new Promise(resolve => setTimeout(resolve, 10));
expect(ws.ping).toHaveBeenCalled();
expect(ws.waitingForPong).toBe(true);
await new Promise(resolve => setTimeout(resolve, 10));
expect(ws.terminate).toHaveBeenCalled();
server.close();

View File

@@ -16,7 +16,7 @@ export class ParseWebSocketServer {
wss.onConnection = ws => {
ws.waitingForPong = false;
ws.on('pong', () => {
this.waitingForPong = false;
ws.waitingForPong = false;
});
ws.on('error', error => {
logger.error(error.message);