mirrors/prosody
1
0
Fork 0
mirror of https://github.com/bjc/prosody.git synced 2025-04-04 13:47:41 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

util.stanza: Check that argument to reply is a stanza

Browse source
This commit is contained in:
Kim Alvefur 2019-11-25 20:44:05 +01:00
parent 10aa40227b
commit b340e5e462
2 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
spec/util_stanza_spec.lua
View file

@ -170,6 +170,12 @@ describe("util.stanza", function()
assert.are.equal(r.attr.type, "result"); assert.are.equal(r.attr.type, "result");
assert.are.equal(#r.tags, 0, "A reply should not include children of the original stanza"); assert.are.equal(#r.tags, 0, "A reply should not include children of the original stanza");
end); end);
it("should reject not-stanzas", function ()
assert.has.error_match(function ()
st.reply(not "a stanza");
end, "expected stanza");
end);
end); end);
describe("#error_reply()", function() describe("#error_reply()", function()

3
util/stanza.lua
View file

@ -434,6 +434,9 @@ local function iq(attr)
end end
local function reply(orig) local function reply(orig)
if not is_stanza(orig) then
error("bad argument to reply: expected stanza, got "..type(orig));
end
return new_stanza(orig.name, return new_stanza(orig.name,
orig.attr and { orig.attr and {
to = orig.attr.from, to = orig.attr.from,