From 2622bc8e0d0aa5cb87e8a65b54bb6b051f0110d6 Mon Sep 17 00:00:00 2001
From: Jonathan Fung <jonathan@dydx.exchange>
Date: Wed, 27 Mar 2024 15:13:32 -0400
Subject: [PATCH] fix tests

---
 indexer/services/vulcan/__tests__/lib/on-message.test.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/indexer/services/vulcan/__tests__/lib/on-message.test.ts b/indexer/services/vulcan/__tests__/lib/on-message.test.ts
index 27167a7143..5c9e83374b 100644
--- a/indexer/services/vulcan/__tests__/lib/on-message.test.ts
+++ b/indexer/services/vulcan/__tests__/lib/on-message.test.ts
@@ -1,7 +1,7 @@
 import { logger, stats } from '@dydxprotocol-indexer/base';
 import { createKafkaMessage, KafkaTopics } from '@dydxprotocol-indexer/kafka';
 import { OffChainUpdateV1 } from '@dydxprotocol-indexer/v4-protos';
-import { KafkaMessage } from 'kafkajs';
+import { IHeaders, KafkaMessage } from 'kafkajs';
 import { onMessage } from '../../src/lib/on-message';
 import { OrderPlaceHandler } from '../../src/handlers/order-place-handler';
 import { OrderRemoveHandler } from '../../src/handlers/order-remove-handler';
@@ -64,7 +64,7 @@ describe('onMessage', () => {
     await onMessage(message);
 
     expect(handler).toHaveBeenCalledTimes(1);
-    expect(handleUpdateMock).toHaveBeenCalledWith(update);
+    expect(handleUpdateMock).toHaveBeenCalledWith(update, message.headers ?? {});
     expect(handleUpdateMock).toHaveBeenCalledTimes(1);
 
     expect(stats.increment).toHaveBeenCalledWith('vulcan.received_kafka_message', 1);