Merge pull request #14 from mkbug-com/feature_add_headerapi

<feat>: add header api get & set

bin/utils.js

@@ -43,6 +43,8 @@ module.exports = {
     ctx.params = req.params;
     ctx.status = 200;
     ctx.type = null;
+    ctx.get = req.get.bind(req);
+    ctx.set = res.set.bind(res);
 
     return ctx;
   },

example/controller/HeaderApiTest.js

@@ -0,0 +1,12 @@
+const { BaseController } = require('../../index');
+
+module.exports = class HeaderApiTest extends BaseController {
+  getAction () {
+    return this.get('from-header');
+  }
+
+  getHeaderAction () {
+    this.set('from-header', 'from-header')
+    return ''
+  }
+}

package.json

@@ -1,6 +1,6 @@
 {
   "name": "mkbugjs",
-  "version": "1.4.2",
+  "version": "1.5.0",
   "description": "An OOP style declare Nodejs Web framework base on Express.js",
   "main": "index.js",
   "scripts": {

test/controller.header.test.js

@@ -0,0 +1,29 @@
+const request = require('superagent');
+
+describe("Controller Header", () => {
+  it("Header from client", async () => {
+    let ret = null;
+    try {
+      ret = await request.get('http://localhost:3000/api/headerapitest')
+      .set('from-header', 'from-header')
+    } catch (err) {
+      ret = err;
+    } finally {
+      expect(ret.status).toBe(200);
+      expect(ret.text).toBe('from-header');
+    }
+  });
+
+  it("Header from server", async () => {
+    let ret = null;
+    try {
+      ret = await request.get('http://localhost:3000/api/headerapitest/header')
+    } catch (err) {
+      ret = err;
+    } finally {
+      expect(ret.status).toBe(200);
+      console.log(ret.header)
+      expect(ret.header['from-header']).toBe('from-header');
+    }
+  });
+})