From 180aaa4d806947da6f7e35f5c3057c10e81cf173 Mon Sep 17 00:00:00 2001
From: Sumate Kongpui <sumatek@sourcecode.co.th>
Date: Mon, 22 Oct 2018 15:49:59 +0700
Subject: [PATCH] update unit test

---
 app.js                | 10 +++++++++-
 unitTest/test.js      | 32 +++++++++++++++++++++++++++++---
 unitTest/testCase.txt |  5 ++++-
 unitTest/unitTest.js  | 85 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 utils/log.js          |  1 +
 utils/utils.js        | 90 +++++++++++++++++++++---------------------------------------------------------------------
 6 files changed, 149 insertions(+), 74 deletions(-)
 create mode 100644 unitTest/unitTest.js

diff --git a/app.js b/app.js
index 1327c0f..b07ccdb 100644
--- a/app.js
+++ b/app.js
@@ -6,7 +6,12 @@ const utils = require('utils/utils');
 var log = require('./utils/log');
 var constants = require('./utils/constants');
 
-utils.testUnit(1,2,3,4,5);
+var testObj = {one:1,two:2,three:3};
+var testArray = [5,6];
+var testObj2 = {test:{
+  lv2 : 10
+}};
+console.log(utils.testUnit(testObj,4,testArray,testObj2));
 // console.log(conf.get('redis.host'));
 
 // setup generate request-id middleware
@@ -187,6 +192,9 @@ app.use( function( req, res, next ) {
     next();
 } );
 
+
+
+
 // handle not found
 // app.all('*', function(req, res) {
 
diff --git a/unitTest/test.js b/unitTest/test.js
index 43cae02..8bc4fd9 100644
--- a/unitTest/test.js
+++ b/unitTest/test.js
@@ -1,6 +1,9 @@
 var readTextFile = require('read-text-file');
+const path = require('path');
 
 
+var dir = path.parse(__dirname);
+
 try{
   var testCaseStr = readTextFile.readSync('./unitTest/testCase.txt').trim("\r\n").split("\r\n");
   var testCase = [];
@@ -12,11 +15,17 @@ try{
     describe('unitTest app',function() {      
       
       for(var i=0;i<testCase.length;i++)
-      {
-        var func = require(testCase[i].from);
+      {        
+        var func = require(dir.dir+path.sep+testCase[i].from);
         var expect = testCase[i].expect;
+        var comment = testCase[i].comment!=""?" => "+testCase[i].comment:testCase[i].comment;
+        
+
+       
+        testCase[i].input = getTrueData(testCase[i].input);
+
         var cal = func[testCase[i].function](...testCase[i].input);
-        it(testCase[i].function ,function(done) {
+        it(testCase[i].function+comment ,function(done) {
           
           if (expect == cal) {
             done();
@@ -32,4 +41,21 @@ catch(err) {
   // console.log(err);
 }
 
+function getTrueData(input)
+{
 
+  // console.log(input);
+  for(var j=0;j<input.length;j++)
+  {
+    try
+    {
+      input[j] = JSON.parse(input[j]);
+    }catch(err) {
+      // console.log(err);
+    }
+    
+    // console.log(input[j]);
+  }
+
+  return input;
+}
diff --git a/unitTest/testCase.txt b/unitTest/testCase.txt
index 9837bf5..19688ba 100644
--- a/unitTest/testCase.txt
+++ b/unitTest/testCase.txt
@@ -1 +1,4 @@
-{"function":"testUnit","input":[1,2,3,4,5],"expect":15,"from":"C:\\myWork\\git\\hospital\\utils\\utils.js"}
+{"function":"testUnit","input":["{\"one\":1,\"two\":2,\"three\":3}",4,"[5,6]","{\"test\":{\"lv2\":10}}"],"expect":31,"from":"utils\\utils.js","comment":"simpleTest"}
+{"function":"findCmdfromMethod","input":["GET"],"expect":"QUERY","from":"utils\\utils.js","comment":"GET"}
+{"function":"findCmdfromMethod","input":["GET"],"expect":"QUERY","from":"utils\\utils.js","comment":"GET"}
+{"function":"findCmdfromMethod","input":["GET"],"expect":"QUERY","from":"utils\\utils.js","comment":"GET"}
diff --git a/unitTest/unitTest.js b/unitTest/unitTest.js
new file mode 100644
index 0000000..bc3c0d6
--- /dev/null
+++ b/unitTest/unitTest.js
@@ -0,0 +1,85 @@
+var fileTempUnitTest = './unitTest/testCase.txt';
+const fs = require('fs');
+const path = require('path');
+const env = 'gentest';
+
+module.exports = function(rawInput,expect,comment)
+{
+
+    if(process.env.NODE_ENV == env)
+    {
+        // console.log(rawInput);
+        var input = [];
+        var data;
+        if(typeof rawInput == "object")
+        {
+            for(var i=0;i<rawInput.length;i++)
+            {
+                if(typeof rawInput[i] == "object")
+                    data = JSON.stringify(rawInput[i]);
+                else
+                    data = rawInput[i];            
+            
+                input.push(data);
+            }
+        }else
+            input.push(rawInput);
+
+            // var track = __stack[1].getFunctionName().split('\.');
+            
+            var testCase = {
+                function : getFunName(2),
+                input : input,
+                expect : expect,
+                from : path.relative('', __stack[1].getFileName()),
+                comment : comment ? comment : ""
+            };
+
+        // console.log(testCase);
+        writeFile(fileTempUnitTest,testCase);
+    }
+}
+
+Object.defineProperty(global, '__stack', {
+    get: function() {
+            var orig = Error.prepareStackTrace;
+            Error.prepareStackTrace = function(_, stack) {
+                return stack;
+            };
+            var err = new Error;
+            Error.captureStackTrace(err, arguments.callee);
+            var stack = err.stack;
+            Error.prepareStackTrace = orig;
+            return stack;
+        }
+    });
+
+function writeFile(fileName,obj){
+    fs.appendFile(fileName, JSON.stringify(obj)+"\r\n", function(err) {
+      if(err) {
+        return console.log(err);
+      }
+        // console.log("The file was saved!");
+    }); 
+}
+
+function getFunName(level)
+{
+    var track;
+    if(level)
+        track = __stack[level].getFunctionName().split('\.');
+    else    
+        track = __stack[1].getFunctionName().split('\.');
+    return track[track.length-1];
+}
+
+if(process.env.NODE_ENV == env)
+{
+    try {
+        fs.unlinkSync(fileTempUnitTest);
+    } catch (error) {
+        
+    }
+}
+    
+
diff --git a/utils/log.js b/utils/log.js
index 8be9017..5401f1a 100644
--- a/utils/log.js
+++ b/utils/log.js
@@ -223,6 +223,7 @@ logger.detailResponseFE = function(req)
     logg.detail(logPrefix(req)+protocal+" "+method+" "+url+" Response to FE - body: "+body);
 }
 
+//mongo
 logger.detailSqlQuery = function(req,method,collection,data)
 {
     // logger4jDetail.info(logPrefix(req)+"BE Send Mongo Collection: \""+collection+"\" Query: "+JSON.stringify(data));
diff --git a/utils/utils.js b/utils/utils.js
index 6430283..e20255c 100644
--- a/utils/utils.js
+++ b/utils/utils.js
@@ -1,8 +1,13 @@
 
 var parseFilter = require('ldapjs').parseFilter;
-const fs = require('fs');
 var constants = require('./constants');
-var fileTempUnitTest = './unitTest/testCase.txt';
+
+// //unit test
+// var fileTempUnitTest = './unitTest/testCase.txt';
+// const fs = require('fs');
+// const path = require('path');
+
+var unitTest = require('../unitTest/unitTest');
 var utils = {};
 
 
@@ -100,75 +105,28 @@ utils.findCmdfromMethod = function (method)
             cmd = 'DELETE';
             break;        
     }
+    unitTest(arguments,cmd,method);
     return cmd;
 }
 
-utils.testUnit = function(a,b,c,d,e)
+utils.testUnit = function(testObj,testInt,testArray,testObj2)
 {
-  var returnData = a+b+c+d+e;
-  genUnitTest(a,b,c,d,e,returnData);
-  return returnData;
-} 
+    var returnData = 0;
+    
+    var keyObj = Object.keys(testObj);
+    for(var i=0;i<keyObj.length;i++) 
+        returnData += testObj[keyObj[i]];
 
-//private
-Object.defineProperty(global, '__stack', {
-    get: function() {
-            var orig = Error.prepareStackTrace;
-            Error.prepareStackTrace = function(_, stack) {
-                return stack;
-            };
-            var err = new Error;
-            Error.captureStackTrace(err, arguments.callee);
-            var stack = err.stack;
-            Error.prepareStackTrace = orig;
-            return stack;
-        }
-    });
-
-function writeFile(fileName,obj){
-    fs.appendFile(fileName, JSON.stringify(obj)+"\r\n", function(err) {
-      if(err) {
-        return console.log(err);
-      }
-        // console.log("The file was saved!");
-    }); 
-}
+    returnData += testInt;
 
-function genUnitTest(...rawdata)
-{
-    if(process.env.NODE_ENV == 'gentest')
-    {
-        console.log(process.env.NODE_ENV);
-        var input = [];
-        var expect;
-        var data;
-        for(var i=0;i<rawdata.length;i++)
-        {
-            if(typeof rawdata[i] == "object")
-            data = JSON.stringify(rawdata[i]);
-            else
-            data = rawdata[i];
-            
-            if(i==rawdata.length-1)
-            expect = data;
-            else
-            input.push(data);
-        }
+    for(var i=0;i<testArray.length;i++) 
+        returnData += testArray[i];
 
+    returnData += testObj2.test.lv2;
 
-            var track = __stack[1].getFunctionName().split('\.');
-            
-            var testCase = {
-                function : track[track.length-1],
-                input : input,
-                expect : expect,
-                from : __filename
-            };
-
-        // console.log(testCase);
-        writeFile(fileTempUnitTest,testCase);
-    }
-}
+    unitTest(arguments,returnData,"simpleTest");
+    return returnData;
+} 
 
 function findValueInBracket(data)
 {
@@ -224,8 +182,6 @@ function findValueInBracket(data)
 
 }
 
-
-
 getQuery = async function(data,query){
     
     if(!query)
@@ -289,8 +245,4 @@ getQuery = async function(data,query){
 
 
 
-if(process.env.NODE_ENV == 'gentest')
-    fs.unlinkSync(fileTempUnitTest);
-
-
 module.exports = utils;
\ No newline at end of file
--
libgit2 0.21.2