Fix issue with message-forwarder tests (#2002)

dydxprotocol · Jul 31, 2024 · 740c181 · 740c181
1 parent 56364db
commit 740c181
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/indexer/services/socks/__tests__/lib/message-forwarder.test.ts b/indexer/services/socks/__tests__/lib/message-forwarder.test.ts
@@ -571,7 +571,7 @@ describe('message-forwarder', () => {
         connectionId = msg.connection_id;
       }
       if (msg.message_id === 1) {
-        // Check that the initial message is correct.
+        // Check that the initial message is a Subscribe Message
         checkInitialMessage(
           msg as SubscribedMessage,
           connectionId,

diff --git a/indexer/services/socks/src/helpers/from-kafka-helpers.ts b/indexer/services/socks/src/helpers/from-kafka-helpers.ts
@@ -100,6 +100,18 @@ export function getMessageToForward(
         version: subaccountMessage.version,
       };
     }
+    case Channel.V4_BLOCK_HEIGHT: {
+      const blockHeightMessage: BlockHeightMessage = BlockHeightMessage.decode(messageBinary);
+      return {
+        channel: Channel.V4_BLOCK_HEIGHT,
+        id: V4_BLOCK_HEIGHT_ID,
+        version: blockHeightMessage.version,
+        contents: {
+          blockHeight: blockHeightMessage.blockHeight,
+          time: blockHeightMessage.time,
+        },
+      };
+    }
     default:
       throw new InvalidForwardMessageError(`Unknown channel: ${channel}`);
   }