/usr/share/lua/5.1/luassert/stub.lua is in lua-luassert 1.7.9-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 | -- module will return a stub module table
local assert = require 'luassert.assert'
local spy = require 'luassert.spy'
local util = require 'luassert.util'
local unpack = require 'luassert.compatibility'.unpack
local stub = {}
function stub.new(object, key, ...)
if object == nil and key == nil then
-- called without arguments, create a 'blank' stub
object = {}
key = ""
end
local return_values_count = select("#", ...)
local return_values = {...}
assert(type(object) == "table" and key ~= nil, "stub.new(): Can only create stub on a table key, call with 2 params; table, key", util.errorlevel())
assert(object[key] == nil or util.callable(object[key]), "stub.new(): The element for which to create a stub must either be callable, or be nil", util.errorlevel())
local old_elem = object[key] -- keep existing element (might be nil!)
local fn = (return_values_count == 1 and util.callable(return_values[1]) and return_values[1])
local defaultfunc = fn or function()
return unpack(return_values, 1, return_values_count)
end
local oncalls = {}
local callbacks = {}
local stubfunc = function(...)
local args = {...}
args.n = select('#', ...)
local match = util.matchargs(oncalls, args)
if match then
return callbacks[match](...)
end
return defaultfunc(...)
end
object[key] = stubfunc -- set the stubfunction
local s = spy.on(object, key) -- create a spy on top of the stub function
local spy_revert = s.revert -- keep created revert function
s.revert = function(self) -- wrap revert function to restore original element
if not self.reverted then
spy_revert(self)
object[key] = old_elem
self.reverted = true
end
return old_elem
end
s.returns = function(...)
local return_args = {...}
local n = select('#', ...)
defaultfunc = function()
return unpack(return_args, 1, n)
end
return s
end
s.invokes = function(func)
defaultfunc = function(...)
return func(...)
end
return s
end
s.by_default = {
returns = s.returns,
invokes = s.invokes,
}
s.on_call_with = function(...)
local match_args = {...}
match_args.n = select('#', ...)
match_args = util.copyargs(match_args)
return {
returns = function(...)
local return_args = {...}
local n = select('#', ...)
table.insert(oncalls, match_args)
callbacks[match_args] = function()
return unpack(return_args, 1, n)
end
return s
end,
invokes = function(func)
table.insert(oncalls, match_args)
callbacks[match_args] = function(...)
return func(...)
end
return s
end
}
end
return s
end
local function set_stub(state, arguments)
state.payload = arguments[1]
state.failure_message = arguments[2]
end
assert:register("modifier", "stub", set_stub)
return setmetatable(stub, {
__call = function(self, ...)
-- stub originally was a function only. Now that it is a module table
-- the __call method is required for backward compatibility
return stub.new(...)
end
})
|