diff --git a/packages/fastboot/src/fastboot-response.js b/packages/fastboot/src/fastboot-response.js index 88b05071..3c6058ea 100644 --- a/packages/fastboot/src/fastboot-response.js +++ b/packages/fastboot/src/fastboot-response.js @@ -4,7 +4,9 @@ const FastBootHeaders = require('./fastboot-headers'); class FastbootResponse { constructor(response) { - this.headers = new FastBootHeaders(response._headers); + this.headers = new FastBootHeaders( + typeof response.getHeaders === 'function' ? response.getHeaders() : response._headers + ); this.statusCode = 200; } } diff --git a/packages/fastboot/test/fastboot-response-test.js b/packages/fastboot/test/fastboot-response-test.js index aa99f6ac..545cf860 100644 --- a/packages/fastboot/test/fastboot-response-test.js +++ b/packages/fastboot/test/fastboot-response-test.js @@ -7,9 +7,11 @@ describe('FastBootResponse', function() { beforeEach(function() { var mockResponse = { - _headers: { - 'i-am-a': ['mock header', 'me too'], - cookie: '', + getHeaders() { + return { + 'i-am-a': ['mock header', 'me too'], + cookie: '', + }; }, }; diff --git a/test-packages/integration-tests/test/basic-test.js b/test-packages/integration-tests/test/basic-test.js index 20be1217..d8a8fea6 100644 --- a/test-packages/integration-tests/test/basic-test.js +++ b/test-packages/integration-tests/test/basic-test.js @@ -18,7 +18,11 @@ function dummyRequest() { }; } function dummyResponse() { - return { _headers: {} }; + return { + getHeaders() { + return {}; + }, + }; } describe("FastBoot", function() {