tests: Have fake module() set _M to satisfy some modules

[prosody.git] / tests / test.lua

diff --git a/tests/test.lua b/tests/test.lua
index 19aa838952c629c550c13c21efbe12374bb0aa58..4f41cd1db369397cf19834e6fa9b893b10106c23 100644 (file)
--- a/tests/test.lua
+++ b/tests/test.lua
@@ -1,15 +1,34 @@
+-- Prosody IM
+-- Copyright (C) 2008-2009 Matthew Wild
+-- Copyright (C) 2008-2009 Waqas Hussain
+-- 
+-- This project is MIT/X11 licensed. Please see the
+-- COPYING file in the source package for more information.
+--
+
+
 
 function run_all_tests()
        dotest "util.jid"
+       dotest "util.multitable"
+       dotest "core.modulemanager"
        dotest "core.stanza_router"
        dotest "core.s2smanager"
        dotest "core.configmanager"
+       dotest "util.stanza"
+               
+       dosingletest("test_sasl.lua", "latin1toutf8");
 end
 
 local verbosity = tonumber(arg[1]) or 2;
 
-package.path = package.path..";../?.lua";
-package.cpath = package.cpath..";../?.so";
+if os.getenv("WINDIR") then
+       package.path = package.path..";..\\?.lua";
+       package.cpath = package.cpath..";..\\?.dll";
+else
+       package.path = package.path..";../?.lua";
+       package.cpath = package.cpath..";../?.so";
+end
 
 require "util.import"
 
@@ -18,14 +37,72 @@ function testlib_new_env(t)
        return setmetatable(t or {}, env_mt);
 end
 
-function assert_equal(a, b, message)
+function assert_equal(a, b, message, level)
        if not (a == b) then
-               error("\n   assert_equal failed: "..tostring(a).." ~= "..tostring(b)..(message and ("\n   Message: "..message) or ""), 2);
+               error("\n   assert_equal failed: "..tostring(a).." ~= "..tostring(b)..(message and ("\n   Message: "..message) or ""), (level or 1) + 1);
        elseif verbosity >= 4 then
                print("assert_equal succeeded: "..tostring(a).." == "..tostring(b));
        end
 end
 
+function assert_table(a, message, level)
+       assert_equal(type(a), "table", message, (level or 1) + 1);
+end
+function assert_function(a, message, level)
+       assert_equal(type(a), "function", message, (level or 1) + 1);
+end
+function assert_string(a, message, level)
+       assert_equal(type(a), "string", message, (level or 1) + 1);
+end
+function assert_boolean(a, message)
+       assert_equal(type(a), "boolean", message);
+end
+function assert_is(a, message)
+       assert_equal(not not a, true, message);
+end
+function assert_is_not(a, message)
+       assert_equal(not not a, false, message);
+end
+
+
+function dosingletest(testname, fname)
+       local tests = setmetatable({}, { __index = _G });
+       tests.__unit = testname;
+       tests.__test = fname;
+       local chunk, err = loadfile(testname);
+       if not chunk then
+               print("WARNING: ", "Failed to load tests for "..testname, err);
+               return;
+       end
+
+       setfenv(chunk, tests);
+       local success, err = pcall(chunk);
+       if not success then
+               print("WARNING: ", "Failed to initialise tests for "..testname, err);
+               return;
+       end
+       
+       if type(tests[fname]) ~= "function" then
+               error(testname.." has no test '"..fname.."'", 0);
+       end
+       
+       
+       local line_hook, line_info = new_line_coverage_monitor(testname);
+       debug.sethook(line_hook, "l")
+       local success, ret = pcall(tests[fname]);
+       debug.sethook();
+       if not success then
+               print("TEST FAILED! Unit: ["..testname.."] Function: ["..fname.."]");
+               print("   Location: "..ret:gsub(":%s*\n", "\n"));
+               line_info(fname, false, report_file);
+       elseif verbosity >= 2 then
+               print("TEST SUCCEEDED: ", testname, fname);
+               print(string.format("TEST COVERED %d/%d lines", line_info(fname, true, report_file)));
+       else
+               line_info(name, success, report_file);
+       end
+end
+
 function dotest(unitname)
        local tests = setmetatable({}, { __index = _G });
        tests.__unit = unitname;
@@ -42,7 +119,7 @@ function dotest(unitname)
                return;
        end
        
-       local unit = setmetatable({}, { __index = setmetatable({ module = function () end }, { __index = _G }) });
+       local unit = setmetatable({}, { __index = setmetatable({ module = function () _M = getfenv(2); end }, { __index = _G }) });
 
        local fn = "../"..unitname:gsub("%.", "/")..".lua";
        local chunk, err = loadfile(fn);
@@ -92,7 +169,7 @@ function runtest(f, msg)
        local success, ret = pcall(f);
        if success and verbosity >= 2 then
                print("SUBTEST PASSED: "..(msg or "(no description)"));
-       elseif (not success) and verbosity >= 1 then
+       elseif (not success) and verbosity >= 0 then
                print("SUBTEST FAILED: "..(msg or "(no description)"));
                error(ret, 0);
        end