Add unittest framework.

* TODO: more UT cases.

Signed-off-by: Eric Wang <[email protected]>

Author: skygragon

.travis.yml

@@ -2,5 +2,5 @@ language: node_js
 node_js:
   - stable
 install:
-  - npm install -g eslint-cli
+  - npm install -g eslint-cli mocha
   - npm install

lib/helper.js

@@ -85,6 +85,7 @@ h.getSetCookieValue = function(resp, key) {
       if (kv[0] === key) return kv[1];
     }
   }
+  return null;
 };
 
 module.exports = h;

package.json

@@ -7,7 +7,7 @@
     "lc": "./bin/lc"
   },
   "scripts": {
-    "test": "npm run lint",
+    "test": "npm run lint && mocha test/",
     "lint": "eslint lib/"
   },
   "repository": {
@@ -38,7 +38,9 @@
     "yargs": "^5.0.0"
   },
   "devDependencies": {
+    "chai": "^3.5.0",
     "eslint": "^3.3.1",
-    "eslint-config-google": "^0.6.0"
+    "eslint-config-google": "^0.6.0",
+    "mocha": "^3.0.2"
   }
 }

test/test_helper.js

@@ -0,0 +1,20 @@
+var assert = require('chai').assert;
+
+var h = require('../lib/helper');
+
+describe('helper', function() {
+  describe('#getSetCookieValue', function() {
+    it('should ok', function() {
+      var resp = {
+        headers: {'set-cookie': [
+          'key1=value1; path=/; Httponly',
+          'key2=value2; path=/; Httponly']
+        }
+      };
+
+      assert.equal(h.getSetCookieValue(resp, 'key1'), 'value1');
+      assert.equal(h.getSetCookieValue(resp, 'key2'), 'value2');
+      assert.equal(h.getSetCookieValue(resp, 'key3'), null);
+    });
+  });
+});