--wip-- [skip ci]

Author: nkaradzhov

packages/test-utils/lib/redis-proxy.ts

@@ -54,11 +54,11 @@ export class RedisProxy extends EventEmitter {
   public readonly config: Required<ProxyConfig>;
   private readonly connections: Map<string, ActiveConnection>;
   private isRunning: boolean;
+  private transformations: ((data: Buffer) => Buffer)[] = [];
 
   constructor(config: ProxyConfig) {
     super();
 
-
     this.config = {
       listenHost: '127.0.0.1',
       timeout: 30000,
@@ -113,6 +113,10 @@ export class RedisProxy extends EventEmitter {
     });
   }
 
+  public setTransformations(transformations: ((data: Buffer) => Buffer)[]): void {
+    this.transformations = transformations;
+  }
+
   public getStats(): ProxyStats {
     const connections = Array.from(this.connections.values());
 
@@ -217,6 +221,11 @@ export class RedisProxy extends EventEmitter {
     });
   }
 
+  private applyTransformations(data: Buffer): Buffer {
+    if(!this.transformations.length) return data
+    return this.transformations.reduce((acc, fn) => fn(acc), data);
+  }
+
   private handleClientConnection(clientSocket: net.Socket): void {
     const connectionId = this.generateConnectionId();
     const serverSocket = net.createConnection({
@@ -249,8 +258,9 @@ export class RedisProxy extends EventEmitter {
     });
 
     serverSocket.on('data', (data) => {
-      this.emit('data', connectionId, 'server->client', data);
-      clientSocket.write(data);
+      const transformedData = this.applyTransformations(data);
+      this.emit('data', connectionId, 'server->client', transformedData);
+      clientSocket.write(transformedData);
     });
 
     clientSocket.on('close', () => {
@@ -326,4 +336,3 @@ export function getFreePortNumber(): Promise<number> {
 
 export { RedisProxy as RedisTransparentProxy };
 export type { ProxyConfig, ConnectionInfo, ProxyEvents, SendResult, DataDirection, ProxyStats };
-