Blame view

unitTest/unitTest.js 2.13 KB
180aaa4d   sumatek   update unit test
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
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];
}

8349d210   sumatek   update unittest
76
77
78
79
80
// if(process.env.NODE_ENV == env)
// {
//     try {
//         fs.unlinkSync(fileTempUnitTest);
//     } catch (error) {
180aaa4d   sumatek   update unit test
81
        
8349d210   sumatek   update unittest
82
83
//     }
// }
180aaa4d   sumatek   update unit test
84