Fix

PostHog · Dec 25, 2024 · dc27719 · dc27719
1 parent f97321c
commit dc27719
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/plugin-server/tests/cdp/cdp-processed-events-consumer.test.ts b/plugin-server/tests/cdp/cdp-processed-events-consumer.test.ts
@@ -332,7 +332,7 @@ describe('CDP Processed Events Consumer', () => {
                 const invocations = await processor.processBatch([globals])
 
                 expect(invocations).toHaveLength(0)
-                expect(mockProducer.produce).toHaveBeenCalledTimes(2)
+                expect(mockProducer.queueMessages).toHaveBeenCalledTimes(1)
 
                 expect(decodeAllKafkaMessages()).toMatchObject([
                     {

diff --git a/...erver/tests/main/ingestion-queues/session-recording/services/console-log-ingester.test.ts b/...erver/tests/main/ingestion-queues/session-recording/services/console-log-ingester.test.ts
@@ -1,6 +1,6 @@
 import { HighLevelProducer } from 'node-rdkafka'
 
-import { produce } from '../../../../../src/kafka/producer'
+import { KafkaProducerWrapper, produce } from '../../../../../src/kafka/producer'
 import { ConsoleLogsIngester } from '../../../../../src/main/ingestion-queues/session-recording/services/console-logs-ingester'
 import { OffsetHighWaterMarker } from '../../../../../src/main/ingestion-queues/session-recording/services/offset-high-water-marker'
 import { IncomingRecordingMessage } from '../../../../../src/main/ingestion-queues/session-recording/types'
@@ -43,7 +43,7 @@ describe('console log ingester', () => {
 
         const mockedHighWaterMarker = { isBelowHighWaterMark: jest.fn() } as unknown as OffsetHighWaterMarker
         consoleLogIngester = new ConsoleLogsIngester(
-            mockProducer as unknown as HighLevelProducer,
+            mockProducer as unknown as KafkaProducerWrapper,
             mockedHighWaterMarker
         )
     })