diff --git a/static/app/utils/replayCount/useReplayCount.spec.tsx b/static/app/utils/replayCount/useReplayCount.spec.tsx index 039c6d3824945f..0d70696a650d24 100644 --- a/static/app/utils/replayCount/useReplayCount.spec.tsx +++ b/static/app/utils/replayCount/useReplayCount.spec.tsx @@ -49,7 +49,7 @@ describe('useReplayCount', () => { ); }); - expect(result.current.getOne('1111')).toBe(5); + await waitFor(() => expect(result.current.getOne('1111')).toBe(5)); expect(result.current.getOne('2222')).toBe(7); expect(result.current.getOne('3333')).toBe(0); expect(result.current.hasOne('1111')).toBeTruthy(); @@ -82,7 +82,7 @@ describe('useReplayCount', () => { ); }); - expect(result.current.getOne('1111')).toBe(0); + await waitFor(() => expect(result.current.getOne('1111')).toBe(0)); expect(result.current.getOne('2222')).toBe(7); expect(result.current.hasOne('1111')).toBeFalsy(); expect(result.current.hasOne('2222')).toBeTruthy(); @@ -115,11 +115,13 @@ describe('useReplayCount', () => { ); }); - expect(result.current.getMany(['1111', '2222', '3333'])).toStrictEqual({ - '1111': 5, - '2222': 7, - '3333': 0, - }); + await waitFor(() => + expect(result.current.getMany(['1111', '2222', '3333'])).toStrictEqual({ + '1111': 5, + '2222': 7, + '3333': 0, + }) + ); expect(result.current.hasMany(['1111', '2222', '3333'])).toStrictEqual({ '1111': true, '2222': true, @@ -150,10 +152,12 @@ describe('useReplayCount', () => { ); }); - expect(result.current.getMany(['1111', '2222'])).toStrictEqual({ - '1111': 0, - '2222': 7, - }); + await waitFor(() => + expect(result.current.getMany(['1111', '2222'])).toStrictEqual({ + '1111': 0, + '2222': 7, + }) + ); expect(result.current.hasMany(['1111', '2222'])).toStrictEqual({ '1111': false, '2222': true,