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
|
--
-- (C) 2017-22 - ntop.org
--
local dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
require "lua_trace"
local unittest = {}
function unittest:runner(test_no, test_name, f)
return function(...)
-- for each function run by this runner, failed assertions are counted
-- so we start clean with a reset
self.failed_assertions = 0
local f_name = debug.getinfo(1, "n").name
local start = os.time()
local result = {f(...)}
local finish = os.time()
local result = table.unpack(result)
traceError(TRACE_NORMAL,TRACE_CONSOLE, string.format("[%s][test: %.2u][%s] executed in %i secs",
self.failed_assertions > 0 and "FAILED" or "OK",
test_no, test_name, finish - start))
return self.failed_assertions == 0
end
end
function unittest:new(options)
local obj = {
test_fns = {},
failed_assertions = 0,
-- add here other class members and possibly initialize them from the options
}
setmetatable(obj, self)
self.__index = self
return obj
end
function unittest:assertEqual(actual, expected, msg)
if actual ~= expected then
traceError(TRACE_ERROR, TRACE_CONSOLE,
string.format("Assertion failed [actual: %s][expected: %s][%s]", tostring(actual), tostring(expected), msg))
self.failed_assertions = self.failed_assertions + 1
return false
end
return true
end
function unittest:appendTest(test_name, test_fn)
self.test_fns[#self.test_fns + 1] = {test_name = test_name, test_fn = test_fn}
end
function unittest:run()
local unittest_file = debug.getinfo(2).short_src
traceError(TRACE_NORMAL, TRACE_CONSOLE, string.format("Running unittest [%s]", unittest_file))
for test_no, test_fn in ipairs(self.test_fns) do
local test_name = test_fn["test_name"]
local test_fn = test_fn["test_fn"]
self:runner(test_no, test_name, test_fn)()
end
traceError(TRACE_NORMAL, TRACE_CONSOLE, string.format("Unittest done [%s]", unittest_file))
end
return unittest
|
