[CAE-342] Minor refactor

Author: htemelski-redis

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

@@ -1,12 +1,11 @@
 import { strict as assert } from 'node:assert';
 import { setTimeout } from 'node:timers/promises';
-import testUtils, { GLOBAL } from '../test-utils';
+import testUtils, { GLOBAL, MATH_FUNCTION } from '../test-utils';
 import { RESP_TYPES } from '../RESP/decoder';
 import { WatchError } from "../errors";
-import { MATH_FUNCTION } from '../commands/FUNCTION_LOAD.spec';
 import { RedisSentinelConfig, SentinelFramework } from "./test-util";
 import { RedisSentinelEvent, RedisSentinelType, RedisSentinelClientType, RedisNode } from "./types";
-import { RedisModules, RedisFunctions, RedisScripts, RespVersions, TypeMapping } from '../RESP/types';
+import { RedisModules, RedisFunctions, RedisScripts, RespVersions, TypeMapping, NumberReply } from '../RESP/types';
 import { promisify } from 'node:util';
 import { exec } from 'node:child_process';
 const execAsync = promisify(exec);
@@ -417,7 +416,7 @@ describe('legacy tests', () => {
   })
 
   describe('Sentinel Client', function () {
-    let sentinel: RedisSentinelType<    RedisModules, RedisFunctions, RedisScripts, RespVersions, TypeMapping> | undefined;
+    let sentinel: RedisSentinelType<RedisModules, RedisFunctions, RedisScripts, RespVersions, TypeMapping> | undefined;
 
     beforeEach(async function () {
       this.timeout(0);

packages/client/lib/test-utils.ts

@@ -4,7 +4,6 @@ import { setTimeout } from 'node:timers/promises';
 import { CredentialsProvider } from './authx';
 import { Command, NumberReply } from './RESP/types';
 import { BasicCommandParser, CommandParser } from './client/parser';
-import { MATH_FUNCTION } from './commands/FUNCTION_LOAD.spec';
 import { defineScript } from './lua-script';
 import RedisBloomModules from '@redis/bloom';
 const utils = TestUtils.createFromConfig({
@@ -56,6 +55,33 @@ const SQUARE_SCRIPT = defineScript({
   transformReply: undefined as unknown as () => NumberReply
 });
 
+export const MATH_FUNCTION = {
+  name: 'math',
+  engine: 'LUA',
+  code:
+    `#!LUA name=math
+    redis.register_function {
+      function_name = "square",
+      callback = function(keys, args)
+        local number = redis.call('GET', keys[1])
+        return number * number
+      end,
+      flags = { "no-writes" }
+    }`,
+  library: {
+    square: {
+      NAME: 'square',
+      IS_READ_ONLY: true,
+      NUMBER_OF_KEYS: 1,
+      FIRST_KEY_INDEX: 0,
+      parseCommand(parser: CommandParser, key: string) {
+        parser.pushKey(key);
+      },
+      transformReply: undefined as unknown as () => NumberReply
+    }
+  }
+};
+
 export const GLOBAL = {
   SERVERS: {
     OPEN: {
@@ -122,7 +148,7 @@ export const GLOBAL = {
       password: undefined,
       functions: {
         math: MATH_FUNCTION.library,
-      } 
+      },
     },
     WITH_MODULE: {
       serverArguments: [...DEBUG_MODE_ARGS],