Mercurial > libervia-backend
view tests/unit/test_plugin_xep_0339.py @ 4318:27bb22eace65
tests (unit/email gateway): add test for XEP-0131 handling:
rel 451
author | Goffi <goffi@goffi.org> |
---|---|
date | Sat, 28 Sep 2024 15:59:48 +0200 |
parents | 4b842c1fb686 |
children |
line wrap: on
line source
#!/usr/bin/env python3 # Libervia: an XMPP client # Copyright (C) 2009-2023 Jérôme Poisson (goffi@goffi.org) # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Affero General Public License for more details. # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. from twisted.words.xish import domish from libervia.backend.plugins.plugin_xep_0339 import NS_JINGLE_RTP_SSMA, XEP_0339 from libervia.backend.tools.xml_tools import parse class TestXEP0339: def test_parse_sdp(self, host): """'ssrc' and 'ssrc-group' attributes in SDP are correctly parsed""" xep_0339 = XEP_0339(host) call_data = {} metadata = {} media_type = "video" application_data = {} transport_data = {} # SDP: a=ssrc:123 label:stream_label attribute = "ssrc" parts = ["123", "label:stream_label"] xep_0339._parse_sdp_a_trigger( attribute, parts, call_data, metadata, media_type, application_data, transport_data, ) assert application_data == {"ssrc": {123: {"label": "stream_label"}}} application_data = {} # SDP: a=ssrc-group:FID 123 456 attribute = "ssrc-group" parts = ["FID", "123", "456"] xep_0339._parse_sdp_a_trigger( attribute, parts, call_data, metadata, media_type, application_data, transport_data, ) assert application_data == {"ssrc-group": {"FID": [123, 456]}} def test_generate_sdp(self, host): """'ssrc' and 'ssrc-group' attributes in SDP are correctly generated""" xep_0339 = XEP_0339(host) media_data = { "ssrc": {123: {"label": "stream_label"}}, "ssrc-group": {"FID": [123, 456]}, } sdp_lines = [] xep_0339._generate_sdp_content_trigger( {}, True, 1, {}, sdp_lines, {}, "", media_data, "video" ) assert sdp_lines == ["a=ssrc:123 label:stream_label", "a=ssrc-group:FID 123 456"] def test_parse_description(self, host): """'<source>' and '<ssrc-group>' elements are correctly parsed""" xep_0339 = XEP_0339(host) media_data = {} raw_xml = """ <description xmlns='urn:xmpp:jingle:apps:rtp:1' media='audio'> <source xmlns='urn:xmpp:jingle:apps:rtp:ssma:0' ssrc='1'> <parameter name='cname' value='some_name'/> <parameter name='msid' value='media'/> </source> <ssrc-group xmlns='urn:xmpp:jingle:apps:rtp:ssma:0' semantics='FEC'> <source ssrc='2'/> <source ssrc='3'/> </ssrc-group> </description> """ desc_elt = parse(raw_xml) xep_0339._parse_description_trigger(desc_elt, media_data) assert media_data == { "msid": "media", "ssrc": { 1: { "cname": "some_name", "msid": "media", } }, "ssrc-group": {"FEC": [2, 3]}, } def test_generate_description(self, host): """'<source>' and '<ssrc-group>' elements are correctly generated""" xep_0339 = XEP_0339(host) media_data = { "ssrc": { 1: { "cname": "some_name", "msid": "media", } }, "ssrc-group": {"FEC": [2, 3]}, } session = {} desc_elt = domish.Element((None, "description")) xep_0339._build_description_trigger(desc_elt, media_data, session) source_elts = list(desc_elt.elements(NS_JINGLE_RTP_SSMA, "source")) assert len(source_elts) == 1 source_elt = source_elts[0] assert source_elt["ssrc"] == "1" param_elts = list(source_elt.elements(NS_JINGLE_RTP_SSMA, "parameter")) assert len(param_elts) == 2 assert param_elts[0]["name"] == "cname" assert param_elts[0]["value"] == "some_name" assert param_elts[1]["name"] == "msid" assert param_elts[1]["value"] == "media" ssrc_group_elts = list(desc_elt.elements(NS_JINGLE_RTP_SSMA, "ssrc-group")) assert len(ssrc_group_elts) == 1 ssrc_group_elt = ssrc_group_elts[0] assert ssrc_group_elt["semantics"] == "FEC" ssrc_ids = [ int(source_elt["ssrc"]) for source_elt in ssrc_group_elt.elements(NS_JINGLE_RTP_SSMA, "source") ] assert ssrc_ids == [2, 3]