fix: tests

Author: vasil-cf

test/addNewMask.unit.spec.js

@@ -57,6 +57,11 @@ describe('addNewMask', () => {
             stubGot.post.resolves({ statusCode: 201 });
 
             const { updateMasks, exitHandler } = proxyquire('../lib/addNewMask', {
+                '@codefresh-io/cf-telemetry/init': {
+                    terminate: () => ({
+                        finally: callback => callback(),
+                    })
+                },
                 './helpers': { getServerAddress: stubGetServerAddress },
             });
             process.listeners('exit').forEach((listener) => {
@@ -77,6 +82,11 @@ describe('addNewMask', () => {
         it('should fail if the server address is not available', async () => {
             stubGetServerAddress.rejects('could not get server address');
             const { updateMasks, exitHandler } = proxyquire('../lib/addNewMask', {
+                '@codefresh-io/cf-telemetry/init': {
+                    terminate: () => ({
+                        finally: callback => callback(),
+                    })
+                },
                 '@codefresh-io/cf-telemetry/logs': {
                     Logger: function() { return stubLogger },
                 },
@@ -97,6 +107,11 @@ describe('addNewMask', () => {
         it('should fail if the server address is not valid URL', async () => {
             stubGetServerAddress.resolves('foo');
             const { updateMasks, exitHandler } = proxyquire('../lib/addNewMask', {
+                '@codefresh-io/cf-telemetry/init': {
+                    terminate: () => ({
+                        finally: callback => callback(),
+                    })
+                },
                 '@codefresh-io/cf-telemetry/logs': {
                     Logger: function() { return stubLogger },
                 },
@@ -122,6 +137,11 @@ describe('addNewMask', () => {
                 body: 'Internal Server Error',
             });
             const { updateMasks, exitHandler } = proxyquire('../lib/addNewMask', {
+                '@codefresh-io/cf-telemetry/init': {
+                    terminate: () => ({
+                        finally: callback => callback(),
+                    })
+                },
                 '@codefresh-io/cf-telemetry/logs': {
                     Logger: function() { return stubLogger },
                 },

test/isReady.unit.spec.js

@@ -22,6 +22,11 @@ describe('isReady script', () => {
             const state = JSON.stringify({ status: 'ready', containers: {} })
             process.argv = [];
             proxyquire('../lib/isReady.js', {
+                '@codefresh-io/cf-telemetry/init': {
+                    terminate: () => ({
+                        finally: callback => callback(),
+                    })
+                },
                 'fs': {
                     readFileSync: () => Buffer.from(state),
                 },
@@ -32,6 +37,11 @@ describe('isReady script', () => {
             const state = JSON.stringify({ status: 'notReady', containers: {} })
             process.argv = [];
             proxyquire('../lib/isReady.js', {
+                '@codefresh-io/cf-telemetry/init': {
+                    terminate: () => ({
+                        finally: callback => callback(),
+                    })
+                },
                 'fs': {
                     readFileSync: () => Buffer.from(state),
                 },
@@ -44,6 +54,11 @@ describe('isReady script', () => {
             const state = JSON.stringify({ status: 'ready', containers: { 'container-id': { status: ContainerHandlingStatus.LISTENING } } })
             process.argv = ['foo', 'bar', 'container-id'];
             proxyquire('../lib/isReady.js', {
+                '@codefresh-io/cf-telemetry/init': {
+                    terminate: () => ({
+                        finally: callback => callback(),
+                    })
+                },
                 'fs': {
                     readFileSync: () => Buffer.from(state),
                 },
@@ -54,6 +69,11 @@ describe('isReady script', () => {
             const state = JSON.stringify({ status: 'ready', containers: { 'container-id': { status: ContainerHandlingStatus.WAITING_FOR_START } } })
             process.argv = ['foo', 'bar', 'container-id'];
             proxyquire('../lib/isReady.js', {
+                '@codefresh-io/cf-telemetry/init': {
+                    terminate: () => ({
+                        finally: callback => callback(),
+                    })
+                },
                 'fs': {
                     readFileSync: () => Buffer.from(state),
                 },
@@ -64,6 +84,11 @@ describe('isReady script', () => {
             const state = JSON.stringify({ status: 'ready', containers: { 'container-id': { status: ContainerHandlingStatus.FINISHED } } })
             process.argv = ['foo', 'bar', 'container-id'];
             proxyquire('../lib/isReady.js', {
+                '@codefresh-io/cf-telemetry/init': {
+                    terminate: () => ({
+                        finally: callback => callback(),
+                    })
+                },
                 'fs': {
                     readFileSync: () => Buffer.from(state),
                 },
@@ -74,11 +99,16 @@ describe('isReady script', () => {
             const state = JSON.stringify({ status: 'ready', containers: { 'container-id': { status: ContainerHandlingStatus.INITIALIZING } } })
             process.argv = ['foo', 'bar', 'container-id'];
             proxyquire('../lib/isReady.js', {
+                '@codefresh-io/cf-telemetry/init': {
+                    terminate: () => ({
+                        finally: callback => callback(),
+                    })
+                },
                 'fs': {
                     readFileSync: () => Buffer.from(state),
                 },
             });
             expect(process.exit).to.have.been.calledOnceWith(1);
         });
     });
-});
+});