1
0
mirror of https://github.com/Refactorio/RedMew.git synced 2024-12-14 10:13:13 +02:00
RedMew/utils/test/module_store.lua
2020-10-03 21:11:15 +01:00

146 lines
3.7 KiB
Lua

local Public = {}
local function new_module(module_name)
return {
id = nil,
name = module_name,
parent = nil,
children = {},
startup_func = nil,
startup_steps = nil,
startup_current_step = nil,
startup_error = nil,
teardown_func = nil,
teardown_steps = nil,
teardown_current_step = nil,
teardown_error = nil,
test_funcs = {},
tests = nil,
is_open = false,
depth = nil,
count = nil,
passed = nil
}
end
local root_module = new_module(nil)
root_module.is_open = true
Public.root_module = root_module
local parent_module = nil
local function add_module(module_name, module_func, parent)
local parent_children = parent.children
local module = parent_children[module_name]
if not module then
module = new_module(module_name)
parent_children[module_name] = module
module.parent = parent_module
end
parent_module = module
module_func()
end
local function no_op()
end
local function add_module_range(modules_names, module_func, parent)
for i = 1, #modules_names - 1 do
local name = modules_names[i]
add_module(name, no_op, parent)
parent = parent_module
end
add_module(modules_names[#modules_names], module_func, parent)
end
function Public.module(module_name, module_func)
local module_name_type = type(module_name)
if module_name_type ~= 'string' and module_name_type ~= 'table' then
error('module_name must be of type string or array of strings.', 2)
end
if module_name_type == 'table' and #module_name == 0 then
error('when module_name is array must be non empty.', 2)
end
if type(module_func) ~= 'function' then
error('module_func must be of type function.', 2)
end
local old_parent = parent_module
local parent = parent_module or root_module
if module_name_type == 'string' then
add_module(module_name, module_func, parent)
else
add_module_range(module_name, module_func, parent)
end
parent_module = old_parent
end
function Public.test(test_name, test_func)
if not parent_module then
error('test can not be declared outisde of a module.', 2)
end
if type(test_name) ~= 'string' then
error('test_name must be of type string.', 2)
end
if type(test_func) ~= 'function' then
error('test_func must be of type function.', 2)
end
local test_funcs = parent_module.test_funcs
if test_funcs[test_name] then
error(
table.concat {
"test '",
test_name,
"' already exists, can not have duplicate test names in the same module."
},
2
)
end
test_funcs[test_name] = test_func
end
function Public.module_startup(startup_func)
if type(startup_func) ~= 'function' then
error('startup_func must be of type function.', 2)
end
if parent_module == nil then
error('root module can not have startup_func.', 2)
end
if parent_module.startup_func ~= nil then
error('startup_func can not be declared twice for the same module.', 2)
end
parent_module.startup_func = startup_func
end
function Public.module_teardown(teardown_func)
if type(teardown_func) ~= 'function' then
error('teardown_func must be of type function.', 2)
end
if parent_module == nil then
error('root module can not have teardown_func.', 2)
end
if parent_module.teardown_func ~= nil then
error('teardown_func can not be declared twice for the same module.', 2)
end
parent_module.teardown_func = teardown_func
end
return Public