@@ -160,27 +160,21 @@ describe('updateResponseHeadersForAgentDownload', () => {
160160} )
161161
162162describe ( 'prepareHeadersForIngressAPI' , ( ) => {
163- it ( 'should set client ip and proxy secret' , ( ) => {
163+ it ( 'should set all proxy headers if proxy secret is defined, preserving the original headers ' , ( ) => {
164164 const result = prepareHeadersForIngressAPI ( mockReq , 'secret' )
165165
166166 expect ( result [ 'fpjs-proxy-client-ip' ] ) . toBe ( '127.0.0.1' )
167167 expect ( result [ 'fpjs-proxy-secret' ] ) . toBe ( 'secret' )
168168 expect ( result [ 'fpjs-proxy-forwarded-host' ] ) . toBe ( 'fpjs.sh' )
169+ expect ( result [ 'x-custom-header' ] ) . toBe ( mockReq . headers [ 'x-custom-header' ] )
169170 } )
170171
171- it ( 'should set correct host' , ( ) => {
172- const result = prepareHeadersForIngressAPI ( mockReq , 'secret' )
173-
174- expect ( result [ 'fpjs-proxy-client-ip' ] ) . toBe ( '127.0.0.1' )
175- expect ( result [ 'fpjs-proxy-secret' ] ) . toBe ( 'secret' )
176- expect ( result [ 'fpjs-proxy-forwarded-host' ] ) . toBe ( 'fpjs.sh' )
177- } )
178-
179- it ( 'should not set secret if it is undefined' , ( ) => {
172+ it ( 'should set the other proxy headers, even if proxy secret is not defined, preserving the original headers' , ( ) => {
180173 const result = prepareHeadersForIngressAPI ( mockReq , undefined )
181174
182175 expect ( result [ 'fpjs-proxy-client-ip' ] ) . toBe ( '127.0.0.1' )
176+ expect ( result [ 'fpjs-proxy-forwarded-host' ] ) . toBe ( 'fpjs.sh' )
183177 expect ( result [ 'fpjs-proxy-secret' ] ) . toBe ( undefined )
184- expect ( result [ 'fpjs-proxy-forwarded-host ' ] ) . toBe ( undefined )
178+ expect ( result [ 'x-custom-header ' ] ) . toBe ( mockReq . headers [ 'x-custom-header' ] )
185179 } )
186180} )
0 commit comments