view tests/unit/test_plugin_xep_0339.py @ 4095:684ba556a617

core (memory/sqla_mapping): fix legacy pickled values: folloing packages refactoring, legacy pickled values could not be unpickled (due to use of old classes). This temporary workaround fix it, but the right thing to do will be to move from pickle to JSON at some point.
author Goffi <goffi@goffi.org>
date Mon, 12 Jun 2023 14:57:27 +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]