@@ -8,11 +8,13 @@ import { WebSocketServer } from 'ws';
8
8
describe ( 'events' , ( ) => {
9
9
let wss : WebSocketServer = null ;
10
10
const config = Configuration . getGlobalConfig ( ) ;
11
- const events = new PlatformEventManager ( config ) ;
12
- config . useEventManager ( events ) ;
11
+ let events : PlatformEventManager | null = null ;
13
12
14
13
beforeEach ( ( ) => {
15
14
wss = new WebSocketServer ( { port : 9099 } ) ;
15
+ events = new PlatformEventManager ( config ) ;
16
+ config . useEventManager ( events ) ;
17
+
16
18
vitest . useFakeTimers ( ) ;
17
19
process . env [ ACTOR_ENV_VARS . EVENTS_WEBSOCKET_URL ] = 'ws://localhost:9099/someRunId' ;
18
20
process . env [ APIFY_ENV_VARS . TOKEN ] = 'dummy' ;
@@ -112,11 +114,13 @@ describe('events', () => {
112
114
113
115
test ( 'should send persist state events in regular interval' , async ( ) => {
114
116
const eventsReceived = [ ] ;
117
+ const interval = config . get ( 'persistStateIntervalMillis' ) ;
118
+
115
119
events . on ( EventType . PERSIST_STATE , ( data ) => eventsReceived . push ( data ) ) ;
116
120
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 ) ;
120
124
await events . close ( ) ;
121
125
expect ( eventsReceived . length ) . toBe ( 5 ) ;
122
126
} ) ;
0 commit comments