fix: Add keys and watch to LiveQueryClient.resubscribe

Author: dplewis

integration/test/ParseLiveQueryTest.js

@@ -90,6 +90,42 @@ describe('Parse LiveQuery', () => {
     await promise;
   });
 
+  it('can resubscribe', async () => {
+    const client = new Parse.LiveQueryClient({
+      applicationId: 'integration',
+      serverURL: 'ws://localhost:1337',
+      javascriptKey: null,
+      masterKey: null,
+      sessionToken: null,
+    });
+    client.open();
+    const resubscribeSpy = spyOn(client, 'resubscribe').and.callThrough();
+    const consoleSpy = spyOn(console, 'error').and.callFake(() => {});
+    const subscribeRequest = {
+      op: 'subscribe',
+      requestId: 1,
+      query: {
+        className: 'TestObject',
+        where: { objectId: 'HEXkuHFm0D' },
+        keys: ['foo', 'objectId'],
+        watch: undefined,
+        unknownField: 'throws Additional properties not allowed error',
+      },
+      sessionToken: undefined,
+    };
+    await client.connectPromise;
+    client.socket.send(JSON.stringify(subscribeRequest));
+    await sleep(1000);
+    expect(resubscribeSpy).toHaveBeenCalled();
+    expect(consoleSpy).toHaveBeenCalledWith(
+      'LiveQuery reconnecting with error:',
+      'Additional properties not allowed',
+      'code:',
+      1
+    );
+    await client.close();
+  });
+
   it('can subscribe to multiple queries', async () => {
     const objectA = new TestObject();
     const objectB = new TestObject();

src/LiveQueryClient.ts

@@ -55,6 +55,7 @@ const SUBSCRIPTION_EMMITER_TYPES = {
   DELETE: 'delete',
 };
 
+// Exponentially-growing random delay
 const generateInterval = k => {
   return Math.random() * Math.min(30, Math.pow(2, k) - 1) * 1000;
 };
@@ -291,7 +292,8 @@ class LiveQueryClient {
       const query = subscription.query;
       const queryJSON = query.toJSON();
       const where = queryJSON.where;
-      const fields = queryJSON.keys ? queryJSON.keys.split(',') : undefined;
+      const keys = queryJSON.keys?.split(',');
+      const watch = queryJSON.watch?.split(',');
       const className = query.className;
       const sessionToken = subscription.sessionToken;
       const subscribeRequest = {
@@ -300,7 +302,8 @@ class LiveQueryClient {
         query: {
           className,
           where,
-          fields,
+          keys,
+          watch,
         },
         sessionToken: undefined as string | undefined,
       };
@@ -347,7 +350,6 @@ class LiveQueryClient {
   }
 
   _handleWebSocketOpen() {
-    this.attempts = 1;
     const connectRequest = {
       op: OP_TYPES.CONNECT,
       applicationId: this.applicationId,
@@ -387,6 +389,7 @@ class LiveQueryClient {
       break;
     case OP_EVENTS.SUBSCRIBED:
       if (subscription) {
+        this.attempts = 1;
         subscription.subscribed = true;
         subscription.subscribePromise.resolve();
         setTimeout(() => subscription.emit(SUBSCRIPTION_EMMITER_TYPES.OPEN, response), 200);
@@ -410,6 +413,7 @@ class LiveQueryClient {
         this.additionalProperties = false;
       }
       if (data.reconnect) {
+        console.error('LiveQuery reconnecting with error:', data.error, 'code:', data.code);
         this._handleReconnect();
       }
       break;
@@ -485,19 +489,16 @@ class LiveQueryClient {
     if (this.state === CLIENT_STATE.DISCONNECTED) {
       return;
     }
-
     this.state = CLIENT_STATE.RECONNECTING;
     const time = generateInterval(this.attempts);
 
     // handle case when both close/error occur at frequent rates we ensure we do not reconnect unnecessarily.
     // we're unable to distinguish different between close/error when we're unable to reconnect therefore
     // we try to reconnect in both cases
     // server side ws and browser WebSocket behave differently in when close/error get triggered
-
     if (this.reconnectHandle) {
       clearTimeout(this.reconnectHandle);
     }
-
     this.reconnectHandle = setTimeout(
       (() => {
         this.attempts++;

src/__tests__/LiveQueryClient-test.js

@@ -748,6 +748,7 @@ describe('LiveQueryClient', () => {
       expect(error).toEqual(data.error);
     });
     const spy = jest.spyOn(liveQueryClient, '_handleReconnect');
+    jest.spyOn(console, 'error').mockImplementationOnce(() => {});
     try {
       liveQueryClient._handleWebSocketMessage(event);
       await liveQueryClient.connectPromise;
@@ -944,6 +945,8 @@ describe('LiveQueryClient', () => {
     };
     const query = new ParseQuery('Test');
     query.equalTo('key', 'value');
+    query.select(['key']);
+    query.watch(['key']);
     liveQueryClient.subscribe(query);
     liveQueryClient.connectPromise.resolve();
 
@@ -961,6 +964,8 @@ describe('LiveQueryClient', () => {
         where: {
           key: 'value',
         },
+        keys: ['key'],
+        watch: ['key'],
       },
     });
   });
@@ -978,6 +983,8 @@ describe('LiveQueryClient', () => {
     };
     const query = new ParseQuery('Test');
     query.equalTo('key', 'value');
+    query.select(['key']);
+    query.watch(['key']);
     liveQueryClient.subscribe(query, 'mySessionToken');
     liveQueryClient.connectPromise.resolve();
 
@@ -996,6 +1003,8 @@ describe('LiveQueryClient', () => {
         where: {
           key: 'value',
         },
+        keys: ['key'],
+        watch: ['key'],
       },
     });
   });