156 lines
5 KiB
Python
156 lines
5 KiB
Python
#!/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]
|