@@ -241,32 +241,33 @@ describe('HLFTxEventHandler', () => {
241241 } ) ;
242242
243243 describe ( '#waitForEvents' , ( ) => {
244- it ( 'Should do nothing if no events hubs' , ( ) => {
244+ it ( 'Should do nothing if no events hubs' , async ( ) => {
245245 let evHandler = new HLFTxEventHandler ( null , null , null ) ;
246- evHandler . waitForEvents ( ) . should . eventually . be . resolved ;
246+ await evHandler . waitForEvents ( ) . should . be . resolved ;
247247 sinon . assert . calledOnce ( logWarnSpy ) ;
248248 evHandler = new HLFTxEventHandler ( [ ] , '1234' , 100 ) ;
249- evHandler . waitForEvents ( ) . should . eventually . be . resolved ;
249+ await evHandler . waitForEvents ( ) . should . be . resolved ;
250250 sinon . assert . calledTwice ( logWarnSpy ) ;
251251 } ) ;
252252
253- it ( 'Should do wait for 1 event' , ( ) => {
253+ it ( 'Should do wait for 1 event' , async ( ) => {
254254 sandbox . stub ( global , 'setTimeout' ) ;
255255 sandbox . stub ( HLFUtil , 'eventHubConnected' ) . withArgs ( eventhub1 ) . returns ( true ) ;
256256 let evHandler = new HLFTxEventHandler ( [ eventhub1 ] , '1234' , 31 ) ;
257257 evHandler . startListening ( ) ;
258258 evHandler . listenerPromises [ 0 ] . should . be . instanceOf ( Promise ) ;
259259 evHandler . listenerPromises [ 0 ] = Promise . resolve ( ) ;
260- evHandler . waitForEvents ( ) . should . eventually . be . resolved ;
260+ evHandler . responseCount ++ ;
261+ await evHandler . waitForEvents ( ) . should . be . resolved ;
261262
262263 evHandler = new HLFTxEventHandler ( [ eventhub1 ] , '1234' , 31 ) ;
263264 evHandler . startListening ( ) ;
264265 evHandler . listenerPromises [ 0 ] . should . be . instanceOf ( Promise ) ;
265266 evHandler . listenerPromises [ 0 ] = Promise . reject ( new Error ( 'some error' ) ) ;
266- evHandler . waitForEvents ( ) . should . eventually . be . rejectedWith ( / s o m e e r r o r / ) ;
267+ await evHandler . waitForEvents ( ) . should . be . rejectedWith ( / s o m e e r r o r / ) ;
267268 } ) ;
268269
269- it ( 'Should do wait more than 1 event' , ( ) => {
270+ it ( 'Should do wait more than 1 event' , async ( ) => {
270271 sandbox . stub ( global , 'setTimeout' ) ;
271272 const ehc = sandbox . stub ( HLFUtil , 'eventHubConnected' ) ;
272273 ehc . withArgs ( eventhub1 ) . returns ( true ) ;
@@ -275,9 +276,10 @@ describe('HLFTxEventHandler', () => {
275276 evHandler . startListening ( ) ;
276277 evHandler . listenerPromises [ 0 ] . should . be . instanceOf ( Promise ) ;
277278 evHandler . listenerPromises [ 0 ] = Promise . resolve ( ) ;
279+ evHandler . responseCount ++ ;
278280 evHandler . listenerPromises [ 1 ] . should . be . instanceOf ( Promise ) ;
279281 evHandler . listenerPromises [ 1 ] = Promise . reject ( new Error ( 'some error' ) ) ;
280- evHandler . waitForEvents ( ) . should . eventually . be . rejectedWith ( / s o m e e r r o r / ) ;
282+ await evHandler . waitForEvents ( ) . should . be . rejectedWith ( / s o m e e r r o r / ) ;
281283 } ) ;
282284
283285 it ( 'Should handle timeout for an event' , ( ) => {
@@ -287,7 +289,7 @@ describe('HLFTxEventHandler', () => {
287289 ehc . withArgs ( eventhub2 ) . returns ( true ) ;
288290 let evHandler = new HLFTxEventHandler ( [ eventhub1 , eventhub2 ] , '1234' , 31 ) ;
289291 evHandler . startListening ( ) ;
290- evHandler . waitForEvents ( ) . should . eventually . be . rejectedWith ( / c o m m i t n o t i f i c a t i o n / )
292+ return evHandler . waitForEvents ( ) . should . eventually . be . rejectedWith ( / c o m m i t n o t i f i c a t i o n / )
291293 . then ( ( ) => {
292294 sinon . assert . calledOnce ( eventhub1 . unregisterTxEvent ) ;
293295 sinon . assert . calledWith ( eventhub1 . unregisterTxEvent , '1234' ) ;
@@ -308,7 +310,7 @@ describe('HLFTxEventHandler', () => {
308310 ehc . withArgs ( eventhub2 ) . returns ( true ) ;
309311 let evHandler = new HLFTxEventHandler ( [ eventhub1 , eventhub2 ] , '1234' , 31 ) ;
310312 evHandler . startListening ( ) ;
311- evHandler . waitForEvents ( )
313+ return evHandler . waitForEvents ( )
312314 . then ( ( ) => {
313315 sinon . assert . calledOnce ( eventhub1 . unregisterTxEvent ) ;
314316 sinon . assert . calledWith ( eventhub1 . unregisterTxEvent , '1234' ) ;
@@ -332,7 +334,7 @@ describe('HLFTxEventHandler', () => {
332334 ehc . withArgs ( eventhub2 ) . returns ( true ) ;
333335 let evHandler = new HLFTxEventHandler ( [ eventhub1 , eventhub2 ] , '1234' , 31 ) ;
334336 evHandler . startListening ( ) ;
335- evHandler . waitForEvents ( ) . should . eventually . be . rejectedWith ( / r e j e c t e d t r a n s a c t i o n / )
337+ return evHandler . waitForEvents ( ) . should . eventually . be . rejectedWith ( / r e j e c t e d t r a n s a c t i o n / )
336338 . then ( ( ) => {
337339 sinon . assert . calledOnce ( eventhub1 . unregisterTxEvent ) ;
338340 sinon . assert . calledWith ( eventhub1 . unregisterTxEvent , '1234' ) ;
0 commit comments