2 -- Copyright (C) 2008-2010 Matthew Wild
3 -- Copyright (C) 2008-2010 Waqas Hussain
5 -- This project is MIT/X11 licensed. Please see the
6 -- COPYING file in the source package for more information.
9 local tests_passed = true;
11 function run_all_tests()
12 package.loaded["net.connlisteners"] = { get = function () return {} end };
14 dotest "util.multitable"
17 dotest "core.stanza_router"
18 dotest "core.s2smanager"
19 dotest "core.configmanager"
22 dotest "util.sasl.scram"
24 dotest "util.throttle"
27 dosingletest("test_sasl.lua", "latin1toutf8");
28 dosingletest("test_utf8.lua", "valid");
31 local verbosity = tonumber(arg[1]) or 2;
33 if os.getenv("WINDIR") then
34 package.path = package.path..";..\\?.lua";
35 package.cpath = package.cpath..";..\\?.dll";
37 package.path = package.path..";../?.lua";
38 package.cpath = package.cpath..";../?.so";
45 local env_mt = { __index = function (t,k) return rawget(_realG, k) or print("WARNING: Attempt to access nil global '"..tostring(k).."'"); end };
46 function testlib_new_env(t)
47 return setmetatable(t or {}, env_mt);
50 function assert_equal(a, b, message, level)
52 error("\n assert_equal failed: "..tostring(a).." ~= "..tostring(b)..(message and ("\n Message: "..message) or ""), (level or 1) + 1);
53 elseif verbosity >= 4 then
54 print("assert_equal succeeded: "..tostring(a).." == "..tostring(b));
58 function assert_table(a, message, level)
59 assert_equal(type(a), "table", message, (level or 1) + 1);
61 function assert_function(a, message, level)
62 assert_equal(type(a), "function", message, (level or 1) + 1);
64 function assert_string(a, message, level)
65 assert_equal(type(a), "string", message, (level or 1) + 1);
67 function assert_boolean(a, message)
68 assert_equal(type(a), "boolean", message);
70 function assert_is(a, message)
71 assert_equal(not not a, true, message);
73 function assert_is_not(a, message)
74 assert_equal(not not a, false, message);
78 function dosingletest(testname, fname)
79 local tests = setmetatable({}, { __index = _realG });
80 tests.__unit = testname;
82 local chunk, err = loadfile(testname);
84 print("WARNING: ", "Failed to load tests for "..testname, err);
88 setfenv(chunk, tests);
89 local success, err = pcall(chunk);
91 print("WARNING: ", "Failed to initialise tests for "..testname, err);
95 if type(tests[fname]) ~= "function" then
96 error(testname.." has no test '"..fname.."'", 0);
100 local line_hook, line_info = new_line_coverage_monitor(testname);
101 debug.sethook(line_hook, "l")
102 local success, ret = pcall(tests[fname]);
105 tests_passed = false;
106 print("TEST FAILED! Unit: ["..testname.."] Function: ["..fname.."]");
107 print(" Location: "..ret:gsub(":%s*\n", "\n"));
108 line_info(fname, false, report_file);
109 elseif verbosity >= 2 then
110 print("TEST SUCCEEDED: ", testname, fname);
111 print(string.format("TEST COVERED %d/%d lines", line_info(fname, true, report_file)));
113 line_info(name, success, report_file);
117 function dotest(unitname)
118 local _fakeG = setmetatable({}, {__index = _realG});
120 local tests = setmetatable({}, { __index = _fakeG });
121 tests.__unit = unitname;
122 local chunk, err = loadfile("test_"..unitname:gsub("%.", "_")..".lua");
124 print("WARNING: ", "Failed to load tests for "..unitname, err);
128 setfenv(chunk, tests);
129 local success, err = pcall(chunk);
131 print("WARNING: ", "Failed to initialise tests for "..unitname, err);
134 if tests.env then setmetatable(tests.env, { __index = _realG }); end
135 local unit = setmetatable({}, { __index = setmetatable({ _G = tests.env or _fakeG }, { __index = tests.env or _fakeG }) });
136 local fn = "../"..unitname:gsub("%.", "/")..".lua";
137 local chunk, err = loadfile(fn);
139 print("WARNING: ", "Failed to load module: "..unitname, err);
143 local oldmodule, old_M = _fakeG.module, _fakeG._M;
144 _fakeG.module = function ()
145 setmetatable(unit, nil);
148 setfenv(chunk, unit);
149 local success, ret = pcall(chunk);
150 _fakeG.module, _fakeG._M = oldmodule, old_M;
152 print("WARNING: ", "Failed to initialise module: "..unitname, ret);
156 if type(ret) == "table" then
157 for k,v in pairs(ret) do
162 for name, f in pairs(unit) do
163 local test = rawget(tests, name);
164 if type(f) ~= "function" then
165 if verbosity >= 3 then
166 print("INFO: ", "Skipping "..unitname.."."..name.." because it is not a function");
168 elseif type(test) ~= "function" then
169 if verbosity >= 1 then
170 print("WARNING: ", unitname.."."..name.." has no test!");
173 if verbosity >= 4 then
174 print("INFO: ", "Testing "..unitname.."."..name);
176 local line_hook, line_info = new_line_coverage_monitor(fn);
177 debug.sethook(line_hook, "l")
178 local success, ret = pcall(test, f, unit);
181 tests_passed = false;
182 print("TEST FAILED! Unit: ["..unitname.."] Function: ["..name.."]");
183 print(" Location: "..ret:gsub(":%s*\n", "\n"));
184 line_info(name, false, report_file);
185 elseif verbosity >= 2 then
186 print("TEST SUCCEEDED: ", unitname, name);
187 print(string.format("TEST COVERED %d/%d lines", line_info(name, true, report_file)));
189 line_info(name, success, report_file);
195 function runtest(f, msg)
196 if not f then print("SUBTEST NOT FOUND: "..(msg or "(no description)")); return; end
197 local success, ret = pcall(f);
198 if success and verbosity >= 2 then
199 print("SUBTEST PASSED: "..(msg or "(no description)"));
200 elseif (not success) and verbosity >= 0 then
201 tests_passed = false;
202 print("SUBTEST FAILED: "..(msg or "(no description)"));
207 function new_line_coverage_monitor(file)
208 local lines_hit, funcs_hit = {}, {};
209 local total_lines, covered_lines = 0, 0;
211 for line in io.lines(file) do
212 total_lines = total_lines + 1;
215 return function (event, line) -- Line hook
216 if not lines_hit[line] then
217 local info = debug.getinfo(2, "fSL")
218 if not info.source:find(file) then return; end
219 if not funcs_hit[info.func] and info.activelines then
220 funcs_hit[info.func] = true;
221 for line in pairs(info.activelines) do
222 lines_hit[line] = false; -- Marks it as hittable, but not hit yet
225 if lines_hit[line] == false then
226 --print("New line hit: "..line.." in "..debug.getinfo(2, "S").source);
227 lines_hit[line] = true;
228 covered_lines = covered_lines + 1;
232 function (test_name, success) -- Get info
233 local fn = file:gsub("^%W*", "");
234 local total_active_lines = 0;
235 local coverage_file = io.open("reports/coverage_"..fn:gsub("%W+", "_")..".report", "a+");
236 for line, active in pairs(lines_hit) do
237 if active ~= nil then total_active_lines = total_active_lines + 1; end
238 if coverage_file then
239 if active == false then coverage_file:write(fn, "|", line, "|", name or "", "|miss\n");
240 else coverage_file:write(fn, "|", line, "|", name or "", "|", tostring(success), "\n"); end
243 if coverage_file then coverage_file:close(); end
244 return covered_lines, total_active_lines, lines_hit;
250 os.exit(tests_passed and 0 or 1);