Fix mock server to handle multiple Redis commands in single packet

Author: bobymicroby

src/app.test.ts

@@ -24,35 +24,53 @@ describe('Redis Proxy API', () => {
           const command = data.toString();
           console.log('Mock Redis received:', command.replace(/\r\n/g, '\\r\\n'));
 
+          // Count how many Redis commands are in this data packet
+          // Each command starts with * followed by number of arguments
+          const commandCount = (command.match(/\*\d+\r\n/g) || []).length;
+          console.log('Command count:', commandCount);
+
+          // Send appropriate responses based on command content
+          let responses = '';
+
           // Handle HELLO command (Redis 6+ handshake)
           if (command.includes('HELLO')) {
-            // Return Redis server info for HELLO command
-            socket.write('*7\r\n$6\r\nserver\r\n$5\r\nredis\r\n$7\r\nversion\r\n$5\r\n7.2.0\r\n$5\r\nproto\r\n:3\r\n$2\r\nid\r\n:1\r\n');
+            responses += '*7\r\n$6\r\nserver\r\n$5\r\nredis\r\n$7\r\nversion\r\n$5\r\n7.2.0\r\n$5\r\nproto\r\n:3\r\n$2\r\nid\r\n:1\r\n';
           }
-          // Handle AUTH command
-          else if (command.includes('AUTH')) {
-            socket.write('+OK\r\n');
+          
+          // Handle CLIENT SETINFO commands (send OK for each)
+          if (command.includes('CLIENT')) {
+            const clientCommands = (command.match(/\*4\r\n\$6\r\nCLIENT\r\n/g) || []).length;
+            for (let i = 0; i < clientCommands; i++) {
+              responses += '+OK\r\n';
+            }
           }
-          // Handle PING command
-          else if (command.includes('PING')) {
-            socket.write('+PONG\r\n');
+          
+          // Handle other commands
+          if (command.includes('AUTH') && !command.includes('CLIENT')) {
+            responses += '+OK\r\n';
           }
-          // Handle FOO command (our test command)
-          else if (command.includes('FOO')) {
-            socket.write('+BAR\r\n');
+          if (command.includes('PING') && !command.includes('CLIENT')) {
+            responses += '+PONG\r\n';
           }
-          // Handle SELECT command (database selection)
-          else if (command.includes('SELECT')) {
-            socket.write('+OK\r\n');
+          if (command.includes('FOO')) {
+            responses += '+BAR\r\n';
           }
-          // Handle INFO command
-          else if (command.includes('INFO')) {
-            socket.write('$23\r\n# Server\r\nredis_version:7.2.0\r\n');
+          if (command.includes('SELECT') && !command.includes('CLIENT')) {
+            responses += '+OK\r\n';
           }
-          // Default response for any other command
-          else {
-            socket.write('+OK\r\n');
+          if (command.includes('INFO') && !command.includes('CLIENT')) {
+            responses += '$23\r\n# Server\r\nredis_version:7.2.0\r\n';
           }
+
+          // If no specific responses were generated, send OK for each command
+          if (!responses) {
+            for (let i = 0; i < commandCount; i++) {
+              responses += '+OK\r\n';
+            }
+          }
+
+          console.log('Sending responses:', responses.replace(/\r\n/g, '\\r\\n'));
+          socket.write(responses);
         },
         open(socket) {
           console.log('Mock Redis TCP connection opened');