You've already forked node-redis
mirror of
https://github.com/redis/node-redis.git
synced 2025-08-07 13:22:56 +03:00
feat(auth): add Entra ID identity provider integration for Redis client authentication (#2877)
* feat(auth): refactor authentication mechanism to use CredentialsProvider - Introduce new credential providers: AsyncCredentialsProvider, StreamingCredentialsProvider - Update client handshake process to use the new CredentialsProviders and to support async credentials fetch / credentials refresh - Internal conversion of username/password to a CredentialsProvider - Modify URL parsing to accommodate the new authentication structure - Tests * feat(auth): auth extensions Introduces TokenManager and supporting classes to handle token acquisition, automatic refresh, and updates via identity providers. This foundation enables consistent authentication token management across different identity provider implementations. Key additions: - Add TokenManager to obtain and maintain auth tokens from identity providers with automated refresh scheduling based on TTL and configurable thresholds - Add IdentityProvider interface for token acquisition from auth providers - Implement Token class for managing token state and TTL tracking - Include configurable retry mechanism with exponential backoff and jitter - Add comprehensive test suite covering refresh cycles and error handling This change establishes the core infrastructure needed for reliable token lifecycle management across different authentication providers. * feat(auth): add Entra ID identity provider integration Introduces Entra ID (former Azure AD) authentication support with multiple authentication flows and automated token lifecycle management. Key additions: - Add EntraIdCredentialsProvider for handling Entra ID authentication flows - Implement MSALIdentityProvider to integrate with MSAL/EntraID authentication library - Add support for multiple authentication methods: - Managed identities (system and user-assigned) - Client credentials with certificate - Client credentials with secret - Authorization Code flow with PKCE - Add factory class with builder methods for each authentication flow - Include sample Express server implementation for Authorization Code flow - Add comprehensive configuration options for authority and token management * feat(test-utils): improve cluster testing - Add support for configuring replica authentication with 'masterauth' - Allow default client configuration during test cluster creation This improves the testing framework's flexibility by automatically configuring replica authentication when '--requirepass' is used and enabling custom client configurations across cluster nodes. * feat(auth): add EntraId integration tests - Add integration tests for token renewal and re-authentication flows - Update credentials provider to use uniqueId as username instead of account username - Add test utilities for loading Redis endpoint configurations - Split TypeScript configs into separate files for samples and integration tests - Remove `@redis/authx` package and nest it under `@`
This commit is contained in:
217
packages/entraid/integration-tests/entraid-integration.spec.ts
Normal file
217
packages/entraid/integration-tests/entraid-integration.spec.ts
Normal file
@@ -0,0 +1,217 @@
|
||||
import { BasicAuth } from '@redis/client/dist/lib/authx';
|
||||
import { createClient } from '@redis/client';
|
||||
import { EntraIdCredentialsProviderFactory } from '../lib/entra-id-credentials-provider-factory';
|
||||
import { strict as assert } from 'node:assert';
|
||||
import { spy, SinonSpy } from 'sinon';
|
||||
import { randomUUID } from 'crypto';
|
||||
import { loadFromFile, RedisEndpointsConfig } from '@redis/test-utils/lib/cae-client-testing';
|
||||
import { EntraidCredentialsProvider } from '../lib/entraid-credentials-provider';
|
||||
import * as crypto from 'node:crypto';
|
||||
|
||||
describe('EntraID Integration Tests', () => {
|
||||
|
||||
it('client configured with client secret should be able to authenticate/re-authenticate', async () => {
|
||||
const config = await readConfigFromEnv();
|
||||
await runAuthenticationTest(() =>
|
||||
EntraIdCredentialsProviderFactory.createForClientCredentials({
|
||||
clientId: config.clientId,
|
||||
clientSecret: config.clientSecret,
|
||||
authorityConfig: { type: 'multi-tenant', tenantId: config.tenantId },
|
||||
tokenManagerConfig: {
|
||||
expirationRefreshRatio: 0.0001
|
||||
}
|
||||
})
|
||||
);
|
||||
});
|
||||
|
||||
it('client configured with client certificate should be able to authenticate/re-authenticate', async () => {
|
||||
const config = await readConfigFromEnv();
|
||||
await runAuthenticationTest(() =>
|
||||
EntraIdCredentialsProviderFactory.createForClientCredentialsWithCertificate({
|
||||
clientId: config.clientId,
|
||||
certificate: convertCertsForMSAL(config.cert, config.privateKey),
|
||||
authorityConfig: { type: 'multi-tenant', tenantId: config.tenantId },
|
||||
tokenManagerConfig: {
|
||||
expirationRefreshRatio: 0.0001
|
||||
}
|
||||
})
|
||||
);
|
||||
});
|
||||
|
||||
it('client with system managed identity should be able to authenticate/re-authenticate', async () => {
|
||||
const config = await readConfigFromEnv();
|
||||
await runAuthenticationTest(() =>
|
||||
EntraIdCredentialsProviderFactory.createForSystemAssignedManagedIdentity({
|
||||
clientId: config.clientId,
|
||||
authorityConfig: { type: 'multi-tenant', tenantId: config.tenantId },
|
||||
tokenManagerConfig: {
|
||||
expirationRefreshRatio: 0.00001
|
||||
}
|
||||
})
|
||||
);
|
||||
});
|
||||
|
||||
interface TestConfig {
|
||||
clientId: string;
|
||||
clientSecret: string;
|
||||
authority: string;
|
||||
tenantId: string;
|
||||
redisScopes: string;
|
||||
cert: string;
|
||||
privateKey: string;
|
||||
userAssignedManagedId: string;
|
||||
endpoints: RedisEndpointsConfig;
|
||||
}
|
||||
|
||||
const readConfigFromEnv = async (): Promise<TestConfig> => {
|
||||
const requiredEnvVars = {
|
||||
AZURE_CLIENT_ID: process.env.AZURE_CLIENT_ID,
|
||||
AZURE_CLIENT_SECRET: process.env.AZURE_CLIENT_SECRET,
|
||||
AZURE_AUTHORITY: process.env.AZURE_AUTHORITY,
|
||||
AZURE_TENANT_ID: process.env.AZURE_TENANT_ID,
|
||||
AZURE_REDIS_SCOPES: process.env.AZURE_REDIS_SCOPES,
|
||||
AZURE_CERT: process.env.AZURE_CERT,
|
||||
AZURE_PRIVATE_KEY: process.env.AZURE_PRIVATE_KEY,
|
||||
AZURE_USER_ASSIGNED_MANAGED_ID: process.env.AZURE_USER_ASSIGNED_MANAGED_ID,
|
||||
REDIS_ENDPOINTS_CONFIG_PATH: process.env.REDIS_ENDPOINTS_CONFIG_PATH
|
||||
};
|
||||
|
||||
Object.entries(requiredEnvVars).forEach(([key, value]) => {
|
||||
if (value == undefined) {
|
||||
throw new Error(`${key} environment variable must be set`);
|
||||
}
|
||||
});
|
||||
|
||||
return {
|
||||
endpoints: await loadFromFile(requiredEnvVars.REDIS_ENDPOINTS_CONFIG_PATH),
|
||||
clientId: requiredEnvVars.AZURE_CLIENT_ID,
|
||||
clientSecret: requiredEnvVars.AZURE_CLIENT_SECRET,
|
||||
authority: requiredEnvVars.AZURE_AUTHORITY,
|
||||
tenantId: requiredEnvVars.AZURE_TENANT_ID,
|
||||
redisScopes: requiredEnvVars.AZURE_REDIS_SCOPES,
|
||||
cert: requiredEnvVars.AZURE_CERT,
|
||||
privateKey: requiredEnvVars.AZURE_PRIVATE_KEY,
|
||||
userAssignedManagedId: requiredEnvVars.AZURE_USER_ASSIGNED_MANAGED_ID
|
||||
};
|
||||
};
|
||||
|
||||
interface TokenDetail {
|
||||
token: string;
|
||||
exp: number;
|
||||
iat: number;
|
||||
lifetime: number;
|
||||
uti: string;
|
||||
}
|
||||
|
||||
const setupTestClient = async (credentialsProvider: EntraidCredentialsProvider) => {
|
||||
const config = await readConfigFromEnv();
|
||||
const client = createClient({
|
||||
url: config.endpoints['standalone-entraid-acl'].endpoints[0],
|
||||
credentialsProvider
|
||||
});
|
||||
|
||||
const clientInstance = (client as any)._self;
|
||||
const reAuthSpy: SinonSpy = spy(clientInstance, 'reAuthenticate');
|
||||
|
||||
return { client, reAuthSpy };
|
||||
};
|
||||
|
||||
const runClientOperations = async (client: any) => {
|
||||
const startTime = Date.now();
|
||||
while (Date.now() - startTime < 1000) {
|
||||
const key = randomUUID();
|
||||
await client.set(key, 'value');
|
||||
const value = await client.get(key);
|
||||
assert.equal(value, 'value');
|
||||
await client.del(key);
|
||||
}
|
||||
};
|
||||
|
||||
const validateTokens = (reAuthSpy: SinonSpy) => {
|
||||
assert(reAuthSpy.callCount >= 1,
|
||||
`reAuthenticate should have been called at least once, but was called ${reAuthSpy.callCount} times`);
|
||||
|
||||
const tokenDetails: TokenDetail[] = reAuthSpy.getCalls().map(call => {
|
||||
const creds = call.args[0] as BasicAuth;
|
||||
const tokenPayload = JSON.parse(
|
||||
Buffer.from(creds.password.split('.')[1], 'base64').toString()
|
||||
);
|
||||
|
||||
return {
|
||||
token: creds.password,
|
||||
exp: tokenPayload.exp,
|
||||
iat: tokenPayload.iat,
|
||||
lifetime: tokenPayload.exp - tokenPayload.iat,
|
||||
uti: tokenPayload.uti
|
||||
};
|
||||
});
|
||||
|
||||
// Verify unique tokens
|
||||
const uniqueTokens = new Set(tokenDetails.map(detail => detail.token));
|
||||
assert.equal(
|
||||
uniqueTokens.size,
|
||||
reAuthSpy.callCount,
|
||||
`Expected ${reAuthSpy.callCount} different tokens, but got ${uniqueTokens.size} unique tokens`
|
||||
);
|
||||
|
||||
// Verify all tokens are not cached (i.e. have the same lifetime)
|
||||
const uniqueLifetimes = new Set(tokenDetails.map(detail => detail.lifetime));
|
||||
assert.equal(
|
||||
uniqueLifetimes.size,
|
||||
1,
|
||||
`Expected all tokens to have the same lifetime, but found ${uniqueLifetimes.size} different lifetimes: ${[uniqueLifetimes].join(', ')} seconds`
|
||||
);
|
||||
|
||||
// Verify that all tokens have different uti (unique token identifier)
|
||||
const uniqueUti = new Set(tokenDetails.map(detail => detail.uti));
|
||||
assert.equal(
|
||||
uniqueUti.size,
|
||||
reAuthSpy.callCount,
|
||||
`Expected all tokens to have different uti, but found ${uniqueUti.size} different uti in: ${[uniqueUti].join(', ')}`
|
||||
);
|
||||
};
|
||||
|
||||
const runAuthenticationTest = async (setupCredentialsProvider: () => any) => {
|
||||
const { client, reAuthSpy } = await setupTestClient(setupCredentialsProvider());
|
||||
|
||||
try {
|
||||
await client.connect();
|
||||
await runClientOperations(client);
|
||||
validateTokens(reAuthSpy);
|
||||
} finally {
|
||||
await client.destroy();
|
||||
}
|
||||
};
|
||||
|
||||
});
|
||||
|
||||
function getCertificate(certBase64) {
|
||||
try {
|
||||
const decodedCert = Buffer.from(certBase64, 'base64');
|
||||
const cert = new crypto.X509Certificate(decodedCert);
|
||||
return cert;
|
||||
} catch (error) {
|
||||
console.error('Error parsing certificate:', error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
function getCertificateThumbprint(certBase64) {
|
||||
const cert = getCertificate(certBase64);
|
||||
return cert.fingerprint.replace(/:/g, '');
|
||||
}
|
||||
|
||||
function convertCertsForMSAL(certBase64, privateKeyBase64) {
|
||||
const thumbprint = getCertificateThumbprint(certBase64);
|
||||
|
||||
const privateKeyPEM = `-----BEGIN PRIVATE KEY-----\n${privateKeyBase64}\n-----END PRIVATE KEY-----`;
|
||||
|
||||
return {
|
||||
thumbprint: thumbprint,
|
||||
privateKey: privateKeyPEM,
|
||||
x5c: certBase64
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user