@@ -1101,11 +1101,9 @@ describe('WebSocket', () => {
1101
1101
} ) ;
1102
1102
1103
1103
it ( 'emits an error if the redirect URL is invalid (1/2)' , ( done ) => {
1104
- const onUpgrade = ( req , socket ) => {
1104
+ server . once ( 'upgrade' , ( req , socket ) => {
1105
1105
socket . end ( 'HTTP/1.1 302 Found\r\nLocation: ws://\r\n\r\n' ) ;
1106
- } ;
1107
-
1108
- server . on ( 'upgrade' , onUpgrade ) ;
1106
+ } ) ;
1109
1107
1110
1108
const ws = new WebSocket ( `ws://localhost:${ server . address ( ) . port } ` , {
1111
1109
followRedirects : true
@@ -1117,17 +1115,14 @@ describe('WebSocket', () => {
1117
1115
assert . strictEqual ( err . message , 'Invalid URL: ws://' ) ;
1118
1116
assert . strictEqual ( ws . _redirects , 1 ) ;
1119
1117
1120
- server . removeListener ( 'upgrade' , onUpgrade ) ;
1121
1118
ws . on ( 'close' , ( ) => done ( ) ) ;
1122
1119
} ) ;
1123
1120
} ) ;
1124
1121
1125
1122
it ( 'emits an error if the redirect URL is invalid (2/2)' , ( done ) => {
1126
- const onUpgrade = ( req , socket ) => {
1123
+ server . once ( 'upgrade' , ( req , socket ) => {
1127
1124
socket . end ( 'HTTP/1.1 302 Found\r\nLocation: http://localhost\r\n\r\n' ) ;
1128
- } ;
1129
-
1130
- server . on ( 'upgrade' , onUpgrade ) ;
1125
+ } ) ;
1131
1126
1132
1127
const ws = new WebSocket ( `ws://localhost:${ server . address ( ) . port } ` , {
1133
1128
followRedirects : true
@@ -1142,7 +1137,6 @@ describe('WebSocket', () => {
1142
1137
) ;
1143
1138
assert . strictEqual ( ws . _redirects , 1 ) ;
1144
1139
1145
- server . removeListener ( 'upgrade' , onUpgrade ) ;
1146
1140
ws . on ( 'close' , ( ) => done ( ) ) ;
1147
1141
} ) ;
1148
1142
} ) ;
0 commit comments