node-redis/packages/entraid/lib/test-utils.ts

import { AuthenticationResult } from '@azure/msal-node';
import { IdentityProvider, StreamingCredentialsProvider, TokenManager, TokenResponse } from '@redis/client/dist/lib/authx';
import TestUtils from '@redis/test-utils';
import { EntraidCredentialsProvider } from './entraid-credentials-provider';

export const testUtils = TestUtils.createFromConfig({
  dockerImageName: 'redislabs/client-libs-test',
  dockerImageVersionArgument: 'redis-version',
  defaultDockerVersion: '8.2-M01-pre'
});

const DEBUG_MODE_ARGS = testUtils.isVersionGreaterThan([7]) ?
  ['--enable-debug-command', 'yes'] :
  [];

const idp: IdentityProvider<AuthenticationResult> = {
  requestToken(): Promise<TokenResponse<AuthenticationResult>> {
    // @ts-ignore
    return Promise.resolve({
      ttlMs: 100000,
      token: {
        accessToken: 'password'
      }
    })
  }
}

const tokenManager = new TokenManager<AuthenticationResult>(idp, { expirationRefreshRatio: 0.8 });
const entraIdCredentialsProvider: StreamingCredentialsProvider = new EntraidCredentialsProvider(tokenManager, idp)

const PASSWORD_WITH_REPLICAS = {
  serverArguments: ['--requirepass', 'password', ...DEBUG_MODE_ARGS],
  numberOfMasters: 2,
  numberOfReplicas: 1,
  clusterConfiguration: {
    defaults: {
      credentialsProvider: entraIdCredentialsProvider
    }
  }
}

export const GLOBAL = {
  CLUSTERS: {
    PASSWORD_WITH_REPLICAS
  }
}