# HG changeset patch
# User Goffi <goffi@goffi.org>
# Date 1685648251 -7200
# Node ID 5241267a92b3543148b329b9757d2bce237f21f6
# Parent  3f62c2445df1770692402cdd9155b99f7a5d493e
tests (units): tests for plugin XEP-0339:

fix 439

diff -r 3f62c2445df1 -r 5241267a92b3 tests/unit/test_plugin_xep_0339.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/unit/test_plugin_xep_0339.py	Thu Jun 01 21:37:31 2023 +0200
@@ -0,0 +1,156 @@
+#!/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 sat.plugins.plugin_xep_0339 import NS_JINGLE_RTP_SSMA, XEP_0339
+from sat.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]