fix: Flakey test (#17649)

PostHog · Sep 28, 2023 · 3675bf4 · 3675bf4
1 parent 3e13484
commit 3675bf4
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/...rver/tests/main/ingestion-queues/session-recording/session-recordings-consumer-v2.test.ts b/...rver/tests/main/ingestion-queues/session-recording/session-recordings-consumer-v2.test.ts
@@ -496,9 +496,6 @@ describe('ingester', () => {
 
         it("flushes and commits as it's revoked", async () => {
             await ingester.handleEachBatch([createMessage('sid1'), createMessage('sid2'), createMessage('sid3', 2)])
-            const revokePromise = ingester.onRevokePartitions([createTP(1)])
-
-            expect(Object.keys(ingester.sessions)).toEqual([`${team.id}-sid3`])
 
             expect(readdirSync(config.SESSION_RECORDING_LOCAL_DIRECTORY + '/session-buffer-files')).toEqual([
                 expect.stringContaining(`${team.id}.sid1.`), // gz
@@ -509,6 +506,10 @@ describe('ingester', () => {
                 expect.stringContaining(`${team.id}.sid3.`), // json
             ])
 
+            const revokePromise = ingester.onRevokePartitions([createTP(1)])
+
+            expect(Object.keys(ingester.sessions)).toEqual([`${team.id}-sid3`])
+
             await revokePromise
 
             // Only files left on the system should be the sid3 ones