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

util.stanza: Add :get_child_with_attr() + tests

Browse source
This commit is contained in:
Matthew Wild 2021-09-12 10:31:02 +01:00
parent bee45656d7
commit eba0bacfda
2 changed files with 28 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

20
spec/util_stanza_spec.lua
View file

@ -455,6 +455,26 @@ describe("util.stanza", function()
end);
end);
describe("get_child_with_attr", function ()
local s = st.message({ type = "chat" })
:text_tag("body", "Hello world", { ["xml:lang"] = "en" })
:text_tag("body", "Bonjour le monde", { ["xml:lang"] = "fr" })
:text_tag("body", "Hallo Welt", { ["xml:lang"] = "de" })
it("works", function ()
assert.equal(s:get_child_with_attr("body", nil, "xml:lang", "en"):get_text(), "Hello world");
assert.equal(s:get_child_with_attr("body", nil, "xml:lang", "de"):get_text(), "Hallo Welt");
assert.equal(s:get_child_with_attr("body", nil, "xml:lang", "fr"):get_text(), "Bonjour le monde");
assert.is_nil(s:get_child_with_attr("body", nil, "xml:lang", "FR"));
assert.is_nil(s:get_child_with_attr("body", nil, "xml:lang", "es"));
end);
it("supports normalization", function ()
assert.equal(s:get_child_with_attr("body", nil, "xml:lang", "EN", string.upper):get_text(), "Hello world");
assert.is_nil(s:get_child_with_attr("body", nil, "xml:lang", "ES", string.upper));
end);
end);
describe("#clone", function ()
it("works", function ()
local s = st.message({type="chat"}, "Hello"):reset();

8
util/stanza.lua
View file

@ -191,6 +191,14 @@ function stanza_mt:child_with_ns(ns)
end
end
function stanza_mt:get_child_with_attr(name, xmlns, attr_name, attr_value, normalize)
for tag in self:childtags(name, xmlns) do
if (normalize and normalize(tag.attr[attr_name]) or tag.attr[attr_name]) == attr_value then
return tag;
end
end
end
function stanza_mt:children()
local i = 0;
return function (a)