fixed unit test

Author: fboucquez

test/infrastructure/Listener.spec.ts

@@ -14,12 +14,13 @@
  * limitations under the License.
  */
 
-import {expect} from 'chai';
-import {Listener} from '../../src/infrastructure/Listener';
-import {Address} from "../../src/model/account/Address";
-import {deepEqual} from "assert";
-import {UInt64} from "../../src/model/UInt64";
-import {timeout} from "rxjs/operators";
+import { expect } from 'chai';
+import { Listener } from '../../src/infrastructure/Listener';
+import { Address } from "../../src/model/account/Address";
+import { deepEqual } from "assert";
+import { UInt64 } from "../../src/model/UInt64";
+import { timeout } from "rxjs/operators";
+import { TransactionStatusError } from "../../src/model/transaction/TransactionStatusError";
 
 describe('Listener', () => {
     it('should createComplete a WebSocket instance given url parameter', () => {
@@ -64,17 +65,25 @@ describe('Listener', () => {
 
             listener.open();
 
-            listener.status(errorAddress).pipe(timeout(2000)).subscribe((transactionStatusError) => {
+            const reportedStatus = new Array<TransactionStatusError>();
+
+            listener.status(errorAddress).subscribe((transactionStatusError) => {
+                reportedStatus.push(transactionStatusError);
+            }, err => {
+                done(err);
+            });
+
+            listener.handleMessage(statusInfoErrorDTO, null);
+
+            setTimeout(() => {
+                expect(reportedStatus.length).to.be.equal(1);
+                const transactionStatusError = reportedStatus[0];
                 expect(transactionStatusError.address).to.deep.equal(errorAddress);
                 expect(transactionStatusError.hash).to.be.equal(statusInfoErrorDTO.hash);
                 expect(transactionStatusError.status).to.be.equal(statusInfoErrorDTO.status);
                 deepEqual(transactionStatusError.deadline.toDTO(), UInt64.fromNumericString(statusInfoErrorDTO.deadline).toDTO());
                 done();
-            }, err => {
-                done('Should have not timed out!');
-            });
-
-            listener.handleMessage(statusInfoErrorDTO, null);
+            }, 100)
 
 
         });
@@ -110,15 +119,21 @@ describe('Listener', () => {
 
             listener.open();
 
-            listener.status(subscribedAddress).pipe(timeout(100)).subscribe(status => {
-                done('Should have timed out!');
+            const reportedStatus = new Array<TransactionStatusError>();
+
+            listener.status(subscribedAddress).subscribe((transactionStatusError) => {
+                reportedStatus.push(transactionStatusError);
             }, err => {
-                expect(err.name).to.be.eq('TimeoutError');
-                done();
+                done(err);
             });
 
             listener.handleMessage(statusInfoErrorDTO, null);
 
+            setTimeout(() => {
+                expect(reportedStatus.length).to.be.equal(0);
+                done();
+            }, 100)
+
 
         });
     });
@@ -127,12 +142,12 @@ describe('Listener', () => {
         it('should reject because of wrong server url', async () => {
             const listener = new Listener('https://notcorrecturl:0000');
             await listener.open()
-                .then((result) => {
-                    throw new Error('This should not be called when expecting error');
-                })
-                .catch((error) => {
-                    expect(error.message.toString()).not.to.be.equal('');
-                });
+            .then((result) => {
+                throw new Error('This should not be called when expecting error');
+            })
+            .catch((error) => {
+                expect(error.message.toString()).not.to.be.equal('');
+            });
         });
     });
 });