diff --git a/lib/connect-logger.js b/lib/connect-logger.js index b9a64b8..b73822a 100755 --- a/lib/connect-logger.js +++ b/lib/connect-logger.js @@ -273,6 +273,7 @@ module.exports = function getLogger(logger4js, options) { res.__statusCode = code; res.__headers = headers || {}; + return res; }; // hook on end request to emit the log entry of the HTTP request. diff --git a/test/tap/connect-context-test.js b/test/tap/connect-context-test.js index ffd88dc..58d440e 100644 --- a/test/tap/connect-context-test.js +++ b/test/tap/connect-context-test.js @@ -58,6 +58,7 @@ class MockResponse extends EE { writeHead(code /* , headers */) { this.statusCode = code; + return this; } } diff --git a/test/tap/connect-logger-test.js b/test/tap/connect-logger-test.js index 4119f31..48dc487 100644 --- a/test/tap/connect-logger-test.js +++ b/test/tap/connect-logger-test.js @@ -58,6 +58,7 @@ class MockResponse extends EE { writeHead(code /* , headers */) { this.statusCode = code; + return this; } } @@ -87,8 +88,7 @@ function request( next = () => {}; } cl(req, res, next); - res.writeHead(code, resHeaders); - res.end('chunk', 'encoding'); + res.writeHead(code, resHeaders).end('chunk', 'encoding'); } test('log4js connect logger', (batch) => { diff --git a/test/tap/connect-nolog-test.js b/test/tap/connect-nolog-test.js index 1d780a7..1c605e4 100644 --- a/test/tap/connect-nolog-test.js +++ b/test/tap/connect-nolog-test.js @@ -49,6 +49,7 @@ class MockResponse extends EE { writeHead(code /* , headers */) { this.statusCode = code; + return this; } }