local verbosity = tonumber(arg[1]) or 2;

package.path = package.path..";../?.lua";

require "util.import"

local env_mt = { __index = function (t,k) return rawget(_G, k) or print("WARNING: Attempt to access nil global '"..tostring(k).."'"); end };

function testlib_new_env(t)

	return setmetatable(t or {}, env_mt);

end

function assert_equal(a, b, message)

	if not (a == b) then

		error("\n   assert_equal failed: "..tostring(a).." ~= "..tostring(b)..(message and ("\n   Message: "..message) or ""), 2);

	elseif verbosity >= 4 then

		print("assert_equal succeeded: "..tostring(a).." == "..tostring(b));

	end

end

function dotest(unitname)

	local tests = setmetatable({}, { __index = _G });

	tests.__unit = unitname;

	local chunk, err = loadfile("test_"..unitname:gsub("%.", "_")..".lua");

	if not chunk then

		print("WARNING: ", "Failed to load tests for "..unitname, err);

		return;

	end

	setfenv(chunk, tests);

	local success, err = pcall(chunk);

	if not success then

		print("WARNING: ", "Failed to initialise tests for "..unitname, err);

		return;

	end

	local unit = setmetatable({}, { __index = setmetatable({ module = function () end }, { __index = _G }) });

	local chunk, err = loadfile("../"..unitname:gsub("%.", "/")..".lua");

	if not chunk then

		print("WARNING: ", "Failed to load module: "..unitname, err);

		return;

	end

	setfenv(chunk, unit);

	local success, err = pcall(chunk);

	if not success then

		print("WARNING: ", "Failed to initialise module: "..unitname, err);

		return;

	end

	for name, f in pairs(unit) do

		if type(f) ~= "function" then

			if verbosity >= 3 then

				print("INFO: ", "Skipping "..unitname.."."..name.." because it is not a function");

			end

		elseif type(tests[name]) ~= "function" then

			if verbosity >= 1 then

				print("WARNING: ", unitname.."."..name.." has no test!");

			end

		else

			local success, ret = pcall(tests[name], f, unit);

			if not success then

				print("TEST FAILED! Unit: ["..unitname.."] Function: ["..name.."]");

				print("   Location: "..ret:gsub(":%s*\n", "\n"));

			elseif verbosity >= 2 then

				print("TEST SUCCEEDED: ", unitname, name);

			end

		end

	end

end

function runtest(f, msg)

	if not f then print("SUBTEST NOT FOUND: "..(msg or "(no description)")); return; end

	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

		print("SUBTEST FAILED: "..(msg or "(no description)"));

		error(ret, 0);

	end

end

dotest "util.jid"

dotest "core.stanza_router"