fix(time-series): honour typeMapping for labels in RESP2 replies (#3412)

Author: abhijeet117

packages/time-series/lib/commands/MGET_WITHLABELS.spec.ts

@@ -2,6 +2,7 @@ import { strict as assert } from 'node:assert';
 import testUtils, { GLOBAL } from '../test-utils';
 import MGET_WITHLABELS from './MGET_WITHLABELS';
 import { parseArgs } from '@redis/client/lib/commands/generic-transformers';
+import { RESP_TYPES } from '@redis/client';
 
 describe('TS.MGET_WITHLABELS', () => {
   it('transformArguments', () => {
@@ -11,6 +12,19 @@ describe('TS.MGET_WITHLABELS', () => {
     );
   });
 
+  it('transformReply maps labels with the MAP type mapping', () => {
+    const reply = MGET_WITHLABELS.transformReply[2](
+      [['key', [['label', 'value']], [0, '1']]] as never,
+      undefined,
+      { [RESP_TYPES.MAP]: Map }
+    ) as unknown as Map<string, { labels: unknown }>;
+
+    assert.deepEqual(
+      reply.get('key')?.labels,
+      new Map([['label', 'value']])
+    );
+  });
+
   testUtils.testWithClient('client.ts.mGetWithLabels', async client => {
     const [, reply] = await Promise.all([
       client.ts.add('key', 0, 0, {

packages/time-series/lib/commands/MGET_WITHLABELS.ts

@@ -34,7 +34,7 @@ export function createTransformMGetLabelsReply<T extends RawLabelValue>() {
     2(reply: MGetLabelsRawReply2<T>, _, typeMapping?: TypeMapping) {
       return resp2MapToValue(reply, ([, labels, sample]) => {
         return {
-          labels: transformRESP2Labels(labels),
+          labels: transformRESP2Labels(labels, typeMapping),
           sample: transformSampleReply[2](sample)
         };
       }, typeMapping);

packages/time-series/lib/commands/MRANGE_WITHLABELS_GROUPBY.spec.ts

@@ -4,6 +4,7 @@ import MRANGE_WITHLABELS_GROUPBY from './MRANGE_WITHLABELS_GROUPBY';
 import { TIME_SERIES_REDUCERS } from './MRANGE_GROUPBY';
 import { TIME_SERIES_AGGREGATION_TYPE } from './CREATERULE';
 import { parseArgs } from '@redis/client/lib/commands/generic-transformers';
+import { RESP_TYPES } from '@redis/client';
 
 describe('TS.MRANGE_WITHLABELS_GROUPBY', () => {
   it('transformArguments', () => {
@@ -40,6 +41,24 @@ describe('TS.MRANGE_WITHLABELS_GROUPBY', () => {
     );
   });
 
+  it('transformReply maps labels with the MAP type mapping', () => {
+    const reply = MRANGE_WITHLABELS_GROUPBY.transformReply[2](
+      [[
+        'key',
+        [['label', 'value'], ['__reducer__', 'avg'], ['__source__', 'source']],
+        [[0, '1']]
+      ]] as never,
+      undefined,
+      { [RESP_TYPES.MAP]: Map }
+    ) as unknown as Map<string, { labels: unknown, sources: Array<string> }>;
+
+    assert.deepEqual(
+      reply.get('key')?.labels,
+      new Map([['label', 'value']])
+    );
+    assert.deepEqual(reply.get('key')?.sources, ['source']);
+  });
+
   testUtils.testWithClient('client.ts.mRangeWithLabelsGroupBy', async client => {
     const [, reply] = await Promise.all([
       client.ts.add('key', 0, 0, {

packages/time-series/lib/commands/MRANGE_WITHLABELS_GROUPBY.ts

@@ -57,7 +57,7 @@ export default {
   transformReply: {
     2(reply: TsMRangeWithLabelsGroupByRawReply2, _?: unknown, typeMapping?: TypeMapping) {
       return resp2MapToValue(reply, ([_key, labels, samples]) => {
-        const transformed = transformRESP2LabelsWithSources(labels);
+        const transformed = transformRESP2LabelsWithSources(labels, typeMapping);
         return {
           labels: transformed.labels,
           sources: transformed.sources,