@@ -13,7 +13,7 @@ describe('addNewMask', () => {
1313
1414 const stubGetServerAddress = sinon . stub ( ) ;
1515 const stubProcessExit = sinon . stub ( ) ;
16- const stubConsole = {
16+ const stubLogger = {
1717 debug : sinon . stub ( ) ,
1818 error : sinon . stub ( ) ,
1919 warn : sinon . stub ( ) ,
@@ -30,16 +30,15 @@ describe('addNewMask', () => {
3030
3131 before ( async ( ) => {
3232 process . exit = stubProcessExit ;
33- console = stubConsole ;
3433 const { default : httpClient } = await import ( 'got' ) ;
3534 sinon . stub ( httpClient , 'post' ) . callsFake ( stubGot . post ) ;
3635 } ) ;
3736
3837 beforeEach ( ( ) => {
3938 stubProcessExit . resetHistory ( ) ;
4039 stubGetServerAddress . resetHistory ( ) ;
41- for ( const stub in stubConsole ) {
42- stubConsole [ stub ] . resetHistory ( ) ;
40+ for ( const stub in stubLogger ) {
41+ stubLogger [ stub ] . resetHistory ( ) ;
4342 }
4443 for ( const stub in stubGot ) {
4544 stubGot [ stub ] . resetHistory ( ) ;
@@ -78,6 +77,9 @@ describe('addNewMask', () => {
7877 it ( 'should fail if the server address is not available' , async ( ) => {
7978 stubGetServerAddress . rejects ( 'could not get server address' ) ;
8079 const { updateMasks, exitHandler } = proxyquire ( '../lib/addNewMask' , {
80+ '@codefresh-io/cf-telemetry/logs' : {
81+ Logger : function ( ) { return stubLogger } ,
82+ } ,
8183 './helpers' : {
8284 getServerAddress : stubGetServerAddress ,
8385 } ,
@@ -88,13 +90,16 @@ describe('addNewMask', () => {
8890 }
8991 } ) ;
9092 await updateMasks ( secret ) ;
91- expect ( stubConsole . error ) . to . have . been . calledOnceWith ( 'could not create mask for secret: 123. Error: could not get server address' ) ;
93+ expect ( stubLogger . error ) . to . have . been . calledOnceWith ( 'could not create mask for secret: 123. Error: could not get server address' ) ;
9294 expect ( stubProcessExit ) . to . have . been . calledOnceWith ( 1 ) ;
9395 } ) ;
9496
9597 it ( 'should fail if the server address is not valid URL' , async ( ) => {
9698 stubGetServerAddress . resolves ( 'foo' ) ;
9799 const { updateMasks, exitHandler } = proxyquire ( '../lib/addNewMask' , {
100+ '@codefresh-io/cf-telemetry/logs' : {
101+ Logger : function ( ) { return stubLogger } ,
102+ } ,
98103 './helpers' : {
99104 getServerAddress : stubGetServerAddress ,
100105 } ,
@@ -105,7 +110,7 @@ describe('addNewMask', () => {
105110 }
106111 } ) ;
107112 await updateMasks ( secret ) ;
108- expect ( stubConsole . error ) . to . have . been . calledOnceWith ( 'could not create mask for secret: 123. Error: TypeError: Invalid URL' ) ;
113+ expect ( stubLogger . error ) . to . have . been . calledOnceWith ( 'could not create mask for secret: 123. Error: TypeError: Invalid URL' ) ;
109114 expect ( stubProcessExit ) . to . have . been . calledOnceWith ( 1 ) ;
110115 } ) ;
111116
@@ -117,6 +122,9 @@ describe('addNewMask', () => {
117122 body : 'Internal Server Error' ,
118123 } ) ;
119124 const { updateMasks, exitHandler } = proxyquire ( '../lib/addNewMask' , {
125+ '@codefresh-io/cf-telemetry/logs' : {
126+ Logger : function ( ) { return stubLogger } ,
127+ } ,
120128 './helpers' : { getServerAddress : stubGetServerAddress } ,
121129 } ) ;
122130 process . listeners ( 'exit' ) . forEach ( ( listener ) => {
@@ -125,27 +133,35 @@ describe('addNewMask', () => {
125133 }
126134 } ) ;
127135 await updateMasks ( secret ) ;
128- expect ( stubConsole . error ) . to . have . been . calledOnceWith ( 'could not create mask for secret: 123. Server responded with: 500\n\nInternal Server Error' ) ;
136+ expect ( stubLogger . error ) . to . have . been . calledOnceWith ( 'could not create mask for secret: 123. Server responded with: 500\n\nInternal Server Error' ) ;
129137 expect ( stubProcessExit ) . to . have . been . calledOnceWith ( 1 ) ;
130138 } ) ;
131139 } ) ;
132140
133141 describe ( 'exitHandler' , ( ) => {
134142 it ( 'should set exit code to 3 if the original exit code is 0 and variable was not masked' , ( ) => {
135- const { exitHandler } = proxyquire ( '../lib/addNewMask' , { } ) ;
143+ const { exitHandler } = proxyquire ( '../lib/addNewMask' , {
144+ '@codefresh-io/cf-telemetry/logs' : {
145+ Logger : function ( ) { return stubLogger } ,
146+ } ,
147+ } ) ;
136148 process . listeners ( 'exit' ) . forEach ( ( listener ) => {
137149 if ( listener === exitHandler ) {
138150 process . removeListener ( 'exit' , listener ) ;
139151 }
140152 } ) ;
141153 exitHandler ( 0 ) ;
142154 expect ( process . exitCode ) . to . be . equal ( 3 ) ;
143- expect ( stubConsole . warn ) . to . have . been . calledOnceWith ( 'Unexpected exit with code 0. Exiting with 3 instead' ) ;
155+ expect ( stubLogger . warn ) . to . have . been . calledOnceWith ( 'Unexpected exit with code 0. Exiting with 3 instead' ) ;
144156 process . exitCode = undefined ;
145157 } ) ;
146158
147159 it ( 'should set exit code to 3 if the original exit code is 0 and variable was not masked' , ( ) => {
148- const { exitHandler } = proxyquire ( '../lib/addNewMask' , { } ) ;
160+ const { exitHandler } = proxyquire ( '../lib/addNewMask' , {
161+ '@codefresh-io/cf-telemetry/logs' : {
162+ Logger : function ( ) { return stubLogger } ,
163+ } ,
164+ } ) ;
149165 process . listeners ( 'exit' ) . forEach ( ( listener ) => {
150166 if ( listener === exitHandler ) {
151167 process . removeListener ( 'exit' , listener ) ;
@@ -157,7 +173,7 @@ describe('addNewMask', () => {
157173 process . exitCode = 0 ;
158174 exitHandler ( ) ;
159175 expect ( process . exitCode ) . to . be . equal ( 3 ) ;
160- expect ( stubConsole . warn ) . to . have . been . calledOnceWith ( 'Unexpected exit with code 0. Exiting with 3 instead' ) ;
176+ expect ( stubLogger . warn ) . to . have . been . calledOnceWith ( 'Unexpected exit with code 0. Exiting with 3 instead' ) ;
161177 process . exitCode = 0 ;
162178 } ) ;
163179
@@ -175,7 +191,7 @@ describe('addNewMask', () => {
175191 } ) ;
176192 await updateMasks ( secret ) ;
177193 expect ( process . exitCode ) . not . to . be . equal ( 3 ) ;
178- expect ( stubConsole . warn ) . not . to . have . been . calledOnceWith ( 'Unexpected exit with code 0. Exiting with 3 instead' ) ;
194+ expect ( stubLogger . warn ) . not . to . have . been . calledOnceWith ( 'Unexpected exit with code 0. Exiting with 3 instead' ) ;
179195 } ) ;
180196 } ) ;
181197} ) ;
0 commit comments