fix tests

dydxprotocol · Mar 27, 2024 · 2622bc8 · 2622bc8
1 parent 01cfc7e
commit 2622bc8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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);