feat(Response): add setHeader as header fn alias (#265)

Author: naorpeled

__tests__/headers.unit.js

@@ -90,6 +90,13 @@ api.get('/hasHeader', function(req,res) {
   })
 })
 
+api.get('/setHeader', function(req,res) {
+  res.status(200).header('TestHeader','test').setHeader('NewHeader','test')
+  res.json({
+    headers: res.getHeaders()
+  })
+});
+
 api.get('/removeHeader', function(req,res) {
   res.status(200).header('TestHeader','test').header('NewHeader','test').removeHeader('testHeader')
   res.json({
@@ -243,6 +250,20 @@ describe('Header Tests:', function() {
       })
     }) // end it
 
+    it('Set Header', async function() {
+      let _event = Object.assign({},event,{ path: '/setHeader'})
+      let result = await new Promise(r => api.run(_event,{},(e,res) => { r(res) }))
+      expect(result).toEqual({
+        multiValueHeaders: {
+          'content-type': ['application/json'],
+          'testheader': ['test'],
+          'newheader': ['test']
+        }, statusCode: 200,
+        body: "{\"headers\":{\"content-type\":[\"application/json\"],\"testheader\":[\"test\"],\"newheader\":[\"test\"]}}",
+        isBase64Encoded: false
+      })
+    }) // end it
+
     it('Remove Header', async function() {
       let _event = Object.assign({},event,{ path: '/removeHeader'})
       let result = await new Promise(r => api.run(_event,{},(e,res) => { r(res) }))

index.d.ts

@@ -205,6 +205,10 @@ export declare class Response {
 
   getHeader(key: string): string;
 
+  getHeaders(): { [key: string]: string };
+
+  setHeader(...args: Parameters<typeof this.header>): this;
+
   hasHeader(key: string): boolean;
 
   removeHeader(key: string): this;

lib/response.js

@@ -96,6 +96,11 @@ class RESPONSE {
       : undefined;
   }
 
+  // Issue #130
+  setHeader(...args) {
+    return this.header(...args);
+  }
+
   getHeaders() {
     return this._headers;
   }