@@ -16,7 +16,12 @@ describe("SocketWrapper", () => {
1616 expect ( socket ) . toBeTruthy ( ) ;
1717 } ) ;
1818
19- it ( "can connect" , ( done ) => {
19+ it ( "can connect" , async ( ) => {
20+ let done ! : ( ( ) => void ) & { fail : ( e ?: any ) => void } ;
21+ const ret = new Promise < void > ( ( resolve , reject ) => {
22+ done = resolve as typeof done ;
23+ done . fail = reject ;
24+ } ) ;
2025 pendingWithoutSocketServer ( ) ;
2126
2227 const socket = new SocketWrapper (
@@ -34,9 +39,16 @@ describe("SocketWrapper", () => {
3439 ) ;
3540 expect ( socket ) . toBeTruthy ( ) ;
3641 socket . connect ( ) ;
42+
43+ return ret ;
3744 } ) ;
3845
39- it ( "fails to connect to non-existing server" , ( done ) => {
46+ it ( "fails to connect to non-existing server" , async ( ) => {
47+ let done ! : ( ( ) => void ) & { fail : ( e ?: any ) => void } ;
48+ const ret = new Promise < void > ( ( resolve , reject ) => {
49+ done = resolve as typeof done ;
50+ done . fail = reject ;
51+ } ) ;
4052 pendingWithoutSocketServer ( ) ;
4153
4254 const socket = new SocketWrapper (
@@ -57,9 +69,16 @@ describe("SocketWrapper", () => {
5769 ) ;
5870 expect ( socket ) . toBeTruthy ( ) ;
5971 socket . connect ( ) ;
72+
73+ return ret ;
6074 } ) ;
6175
62- it ( "fails to connect to non-existing server but timeout is not triggered" , ( done ) => {
76+ it ( "fails to connect to non-existing server but timeout is not triggered" , async ( ) => {
77+ let done ! : ( ( ) => void ) & { fail : ( e ?: any ) => void } ;
78+ const ret = new Promise < void > ( ( resolve , reject ) => {
79+ done = resolve as typeof done ;
80+ done . fail = reject ;
81+ } ) ;
6382 pendingWithoutSocketServer ( ) ;
6483 const timeout = 1200 ; // ms
6584
@@ -82,9 +101,16 @@ describe("SocketWrapper", () => {
82101 ) ;
83102 expect ( socket ) . toBeTruthy ( ) ;
84103 socket . connect ( ) ;
104+
105+ return ret ;
85106 } ) ;
86107
87- it ( "can connect to slow server" , ( done ) => {
108+ it ( "can connect to slow server" , async ( ) => {
109+ let done ! : ( ( ) => void ) & { fail : ( e ?: any ) => void } ;
110+ const ret = new Promise < void > ( ( resolve , reject ) => {
111+ done = resolve as typeof done ;
112+ done . fail = reject ;
113+ } ) ;
88114 pendingWithoutSocketServer ( ) ;
89115
90116 const socket = new SocketWrapper (
@@ -102,6 +128,8 @@ describe("SocketWrapper", () => {
102128 ) ;
103129 expect ( socket ) . toBeTruthy ( ) ;
104130 socket . connect ( ) ;
131+
132+ return ret ;
105133 } ) ;
106134
107135 it ( "times out when establishing connection takes too long" , async ( ) => {
@@ -134,7 +162,12 @@ describe("SocketWrapper", () => {
134162 } ) ;
135163 } ) ;
136164
137- it ( "can connect and disconnect" , ( done ) => {
165+ it ( "can connect and disconnect" , async ( ) => {
166+ let done ! : ( ( ) => void ) & { fail : ( e ?: any ) => void } ;
167+ const ret = new Promise < void > ( ( resolve , reject ) => {
168+ done = resolve as typeof done ;
169+ done . fail = reject ;
170+ } ) ;
138171 pendingWithoutSocketServer ( ) ;
139172
140173 let opened = 0 ;
@@ -160,9 +193,16 @@ describe("SocketWrapper", () => {
160193 } ,
161194 ) ;
162195 socket . connect ( ) ;
196+
197+ return ret ;
163198 } ) ;
164199
165- it ( "can disconnect before waiting for open" , ( done ) => {
200+ it ( "can disconnect before waiting for open" , async ( ) => {
201+ let done ! : ( ( ) => void ) & { fail : ( e ?: any ) => void } ;
202+ const ret = new Promise < void > ( ( resolve , reject ) => {
203+ done = resolve as typeof done ;
204+ done . fail = reject ;
205+ } ) ;
166206 pendingWithoutSocketServer ( ) ;
167207
168208 const socket = new SocketWrapper (
@@ -184,9 +224,16 @@ describe("SocketWrapper", () => {
184224 ) ;
185225 socket . connect ( ) ;
186226 socket . disconnect ( ) ;
227+
228+ return ret ;
187229 } ) ;
188230
189- it ( "can disconnect before waiting for open and timeout will not be triggered" , ( done ) => {
231+ it ( "can disconnect before waiting for open and timeout will not be triggered" , async ( ) => {
232+ let done ! : ( ( ) => void ) & { fail : ( e ?: any ) => void } ;
233+ const ret = new Promise < void > ( ( resolve , reject ) => {
234+ done = resolve as typeof done ;
235+ done . fail = reject ;
236+ } ) ;
190237 pendingWithoutSocketServer ( ) ;
191238 const timeout = 500 ; // ms
192239
@@ -212,9 +259,16 @@ describe("SocketWrapper", () => {
212259 ) ;
213260 socket . connect ( ) ;
214261 socket . disconnect ( ) ;
262+
263+ return ret ;
215264 } ) ;
216265
217- it ( "can send events when connected" , ( done ) => {
266+ it ( "can send events when connected" , async ( ) => {
267+ let done ! : ( ( ) => void ) & { fail : ( e ?: any ) => void } ;
268+ const ret = new Promise < void > ( ( resolve , reject ) => {
269+ done = resolve as typeof done ;
270+ done . fail = reject ;
271+ } ) ;
218272 pendingWithoutSocketServer ( ) ;
219273
220274 const responseMessages = new Array < string > ( ) ;
@@ -243,9 +297,16 @@ describe("SocketWrapper", () => {
243297 } ,
244298 ) ;
245299 socket . connect ( ) ;
300+
301+ return ret ;
246302 } ) ;
247303
248- it ( "can send events after timeout period" , ( done ) => {
304+ it ( "can send events after timeout period" , async ( ) => {
305+ let done ! : ( ( ) => void ) & { fail : ( e ?: any ) => void } ;
306+ const ret = new Promise < void > ( ( resolve , reject ) => {
307+ done = resolve as typeof done ;
308+ done . fail = reject ;
309+ } ) ;
249310 pendingWithoutSocketServer ( ) ;
250311
251312 // The "timeout period" is the period in which a timeout could potentially be triggered
@@ -271,9 +332,16 @@ describe("SocketWrapper", () => {
271332 socket . connect ( ) ;
272333
273334 setTimeout ( ( ) => socket . send ( "Hello world" ) , 2 * timeoutPeriodLength ) ;
335+
336+ return ret ;
274337 } ) ;
275338
276- it ( "cannot send on a disconnect socket (it will never come back)" , ( done ) => {
339+ it ( "cannot send on a disconnect socket (it will never come back)" , async ( ) => {
340+ let done ! : ( ( ) => void ) & { fail : ( e ?: any ) => void } ;
341+ const ret = new Promise < void > ( ( resolve , reject ) => {
342+ done = resolve as typeof done ;
343+ done . fail = reject ;
344+ } ) ;
277345 pendingWithoutSocketServer ( ) ;
278346
279347 const socket = new SocketWrapper (
@@ -300,5 +368,7 @@ describe("SocketWrapper", () => {
300368 } ,
301369 ) ;
302370 socket . connect ( ) ;
371+
372+ return ret ;
303373 } ) ;
304374} ) ;
0 commit comments