@@ -8,11 +8,13 @@ import { WebSocketServer } from 'ws';
88describe ( 'events' , ( ) => {
99 let wss : WebSocketServer = null ;
1010 const config = Configuration . getGlobalConfig ( ) ;
11- const events = new PlatformEventManager ( config ) ;
12- config . useEventManager ( events ) ;
11+ let events : PlatformEventManager = null ;
1312
1413 beforeEach ( ( ) => {
1514 wss = new WebSocketServer ( { port : 9099 } ) ;
15+ events = new PlatformEventManager ( config ) ;
16+ config . useEventManager ( events ) ;
17+
1618 vitest . useFakeTimers ( ) ;
1719 process . env [ ACTOR_ENV_VARS . EVENTS_WEBSOCKET_URL ] = 'ws://localhost:9099/someRunId' ;
1820 process . env [ APIFY_ENV_VARS . TOKEN ] = 'dummy' ;
@@ -112,11 +114,13 @@ describe('events', () => {
112114
113115 test ( 'should send persist state events in regular interval' , async ( ) => {
114116 const eventsReceived = [ ] ;
117+ const interval = config . get ( 'persistStateIntervalMillis' )
118+
115119 events . on ( EventType . PERSIST_STATE , ( data ) => eventsReceived . push ( data ) ) ;
116120 await events . init ( ) ;
117- vitest . advanceTimersByTime ( 60001 ) ;
118- vitest . advanceTimersByTime ( 60001 ) ;
119- vitest . advanceTimersByTime ( 60001 ) ;
121+ await vitest . advanceTimersByTimeAsync ( 1.1 * interval ) ;
122+ await vitest . advanceTimersByTimeAsync ( 1.1 * interval ) ;
123+ await vitest . advanceTimersByTimeAsync ( 1.1 * interval ) ;
120124 await events . close ( ) ;
121125 expect ( eventsReceived . length ) . toBe ( 5 ) ;
122126 } ) ;
0 commit comments