@@ -70,11 +70,29 @@ describe("Base", () => {
7070 expect ( baseFormat . compose ( "test" , { } ) ) . toBeUndefined ( ) ;
7171 } ) ;
7272
73- test ( "Mock Response" , async ( ) => {
73+ test ( "Enhance Request - Host" , async ( ) => {
74+ const next = jest . fn ( ) ;
75+ const socketServer = new SocketServer ( ) ;
76+ let req = { headers : { host : "localhost:4711" } } ;
77+ socketServer . enhanceRequest ( { request : req } , next ) ;
78+ expect ( req . baseUrl ) . toBe ( req . url ) ;
79+ expect ( req . originalUrl ) . toBe ( req . url ) ;
80+ expect ( req . hostname ) . toBe ( "localhost:4711" ) ;
81+ expect ( req . host ) . toBe ( "localhost:4711" ) ;
82+
83+ req = { headers : { host : "localhost:4711" , "x-forwarded-host" : "localhost:4712" } } ;
84+ socketServer . enhanceRequest ( { request : req } , next ) ;
85+ expect ( req . baseUrl ) . toBe ( req . url ) ;
86+ expect ( req . originalUrl ) . toBe ( req . url ) ;
87+ expect ( req . hostname ) . toBe ( "localhost:4712" ) ;
88+ expect ( req . host ) . toBe ( "localhost:4712" ) ;
89+ } ) ;
90+
91+ test ( "Enhance Request - Mock Response" , async ( ) => {
7492 const socketServer = new SocketServer ( ) ;
7593 const req = { } ;
7694 const next = jest . fn ( ) ;
77- socketServer . mockResponse ( { request : req } , next ) ;
95+ socketServer . enhanceRequest ( { request : req } , next ) ;
7896 expect ( req . res ) . toBeDefined ( ) ;
7997 expect ( req . res . headers ) . toEqual ( { } ) ;
8098 expect ( req . res . set ( "A" , "B" ) ) . toBe ( req . res ) ;
0 commit comments