@@ -23,6 +23,9 @@ export const options = getTestOptions(
2323
2424// Change this to test different code
2525const sourceToTest = helloWorld ;
26+ // Set this to true to space out requests every REQUEST_TIME_MS milliseconds. Set to
27+ // false to send as many requests as possible.
28+ const SHOULD_SLEEP = true ;
2629
2730const exceptionCounter = new Counter ( "exceptions" ) ;
2831const errorCounter = new Counter ( "errors" ) ;
@@ -50,10 +53,14 @@ export default function () {
5053 check ( uploadResult , { "upload status is 200" : ( r ) => isResultSuccess ( r ) } ) ;
5154
5255 if ( isResultSuccess ( uploadResult ) ) {
53- const res = ws . connect ( WEBSOCKET_URL + authToken , WEBSOCKET_PARAMS , ( socket ) =>
54- onSocketConnect ( socket , requestStartTime , Date . now ( ) , sessionId )
55- ) ;
56-
56+ let res = null ;
57+ try {
58+ res = ws . connect ( WEBSOCKET_URL + authToken , WEBSOCKET_PARAMS , ( socket ) =>
59+ onSocketConnect ( socket , requestStartTime , Date . now ( ) , sessionId )
60+ ) ;
61+ } catch ( error ) {
62+ console . log ( `ERROR ${ error } for session id ${ sessionId } ` ) ;
63+ }
5764 check ( res , { "websocket status is 101" : ( r ) => r && r . status === 101 } ) ;
5865 } else {
5966 console . log ( `ERROR upload failed for session id ${ sessionId } ` ) ;
@@ -99,9 +106,11 @@ function onSocketConnect(socket, requestStartTime, websocketStartTime, sessionId
99106
100107 // Sleep this VU if we are under the max request time. This is so we maintain
101108 // a reasonable number of total requests across all virtual users.
102- const sleepTime = Math . floor ( ( REQUEST_TIME_MS - totalTime ) / 1000 ) ;
103- if ( sleepTime > 0 ) {
104- sleep ( sleepTime ) ;
109+ if ( SHOULD_SLEEP ) {
110+ const sleepTime = Math . floor ( ( REQUEST_TIME_MS - totalTime ) / 1000 ) ;
111+ if ( sleepTime > 0 ) {
112+ sleep ( sleepTime ) ;
113+ }
105114 }
106115 } ) ;
107116
0 commit comments