server/share/lua/5.4/lunit.lua

760 lines
20 KiB
Lua
Raw Normal View History

--[[--------------------------------------------------------------------------
This file is part of lunitx.
For Details about lunit look at: http://www.mroth.net/lunit/
For Details about lunitx look at: https://github.com/dcurrie/lunit
Author: Michael Roth <mroth@nessie.de>
Copyright (c) 2004, 2006-2010 Michael Roth <mroth@nessie.de>
Copyright (c) 2011-2012 Doug Currie
Permission is hereby granted, free of charge, to any person
obtaining a copy of this software and associated documentation
files (the "Software"), to deal in the Software without restriction,
including without limitation the rights to use, copy, modify, merge,
publish, distribute, sublicense, and/or sell copies of the Software,
and to permit persons to whom the Software is furnished to do so,
subject to the following conditions:
The above copyright notice and this permission notice shall be
included in all copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
--]]--------------------------------------------------------------------------
local orig_assert = assert
local pairs = pairs
local ipairs = ipairs
local next = next
local type = type
local error = error
local tostring = tostring
local setmetatable = setmetatable
local pcall = pcall
local xpcall = xpcall
local require = require
local loadfile = loadfile
local string_sub = string.sub
local string_gsub = string.gsub
local string_format = string.format
local string_lower = string.lower
local string_find = string.find
local table_concat = table.concat
local debug_getinfo = debug.getinfo
local package = require "package"
local setfenv = setfenv
local _G = _G
local IS_LUA52 = _VERSION >= 'Lua 5.2'
local lunit
local module = _G.module
local function TEST_CASE(name)
if not IS_LUA52 then
module(name, package.seeall, lunit.testcase)
setfenv(2, _M)
return _M
else
return lunit.module(name, 'seeall')
end
end
if IS_LUA52 then
lunit = {}
_ENV = lunit
else
module("lunit")
lunit = _M
end
lunit.TEST_CASE = TEST_CASE
local __failure__ = {} -- Type tag for failed assertions
local __skip__ = {} -- Type tag for skipped tests
local typenames = { "nil", "boolean", "number", "string", "table", "function", "thread", "userdata" }
local traceback_hide -- Traceback function which hides lunit internals
local mypcall -- Protected call to a function with own traceback
do
local _tb_hide = setmetatable( {}, {__mode="k"} )
function traceback_hide(func)
_tb_hide[func] = true
end
local function my_traceback(errobj)
if is_table(errobj) and ((errobj.type == __failure__) or (errobj.type == __skip__)) then
local info = debug_getinfo(5, "Sl") -- FIXME: Hardcoded integers are bad...
errobj.where = string_format( "%s:%d", info.short_src, info.currentline)
else
errobj = { msg = tostring(errobj) }
errobj.tb = {}
local i = 2
while true do
local info = debug_getinfo(i, "Snlf")
if not is_table(info) then
break
end
if not _tb_hide[info.func] then
local line = {} -- Ripped from ldblib.c...
line[#line+1] = string_format("%s:", info.short_src)
if info.currentline > 0 then
line[#line+1] = string_format("%d:", info.currentline)
end
if info.namewhat ~= "" then
line[#line+1] = string_format(" in function '%s'", info.name)
else
if info.what == "main" then
line[#line+1] = " in main chunk"
elseif info.what == "C" or info.what == "tail" then
line[#line+1] = " ?"
else
line[#line+1] = string_format(" in function <%s:%d>", info.short_src, info.linedefined)
end
end
errobj.tb[#errobj.tb+1] = table_concat(line)
end
i = i + 1
end
end
return errobj
end
function mypcall(func)
orig_assert( is_function(func) )
local ok, errobj = xpcall(func, my_traceback)
if not ok then
return errobj
end
end
traceback_hide(mypcall)
end
-- Type check functions
for _, typename in ipairs(typenames) do
lunit["is_"..typename] = function(x)
return type(x) == typename
end
end
local is_nil = is_nil
local is_boolean = is_boolean
local is_number = is_number
local is_string = is_string
local is_table = is_table
local is_function = is_function
local is_thread = is_thread
local is_userdata = is_userdata
local function failure(name, usermsg, defaultmsg, ...)
if usermsg then usermsg = tostring(usermsg) end
local errobj = {
type = __failure__,
name = name,
msg = string_format(defaultmsg,...),
usermsg = usermsg
}
error(errobj, 0)
end
traceback_hide( failure )
local function _skip(name, usermsg, defaultmsg, ...)
if usermsg then usermsg = tostring(usermsg) end
local errobj = {
type = __skip__,
name = name,
msg = string_format(defaultmsg,...),
usermsg = usermsg
}
error(errobj, 0)
end
traceback_hide( _skip )
local function format_arg(arg)
local argtype = type(arg)
if argtype == "string" then
return "'"..arg.."'"
elseif argtype == "number" or argtype == "boolean" or argtype == "nil" then
return tostring(arg)
else
return "["..tostring(arg).."]"
end
end
local selected
do
local conv = {
["^"] = "%^",
["$"] = "%$",
["("] = "%(",
[")"] = "%)",
["%"] = "%%",
["."] = "%.",
["["] = "%[",
["]"] = "%]",
["+"] = "%+",
["-"] = "%-",
["?"] = ".",
["*"] = ".*"
}
local function lunitpat2luapat(str)
--return "^" .. string.gsub(str, "%W", conv) .. "$"
-- Above was very annoying, if I want to run all the tests having to do with
-- RSS, I want to be able to do "-t rss" not "-t \*rss\*".
return string_gsub(str, "%W", conv)
end
local function in_patternmap(map, name)
if map[name] == true then
return true
else
for _, pat in ipairs(map) do
if string_find(name, pat) then
return true
end
end
end
return false
end
selected = function (map, name)
if not map then
return true
end
local m = {}
for k,v in pairs(map) do
m[k] = lunitpat2luapat(v)
end
return in_patternmap(m, name)
end
end
function fail(msg)
stats.assertions = stats.assertions + 1
failure( "fail", msg, "failure" )
end
traceback_hide( fail )
function skip(msg)
stats.assertions = stats.assertions + 1
_skip( "skip", msg, "skip" )
end
traceback_hide( skip )
function assert(assertion, ...)
stats.assertions = stats.assertions + 1
if not assertion then
failure( "assert", (...), "assertion failed" )
end
return assertion, ...
end
traceback_hide( assert )
function assert_true(actual, ...)
stats.assertions = stats.assertions + 1
if actual ~= true then
failure( "assert_true", (...), "true expected but was %s", format_arg(actual) )
end
return actual, ...
end
traceback_hide( assert_true )
function assert_false(actual, ...)
stats.assertions = stats.assertions + 1
if actual ~= false then
failure( "assert_false", (...), "false expected but was %s", format_arg(actual) )
end
return actual, ...
end
traceback_hide( assert_false )
function assert_equal(expected, actual, ...)
stats.assertions = stats.assertions + 1
if expected ~= actual then
failure( "assert_equal", (...), "expected %s but was %s", format_arg(expected), format_arg(actual) )
end
return actual, ...
end
traceback_hide( assert_equal )
function assert_not_equal(unexpected, actual, ...)
stats.assertions = stats.assertions + 1
if unexpected == actual then
failure( "assert_not_equal", (...), "%s not expected but was one", format_arg(unexpected) )
end
return actual, ...
end
traceback_hide( assert_not_equal )
function assert_match(pattern, actual, ...)
stats.assertions = stats.assertions + 1
if type(pattern) ~= "string" then
failure( "assert_match", (...), "expected a string as pattern but was %s", format_arg(pattern) )
end
if type(actual) ~= "string" then
failure( "assert_match", (...), "expected a string to match pattern '%s' but was a %s", pattern, format_arg(actual) )
end
if not string_find(actual, pattern) then
failure( "assert_match", (...), "expected '%s' to match pattern '%s' but doesn't", actual, pattern )
end
return actual, ...
end
traceback_hide( assert_match )
function assert_not_match(pattern, actual, ...)
stats.assertions = stats.assertions + 1
if type(pattern) ~= "string" then
failure( "assert_not_match", (...), "expected a string as pattern but was %s", format_arg(pattern) )
end
if type(actual) ~= "string" then
failure( "assert_not_match", (...), "expected a string to not match pattern '%s' but was %s", pattern, format_arg(actual) )
end
if string_find(actual, pattern) then
failure( "assert_not_match", (...), "expected '%s' to not match pattern '%s' but it does", actual, pattern )
end
return actual, ...
end
traceback_hide( assert_not_match )
function assert_error(msg, func)
stats.assertions = stats.assertions + 1
if func == nil then
func, msg = msg, nil
end
if type(func) ~= "function" then
failure( "assert_error", msg, "expected a function as last argument but was %s", format_arg(func) )
end
local ok, errmsg = pcall(func)
if ok then
failure( "assert_error", msg, "error expected but no error occurred" )
end
end
traceback_hide( assert_error )
function assert_error_match(msg, pattern, func)
stats.assertions = stats.assertions + 1
if func == nil then
msg, pattern, func = nil, msg, pattern
end
if type(pattern) ~= "string" then
failure( "assert_error_match", msg, "expected the pattern as a string but was %s", format_arg(pattern) )
end
if type(func) ~= "function" then
failure( "assert_error_match", msg, "expected a function as last argument but was %s", format_arg(func) )
end
local ok, errmsg = pcall(func)
if ok then
failure( "assert_error_match", msg, "error expected but no error occurred" )
end
if type(errmsg) ~= "string" then
failure( "assert_error_match", msg, "error as string expected but was %s", format_arg(errmsg) )
end
if not string_find(errmsg, pattern) then
failure( "assert_error_match", msg, "expected error '%s' to match pattern '%s' but doesn't", errmsg, pattern )
end
end
traceback_hide( assert_error_match )
function assert_pass(msg, func)
stats.assertions = stats.assertions + 1
if func == nil then
func, msg = msg, nil
end
if type(func) ~= "function" then
failure( "assert_pass", msg, "expected a function as last argument but was %s", format_arg(func) )
end
local ok, errmsg = pcall(func)
if not ok then
failure( "assert_pass", msg, "no error expected but error was: '%s'", errmsg )
end
end
traceback_hide( assert_pass )
-- lunit.assert_typename functions
for _, typename in ipairs(typenames) do
local assert_typename = "assert_"..typename
lunit[assert_typename] = function(actual, ...)
stats.assertions = stats.assertions + 1
if type(actual) ~= typename then
failure( assert_typename, (...), "%s expected but was %s", typename, format_arg(actual) )
end
return actual, ...
end
traceback_hide( lunit[assert_typename] )
end
-- lunit.assert_not_typename functions
for _, typename in ipairs(typenames) do
local assert_not_typename = "assert_not_"..typename
lunit[assert_not_typename] = function(actual, ...)
stats.assertions = stats.assertions + 1
if type(actual) == typename then
failure( assert_not_typename, (...), typename.." not expected but was one" )
end
return actual, ...
end
traceback_hide( lunit[assert_not_typename] )
end
function lunit.clearstats()
stats = {
assertions = 0;
passed = 0;
failed = 0;
errors = 0;
skipped = 0;
}
end
local report, reporterrobj
do
local testrunner
function lunit.setrunner(newrunner)
if not ( is_table(newrunner) or is_nil(newrunner) ) then
return error("lunit.setrunner: Invalid argument", 0)
end
local oldrunner = testrunner
testrunner = newrunner
return oldrunner
end
function lunit.loadrunner(name)
if not is_string(name) then
return error("lunit.loadrunner: Invalid argument", 0)
end
local ok, runner = pcall( require, name )
if not ok then
return error("lunit.loadrunner: Can't load test runner: "..runner, 0)
end
return setrunner(runner)
end
function lunit.getrunner()
return testrunner
end
function report(event, ...)
local f = testrunner and testrunner[event]
if is_function(f) then
pcall(f, ...)
end
end
function reporterrobj(context, tcname, testname, errobj)
local fullname = tcname .. "." .. testname
if context == "setup" then
fullname = fullname .. ":" .. setupname(tcname, testname)
elseif context == "teardown" then
fullname = fullname .. ":" .. teardownname(tcname, testname)
end
if errobj.type == __failure__ then
stats.failed = stats.failed + 1
report("fail", fullname, errobj.where, errobj.msg, errobj.usermsg)
elseif errobj.type == __skip__ then
stats.skipped = stats.skipped + 1
report("skip", fullname, errobj.where, errobj.msg, errobj.usermsg)
else
stats.errors = stats.errors + 1
report("err", fullname, errobj.msg, errobj.tb)
end
end
end
local function key_iter(t, k)
return (next(t,k))
end
local testcase
do
-- Array with all registered testcases
local _testcases = {}
-- Marks a module as a testcase.
-- Applied over a module from module("xyz", lunit.testcase).
function lunit.testcase(m)
orig_assert( is_table(m) )
--orig_assert( m._M == m )
orig_assert( is_string(m._NAME) )
--orig_assert( is_string(m._PACKAGE) )
-- Register the module as a testcase
_testcases[m._NAME] = m
-- Import lunit, fail, assert* and is_* function to the module/testcase
m.lunit = lunit
m.fail = lunit.fail
m.skip = lunit.skip
for funcname, func in pairs(lunit) do
if "assert" == string_sub(funcname, 1, 6) or "is_" == string_sub(funcname, 1, 3) then
m[funcname] = func
end
end
end
function lunit.module(name,seeall)
local m = {}
if seeall == "seeall" then
setmetatable(m, { __index = _G })
end
m._NAME = name
lunit.testcase(m)
return m
end
-- Iterator (testcasename) over all Testcases
function lunit.testcases()
-- Make a copy of testcases to prevent confusing the iterator when
-- new testcase are defined
local _testcases2 = {}
for k,v in pairs(_testcases) do
_testcases2[k] = true
end
return key_iter, _testcases2, nil
end
function testcase(tcname)
return _testcases[tcname]
end
end
do
-- Finds a function in a testcase case insensitive
local function findfuncname(tcname, name)
for key, value in pairs(testcase(tcname)) do
if is_string(key) and is_function(value) and string_lower(key) == name then
return key
end
end
end
function lunit.setupname(tcname)
return findfuncname(tcname, "setup")
end
function lunit.teardownname(tcname)
return findfuncname(tcname, "teardown")
end
-- Iterator over all test names in a testcase.
-- Have to collect the names first in case one of the test
-- functions creates a new global and throws off the iteration.
function lunit.tests(tcname)
local testnames = {}
for key, value in pairs(testcase(tcname)) do
if is_string(key) and is_function(value) then
local lfn = string_lower(key)
if string_sub(lfn, 1, 4) == "test" or string_sub(lfn, -4) == "test" then
testnames[key] = true
end
end
end
return key_iter, testnames, nil
end
end
function lunit.runtest(tcname, testname)
orig_assert( is_string(tcname) )
orig_assert( is_string(testname) )
if (not getrunner()) then
loadrunner("lunit.console")
end
local function callit(context, func)
if func then
local err = mypcall(func)
if err then
reporterrobj(context, tcname, testname, err)
return false
end
end
return true
end
traceback_hide(callit)
report("run", tcname, testname)
local tc = testcase(tcname)
local setup = tc[setupname(tcname)]
local test = tc[testname]
local teardown = tc[teardownname(tcname)]
local setup_ok = callit( "setup", setup )
local test_ok = setup_ok and callit( "test", test )
local teardown_ok = setup_ok and callit( "teardown", teardown )
if setup_ok and test_ok and teardown_ok then
stats.passed = stats.passed + 1
report("pass", tcname, testname)
end
end
traceback_hide(runtest)
function lunit.run(testpatterns)
clearstats()
if (not getrunner()) then
loadrunner("lunit.console")
end
report("begin")
for testcasename in lunit.testcases() do
-- Run tests in the testcases
for testname in lunit.tests(testcasename) do
if selected(testpatterns, testname) then
runtest(testcasename, testname)
end
end
end
report("done")
return stats
end
traceback_hide(run)
function lunit.loadonly()
clearstats()
report("begin")
report("done")
return stats
end
-- Called from 'lunit' shell script.
function main(argv)
argv = argv or {}
-- FIXME: Error handling and error messages aren't nice.
local function checkarg(optname, arg)
if not is_string(arg) then
return error("lunit.main: option "..optname..": argument missing.", 0)
end
end
local function loadtestcase(filename)
if not is_string(filename) then
return error("lunit.main: invalid argument")
end
if #filename==0 then return end
local chunk, err = loadfile(filename)
if err then
return error(err)
else
chunk()
end
end
local testpatterns = nil
local doloadonly = false
local i = 0
while i < #argv do
i = i + 1
local arg = argv[i]
if arg == "--loadonly" then
doloadonly = true
elseif arg == "--runner" or arg == "-r" then
local optname = arg; i = i + 1; arg = argv[i]
checkarg(optname, arg)
loadrunner(arg)
elseif arg == "--test" or arg == "-t" then
local optname = arg; i = i + 1; arg = argv[i]
checkarg(optname, arg)
testpatterns = testpatterns or {}
testpatterns[#testpatterns+1] = arg
elseif arg == "--help" or arg == "-h" then
print[[
lunit 0.8
Copyright (c) 2004-2010 Michael Roth <mroth@nessie.de>
Copyright (c) 2011-2015 Doug Currie
This program comes WITHOUT WARRANTY OF ANY KIND.
Usage: lua test [OPTIONS] [--] scripts
Options:
-r, --runner RUNNER Testrunner to use, defaults to 'lunit-console'.
-t, --test PATTERN Which tests to run, may contain * or ? wildcards.
--loadonly Only load the tests.
-h, --help Print this help screen.
Please report bugs to <mroth@nessie.de>.
]]
return
elseif arg == "--" then
while i < #argv do
i = i + 1; arg = argv[i]
loadtestcase(arg)
end
else
loadtestcase(arg)
end
end
if doloadonly then
return loadonly()
else
return run(testpatterns)
end
end
clearstats()
return lunit