1
0
mirror of https://github.com/SquidDev-CC/CC-Tweaked synced 2024-06-21 04:40:02 +00:00

Support __name inside cc.expect

See #1355
This commit is contained in:
Jonathan Coates 2023-03-28 08:55:29 +01:00
parent 5bb2e8e8cd
commit ca279d410a
No known key found for this signature in database
GPG Key ID: B9E431FF07C98D06
2 changed files with 25 additions and 0 deletions

View File

@ -42,6 +42,21 @@ local function get_type_names(...)
return table.concat(types, ", ", 1, #types - 1) .. " or " .. types[#types]
end
end
local function get_display_type(value, t)
-- Lua is somewhat inconsistent in whether it obeys __name just for values which
-- have a per-instance metatable (so tables/userdata) or for everything. We follow
-- Cobalt and only read the metatable for tables/userdata.
if t ~= "table" and t ~= "userdata" then return t end
local metatable = debug.getmetatable(value)
if not metatable then return t end
local name = rawget(metatable, "__name")
if type(name) == "string" then return name else return t end
end
--- Expect an argument to have a specific type.
--
-- @tparam number index The 1-based argument index.
@ -60,6 +75,8 @@ local function expect(index, value, ...)
local ok, info = pcall(debug.getinfo, 3, "nS")
if ok and info.name and info.name ~= "" and info.what ~= "C" then name = info.name end
t = get_display_type(value, t)
local type_names = get_type_names(...)
if name then
error(("bad argument #%d to '%s' (expected %s, got %s)"):format(index, name, type_names, t), 3)
@ -85,6 +102,8 @@ local function field(tbl, index, ...)
if t == native_select(i, ...) then return value end
end
t = get_display_type(value, t)
if value == nil then
error(("field '%s' missing from table"):format(index), 3)
else

View File

@ -33,6 +33,12 @@ describe("cc.expect", function()
expect.error(trampoline):str_match("^[^:]*expect_spec.lua:31: bad argument #1 to 'worker' %(expected string, got nil%)$")
end)
it("supports custom type names", function()
local value = setmetatable({}, { __name = "some type" })
expect.error(e.expect, 1, value, "string"):eq("bad argument #1 (expected string, got some type)")
end)
end)
describe("field", function()