test: reconnect silently with promise

enisdenjo · Dec 9, 2020 · 85f041e · 85f041e
1 parent b6b9c6b
commit 85f041e
Showing 1 changed file with 19 additions and 10 deletions.
diff --git a/src/tests/client.ts b/src/tests/client.ts
@@ -664,21 +664,30 @@ describe('reconnecting', () => {
   it('should reconnect silently after socket closes', async () => {
     const { url, ...server } = await startTServer();
 
-    const sub = tsubscribe(
-      createClient({
-        url,
-        retryAttempts: 1,
-      }),
-      {
-        query: 'subscription { ping }',
-      },
-    );
+    const client = createClient({
+      url,
+      retryAttempts: 3,
+      retryWait: () => Promise.resolve(),
+    });
+    const sub = tsubscribe(client, {
+      query: 'subscription { ping }',
+    });
+
+    await server.waitForClient((client) => {
+      client.close();
+    });
+
+    // retried
+    await server.waitForClient((client) => {
+      client.close();
+    });
 
+    // once more
     await server.waitForClient((client) => {
       client.close();
     });
 
-    // tried again
+    // and once more
     await server.waitForClient((client) => {
       client.close();
     });