53 lines
1.1 KiB
Lua
53 lines
1.1 KiB
Lua
|
local mock = mkmodule('test_util.mock')
|
||
|
|
||
|
|
||
|
--[[
|
||
|
Usage:
|
||
|
patch(table, key, value, callback)
|
||
|
patch({
|
||
|
[{table, key}] = value,
|
||
|
[{table2, key2}] = value2
|
||
|
}, callback)
|
||
|
]]
|
||
|
function mock.patch(...)
|
||
|
local args = {...}
|
||
|
if #args == 4 then
|
||
|
args = {{
|
||
|
[{args[1], args[2]}] = args[3]
|
||
|
}, args[4]}
|
||
|
end
|
||
|
if #args ~= 2 then
|
||
|
error('expected 2 or 4 arguments')
|
||
|
end
|
||
|
|
||
|
local callback = args[2]
|
||
|
local patches = {}
|
||
|
for k, v in pairs(args[1]) do
|
||
|
local p = {
|
||
|
table = k[1],
|
||
|
key = k[2],
|
||
|
new_value = v,
|
||
|
}
|
||
|
p.old_value = p.table[p.key]
|
||
|
-- no-op to ensure that the value can be restored by the finalizer below
|
||
|
p.table[p.key] = p.table[p.key]
|
||
|
table.insert(patches, p)
|
||
|
end
|
||
|
|
||
|
dfhack.with_finalize(
|
||
|
function()
|
||
|
for _, p in ipairs(patches) do
|
||
|
p.table[p.key] = p.old_value
|
||
|
end
|
||
|
end,
|
||
|
function()
|
||
|
for _, p in ipairs(patches) do
|
||
|
p.table[p.key] = p.new_value
|
||
|
end
|
||
|
callback()
|
||
|
end
|
||
|
)
|
||
|
end
|
||
|
|
||
|
return mock
|