Write tests

redis · Jan 10, 2024 · 9720e08 · 9720e08
1 parent fa8a93a
commit 9720e08
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/packages/client/lib/cluster/index.spec.ts b/packages/client/lib/cluster/index.spec.ts
@@ -235,6 +235,20 @@ describe('Cluster', () => {
 
             assert.equal(cluster.pubSubNode, undefined);
         }, GLOBAL.CLUSTERS.OPEN);
+
+        testUtils.testWithCluster('concurrent unsubscribe does not throw an error', async cluster => {
+            const listener = spy();
+
+            await cluster.subscribe('channel', listener);
+            await cluster.subscribe('channel2', listener);
+
+            await Promise.all([
+                cluster.unsubscribe('channel', listener),
+                cluster.unsubscribe('channel2', listener)
+            ]);
+
+            assert.equal(cluster.isOpen, false);
+        }, GLOBAL.CLUSTERS.OPEN);
 
         testUtils.testWithCluster('psubscribe & punsubscribe', async cluster => {
             const listener = spy();
@@ -323,6 +337,24 @@ describe('Cluster', () => {
             minimumDockerVersion: [7]
         });
 
+        testUtils.testWithCluster('concurrent sunsubscribe does not throw an error', async cluster => {
+            const listener = spy();
+
+            await cluster.sSubscribe('channel', listener);
+            await cluster.sSubscribe('channel2', listener);
+
+            await Promise.all([
+                cluster.sUnsubscribe('channel', listener),
+                cluster.sUnsubscribe('channel2', listener)
+            ])
+
+
+            assert.equal(cluster.isOpen, false);
+        }, {
+            ...GLOBAL.CLUSTERS.OPEN,
+            minimumDockerVersion: [7]
+        });
+
         testUtils.testWithCluster('should handle sharded-channel-moved events', async cluster => {
             const SLOT = 10328,
                 migrating = cluster.slots[SLOT].master,