test cluster and sentinel

Author: nkaradzhov

packages/client/lib/client/index.spec.ts

@@ -261,6 +261,22 @@ describe('Client', () => {
       })
     }, GLOBAL.SERVERS.OPEN);
 
+    testUtils.testWithCluster('Timeout with custom timeout config (cluster)', async cluster => {
+      await blockSetImmediate(async () => {
+        await assert.rejects(cluster.sendCommand(undefined, true, ['PING'], {
+          timeout: 5
+        }), TimeoutError);
+      })
+    }, GLOBAL.CLUSTERS.OPEN);
+
+    testUtils.testWithClientSentinel('Timeout with custom timeout config (sentinel)', async sentinel => {
+      await blockSetImmediate(async () => {
+        await assert.rejects(sentinel.sendCommand(true, ['PING'], {
+          timeout: 5
+        }), TimeoutError);
+      })
+    }, GLOBAL.CLUSTERS.OPEN);
+
     testUtils.testWithClient('Timeout with global timeout config', async client => {
       await blockSetImmediate(async () => {
         await assert.rejects(client.ping(), TimeoutError);
@@ -275,6 +291,34 @@ describe('Client', () => {
       }
     });
 
+    testUtils.testWithCluster('Timeout with global timeout config (cluster)', async cluster => {
+      await blockSetImmediate(async () => {
+        await assert.rejects(cluster.HSET('key', 'foo', 'value'), TimeoutError);
+        await assert.rejects(cluster.sendCommand(undefined, true, ['PING']), TimeoutError);
+      });
+    }, {
+      ...GLOBAL.CLUSTERS.OPEN,
+      clusterConfiguration: {
+        commandOptions: {
+          timeout: 5
+        }
+      }
+    });
+
+    testUtils.testWithClientSentinel('Timeout with global timeout config (sentinel)', async sentinel => {
+      await blockSetImmediate(async () => {
+        await assert.rejects(sentinel.HSET('key', 'foo', 'value'), TimeoutError);
+        await assert.rejects(sentinel.sendCommand(true, ['PING']), TimeoutError);
+      });
+    }, {
+      ...GLOBAL.SENTINEL.OPEN,
+      clientOptions: {
+        commandOptions: {
+          timeout: 5
+        }
+      }
+    });
+
     testUtils.testWithClient('undefined and null should not break the client', async client => {
       await assert.rejects(
         client.sendCommand([null as any, undefined as any]),

packages/client/lib/cluster/index.ts

@@ -38,12 +38,12 @@ export interface RedisClusterOptions<
   // POLICIES extends CommandPolicies = CommandPolicies
 > extends ClusterCommander<M, F, S, RESP, TYPE_MAPPING/*, POLICIES*/> {
   /**
-   * Should contain details for some of the cluster nodes that the client will use to discover 
+   * Should contain details for some of the cluster nodes that the client will use to discover
    * the "cluster topology". We recommend including details for at least 3 nodes here.
    */
   rootNodes: Array<RedisClusterClientOptions>;
   /**
-   * Default values used for every client in the cluster. Use this to specify global values, 
+   * Default values used for every client in the cluster. Use this to specify global values,
    * for example: ACL credentials, timeouts, TLS configuration etc.
    */
   defaults?: Partial<RedisClusterClientOptions>;
@@ -68,13 +68,13 @@ export interface RedisClusterOptions<
   nodeAddressMap?: NodeAddressMap;
   /**
    * Client Side Caching configuration for the pool.
-   * 
-   * Enables Redis Servers and Clients to work together to cache results from commands 
+   *
+   * Enables Redis Servers and Clients to work together to cache results from commands
    * sent to a server. The server will notify the client when cached results are no longer valid.
    * In pooled mode, the cache is shared across all clients in the pool.
-   * 
+   *
    * Note: Client Side Caching is only supported with RESP3.
-   * 
+   *
    * @example Anonymous cache configuration
    * ```
    * const client = createCluster({
@@ -86,7 +86,7 @@ export interface RedisClusterOptions<
    *   minimum: 5
    * });
    * ```
-   * 
+   *
    * @example Using a controllable cache
    * ```
    * const cache = new BasicPooledClientSideCache({
@@ -406,7 +406,7 @@ export default class RedisCluster<
     proxy._commandOptions[key] = value;
     return proxy as RedisClusterType<
       M,
-      F, 
+      F,
       S,
       RESP,
       K extends 'typeMapping' ? V extends TypeMapping ? V : {} : TYPE_MAPPING
@@ -489,15 +489,15 @@ export default class RedisCluster<
           myFn = this._handleAsk(fn);
           continue;
         }
-        
+
         if (err.message.startsWith('MOVED')) {
           await this._slots.rediscover(client);
           client = await this._slots.getClient(firstKey, isReadonly);
           continue;
         }
 
         throw err;
-      } 
+      }
     }
   }
 
@@ -508,10 +508,16 @@ export default class RedisCluster<
     options?: ClusterCommandOptions,
     // defaultPolicies?: CommandPolicies
   ): Promise<T> {
+
+    // Merge global options with local options
+    const opts = {
+      ...this._self._commandOptions,
+      ...options
+    }
     return this._self._execute(
       firstKey,
       isReadonly,
-      options,
+      opts,
       (client, opts) => client.sendCommand(args, opts)
     );
   }

packages/client/lib/sentinel/index.ts

@@ -35,7 +35,7 @@ export class RedisSentinelClient<
 
   /**
    * Indicates if the client connection is open
-   * 
+   *
    * @returns `true` if the client connection is open, `false` otherwise
    */
 
@@ -45,7 +45,7 @@ export class RedisSentinelClient<
 
   /**
    * Indicates if the client connection is ready to accept commands
-   * 
+   *
    * @returns `true` if the client connection is ready, `false` otherwise
    */
   get isReady() {
@@ -54,7 +54,7 @@ export class RedisSentinelClient<
 
   /**
    * Gets the command options configured for this client
-   * 
+   *
    * @returns The command options for this client or `undefined` if none were set
    */
   get commandOptions() {
@@ -241,10 +241,10 @@ export class RedisSentinelClient<
 
   /**
    * Releases the client lease back to the pool
-   * 
+   *
    * After calling this method, the client instance should no longer be used as it
    * will be returned to the client pool and may be given to other operations.
-   * 
+   *
    * @returns A promise that resolves when the client is ready to be reused, or undefined
    *          if the client was immediately ready
    * @throws Error if the lease has already been released
@@ -274,7 +274,7 @@ export default class RedisSentinel<
 
   /**
    * Indicates if the sentinel connection is open
-   * 
+   *
    * @returns `true` if the sentinel connection is open, `false` otherwise
    */
   get isOpen() {
@@ -283,7 +283,7 @@ export default class RedisSentinel<
 
   /**
    * Indicates if the sentinel connection is ready to accept commands
-   * 
+   *
    * @returns `true` if the sentinel connection is ready, `false` otherwise
    */
   get isReady() {
@@ -554,15 +554,15 @@ export default class RedisSentinel<
 
   /**
    * Acquires a master client lease for exclusive operations
-   * 
+   *
    * Used when multiple commands need to run on an exclusive client (for example, using `WATCH/MULTI/EXEC`).
    * The returned client must be released after use with the `release()` method.
-   * 
+   *
    * @returns A promise that resolves to a Redis client connected to the master node
    * @example
    * ```javascript
    * const clientLease = await sentinel.acquire();
-   * 
+   *
    * try {
    *   await clientLease.watch('key');
    *   const resp = await clientLease.multi()
@@ -670,7 +670,7 @@ class RedisSentinelInternal<
     super();
 
     this.#validateOptions(options);
-    
+
     this.#name = options.name;
 
     this.#sentinelRootNodes = Array.from(options.sentinelRootNodes);
@@ -728,7 +728,7 @@ class RedisSentinelInternal<
 
   /**
    * Gets a client lease from the master client pool
-   * 
+   *
    * @returns A client info object or a promise that resolves to a client info object
    *          when a client becomes available
    */
@@ -743,10 +743,10 @@ class RedisSentinelInternal<
 
   /**
    * Releases a client lease back to the pool
-   * 
+   *
    * If the client was used for a transaction that might have left it in a dirty state,
    * it will be reset before being returned to the pool.
-   * 
+   *
    * @param clientInfo The client info object representing the client to release
    * @returns A promise that resolves when the client is ready to be reused, or undefined
    *          if the client was immediately ready or no longer exists
@@ -786,10 +786,10 @@ class RedisSentinelInternal<
 
   async #connect() {
     let count = 0;
-    while (true) {      
+    while (true) {
       this.#trace("starting connect loop");
 
-      count+=1;      
+      count+=1;
       if (this.#destroy) {
         this.#trace("in #connect and want to destroy")
         return;
@@ -842,7 +842,7 @@ class RedisSentinelInternal<
 
       try {
         /*
-                // force testing of READONLY errors        
+                // force testing of READONLY errors
                 if (clientInfo !== undefined) {
                   if (Math.floor(Math.random() * 10) < 1) {
                     console.log("throwing READONLY error");
@@ -856,7 +856,7 @@ class RedisSentinelInternal<
           throw err;
         }
 
-        /* 
+        /*
           rediscover and retry if doing a command against a "master"
           a) READONLY error (topology has changed) but we haven't been notified yet via pubsub
           b) client is "not ready" (disconnected), which means topology might have changed, but sentinel might not see it yet
@@ -1569,4 +1569,4 @@ export class RedisSentinelFactory extends EventEmitter {
       }
     });
   }
-}
+}

packages/client/lib/sentinel/utils.ts

@@ -6,7 +6,7 @@ import { NamespaceProxySentinel, NamespaceProxySentinelClient, ProxySentinel, Pr
 
 /* TODO: should use map interface, would need a transform reply probably? as resp2 is list form, which this depends on */
 export function parseNode(node: Record<string, string>): RedisNode | undefined{
- 
+
   if (node.flags.includes("s_down") || node.flags.includes("disconnected") || node.flags.includes("failover_in_progress")) {
     return undefined;
   }

packages/test-utils/lib/index.ts

@@ -179,7 +179,7 @@ export default class TestUtils {
     this.#VERSION_NUMBERS = numbers;
     this.#DOCKER_IMAGE = {
       image: dockerImageName,
-      version: string, 
+      version: string,
       mode: "server"
     };
   }
@@ -315,7 +315,7 @@ export default class TestUtils {
     if (passIndex != 0) {
       password = options.serverArguments[passIndex];
     }
-    
+
     if (this.isVersionGreaterThan(options.minimumDockerVersion)) {
       const dockerImage = this.#DOCKER_IMAGE;
       before(function () {
@@ -333,17 +333,18 @@ export default class TestUtils {
 
       const promises = await dockerPromises;
       const rootNodes: Array<RedisNode> = promises.map(promise => ({
-        host: "127.0.0.1", 
+        host: "127.0.0.1",
         port: promise.port
       }));
 
       const sentinel = createSentinel({
-        name: 'mymaster', 
-        sentinelRootNodes: rootNodes, 
-        nodeClientOptions: { 
+        name: 'mymaster',
+        sentinelRootNodes: rootNodes,
+        nodeClientOptions: {
+          commandOptions: options.clientOptions?.commandOptions,
           password: password || undefined,
         },
-        sentinelClientOptions: { 
+        sentinelClientOptions: {
           password: password || undefined,
         },
         replicaPoolSize: options?.replicaPoolSize || 0,
@@ -505,7 +506,7 @@ export default class TestUtils {
 
     it(title, async function () {
       if (!dockersPromise) return this.skip();
-    
+
       const dockers = await dockersPromise,
         cluster = createCluster({
           rootNodes: dockers.map(({ port }) => ({
@@ -578,12 +579,12 @@ export default class TestUtils {
       const tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), appPrefix));
 
       sentinels.push(await spawnSentinelNode(this.#DOCKER_IMAGE, options.serverArguments, masterPort, sentinelName, tmpDir))
-      
+
       if (tmpDir) {
         fs.rmSync(tmpDir, { recursive: true });
       }
     }
-    
+
     return sentinels
   }
 }