@@ -18,8 +18,8 @@ import generateToken from "./generateToken.js";
1818// Change these options to increase the user goal or time to run the test.
1919export const options = getTestOptions (
2020 /* User goal */ 100 ,
21- /* Ramp up time minutes */ 2 ,
22- /* High load time minutes */ 5
21+ /* Ramp up time minutes */ 1 ,
22+ /* High load time minutes */ 1
2323) ;
2424
2525// Change this to test different code
@@ -36,20 +36,28 @@ const longWebsocketSessions = new Counter("long_websocket_sessions");
3636// websocket sessions > EXTRA_LONG_REQUEST_MS
3737const extraLongWebsocketSessions = new Counter ( "extra_long_websocket_sessions" ) ;
3838
39+
40+ function isResultSuccess ( result ) {
41+ return result && result . status === 200 ;
42+ }
43+
3944export default function ( ) {
4045 const authToken = generateToken ( MiniAppType . CONSOLE ) ;
4146 const uploadResult = http . put (
4247 UPLOAD_URL + authToken ,
4348 sourceToTest ,
4449 UPLOAD_PARAMS
4550 ) ;
46- const res = ws . connect ( WEBSOCKET_URL + authToken , WEBSOCKET_PARAMS , ( socket ) =>
47- onSocketConnect ( socket , Date . now ( ) )
48- ) ;
4951
50- check ( res , { "websocket status is 101" : ( r ) => r && r . status === 101 } ) ;
52+ check ( uploadResult , { "upload status is 200" : ( r ) => isResultSuccess ( r ) } ) ;
53+
54+ if ( isResultSuccess ( uploadResult ) ) {
55+ const res = ws . connect ( WEBSOCKET_URL + authToken , WEBSOCKET_PARAMS , ( socket ) =>
56+ onSocketConnect ( socket , Date . now ( ) )
57+ ) ;
5158
52- check ( uploadResult , { "upload status is 200" : ( r ) => r && r . status === 200 } ) ;
59+ check ( res , { "websocket status is 101" : ( r ) => r && r . status === 101 } ) ;
60+ }
5361}
5462
5563function onSocketConnect ( socket , startTime ) {
@@ -58,6 +66,7 @@ function onSocketConnect(socket, startTime) {
5866 socket . on ( "message" , function ( data ) {
5967 const parsedData = JSON . parse ( data ) ;
6068 if ( parsedData . type === "EXCEPTION" ) {
69+ console . log ( "[EXCEPTION] " + parsedData . value ) ;
6170 exceptionCounter . add ( 1 ) ;
6271 }
6372 } ) ;
@@ -78,6 +87,7 @@ function onSocketConnect(socket, startTime) {
7887 } ) ;
7988
8089 socket . on ( "error" , function ( e ) {
90+ console . log ( "[ERROR] " + e . error ( ) ) ;
8191 errorCounter . add ( 1 ) ;
8292 } ) ;
8393}
0 commit comments