Fixes

PostHog · Dec 23, 2024 · 235807b · 235807b
1 parent f97ae83
commit 235807b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/plugin-server/tests/e2e.timeout.test.ts b/plugin-server/tests/e2e.timeout.test.ts
@@ -54,7 +54,7 @@ describe('e2e ingestion timeout', () => {
         expect((await hub.db.fetchEvents()).length).toBe(0)
         const uuid = new UUIDT().toString()
         await posthog.capture('custom event', { name: 'haha', uuid, randomProperty: 'lololo' })
-        await delayUntilEventIngested(() => hub.db.fetchEvents())
+        await delayUntilEventIngested(() => hub.db.fetchEvents(), { maxDelayMs: 50000 })
 
         await hub.kafkaProducer.flush()
         const events = await hub.db.fetchEvents()

diff --git a/plugin-server/tests/helpers/clickhouse.ts b/plugin-server/tests/helpers/clickhouse.ts
@@ -34,10 +34,13 @@ export async function resetTestDatabaseClickhouse(extraServerConfig?: Partial<Pl
 
 export async function delayUntilEventIngested<T extends any[] | number>(
     fetchData: () => T | Promise<T>,
-    minLength = 1,
-    delayMs = 100,
-    maxDelayMs = 5000
+    options: {
+        minLength?: number
+        delayMs?: number
+        maxDelayMs?: number
+    } = {}
 ): Promise<T> {
+    const { minLength = 1, delayMs = 100, maxDelayMs = 5000 } = options
     const timer = performance.now()
     let data: T
     let dataLength = 0