Mercurial > libervia-backend

view libervia/backend/test/test_helpers_plugins.py @ 4212:5f2d496c633f

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
core: get rid of `pickle`: Use of `pickle` to serialise data was a technical legacy that was causing trouble to store in database, to update (if a class was serialised, a change could break update), and to security (pickle can lead to code execution). This patch remove all use of Pickle in favour in JSON, notably: - for caching data, a Pydantic model is now used instead - for SQLAlchemy model, the LegacyPickle is replaced by JSON serialisation - in XEP-0373 a class `PublicKeyMetadata` was serialised. New method `from_dict` and `to_dict` method have been implemented to do serialisation. - new methods to (de)serialise data can now be specified with Identity data types. It is notably used to (de)serialise `path` of avatars. A migration script has been created to convert data (for upgrade or downgrade), with special care for XEP-0373 case. Depending of size of database, this migration script can be long to run. rel 443
author Goffi <goffi@goffi.org>
date Fri, 23 Feb 2024 13:31:04 +0100
parents 4b842c1fb686
children
line wrap: on
line source

#!/usr/bin/env python3


# SAT: a jabber client
# Copyright (C) 2009-2021 Jérôme Poisson (goffi@goffi.org)
# Copyright (C) 2013-2016 Adrien Cossa (souliane@mailoo.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/>.

""" Test the helper classes to see if they behave well"""

from libervia.backend.test import helpers
from libervia.backend.test import helpers_plugins


class FakeXEP_0045Test(helpers.SatTestCase):
    def setUp(self):
        self.host = helpers.FakeSAT()
        self.plugin = helpers_plugins.FakeXEP_0045(self.host)

    def test_join_room(self):
        self.plugin.join_room(0, 0)
        self.assertEqual("test", self.plugin.get_nick(0, 0))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 0, 0))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 1, 0))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 2, 0))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 3, 0))
        self.assertEqual("", self.plugin.get_nick(0, 1))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 0, 1))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 1, 1))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 2, 1))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 3, 1))
        self.assertEqual("", self.plugin.get_nick(0, 2))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 0, 2))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 1, 2))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 2, 2))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 3, 2))
        self.assertEqual("", self.plugin.get_nick(0, 3))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 0, 3))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 1, 3))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 2, 3))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 3, 3))
        self.plugin.join_room(0, 1)
        self.assertEqual("test", self.plugin.get_nick(0, 0))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 0, 0))
        self.assertEqual("sender", self.plugin.get_nick_of_user(0, 1, 0))
        self.assertEqual("sender", self.plugin.get_nick_of_user(0, 2, 0))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 3, 0))
        self.assertEqual("sender", self.plugin.get_nick(0, 1))
        self.assertEqual("test", self.plugin.get_nick_of_user(0, 0, 1))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 1, 1))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 2, 1))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 3, 1))
        self.assertEqual("", self.plugin.get_nick(0, 2))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 0, 2))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 1, 2))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 2, 2))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 3, 2))
        self.assertEqual("", self.plugin.get_nick(0, 3))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 0, 3))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 1, 3))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 2, 3))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 3, 3))
        self.plugin.join_room(0, 2)
        self.assertEqual("test", self.plugin.get_nick(0, 0))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 0, 0))
        self.assertEqual("sender", self.plugin.get_nick_of_user(0, 1, 0))
        self.assertEqual("sender", self.plugin.get_nick_of_user(0, 2, 0))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 3, 0))
        self.assertEqual("sender", self.plugin.get_nick(0, 1))
        self.assertEqual("test", self.plugin.get_nick_of_user(0, 0, 1))
        self.assertEqual(
            "sender", self.plugin.get_nick_of_user(0, 1, 1)
        )  # Const.JID[2] is in the roster for Const.PROFILE[1]
        self.assertEqual("sender", self.plugin.get_nick_of_user(0, 2, 1))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 3, 1))
        self.assertEqual("sender", self.plugin.get_nick(0, 2))
        self.assertEqual("test", self.plugin.get_nick_of_user(0, 0, 2))
        self.assertEqual("sender", self.plugin.get_nick_of_user(0, 1, 2))
        self.assertEqual(
            "sender", self.plugin.get_nick_of_user(0, 2, 2)
        )  # Const.JID[1] is in the roster for Const.PROFILE[2]
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 3, 2))
        self.assertEqual("", self.plugin.get_nick(0, 3))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 0, 3))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 1, 3))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 2, 3))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 3, 3))
        self.plugin.join_room(0, 3)
        self.assertEqual("test", self.plugin.get_nick(0, 0))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 0, 0))
        self.assertEqual("sender", self.plugin.get_nick_of_user(0, 1, 0))
        self.assertEqual("sender", self.plugin.get_nick_of_user(0, 2, 0))
        self.assertEqual("sender_", self.plugin.get_nick_of_user(0, 3, 0))
        self.assertEqual("sender", self.plugin.get_nick(0, 1))
        self.assertEqual("test", self.plugin.get_nick_of_user(0, 0, 1))
        self.assertEqual(
            "sender", self.plugin.get_nick_of_user(0, 1, 1)
        )  # Const.JID[2] is in the roster for Const.PROFILE[1]
        self.assertEqual("sender", self.plugin.get_nick_of_user(0, 2, 1))
        self.assertEqual("sender_", self.plugin.get_nick_of_user(0, 3, 1))
        self.assertEqual("sender", self.plugin.get_nick(0, 2))
        self.assertEqual("test", self.plugin.get_nick_of_user(0, 0, 2))
        self.assertEqual("sender", self.plugin.get_nick_of_user(0, 1, 2))
        self.assertEqual(
            "sender", self.plugin.get_nick_of_user(0, 2, 2)
        )  # Const.JID[1] is in the roster for Const.PROFILE[2]
        self.assertEqual("sender_", self.plugin.get_nick_of_user(0, 3, 2))
        self.assertEqual("sender_", self.plugin.get_nick(0, 3))
        self.assertEqual("test", self.plugin.get_nick_of_user(0, 0, 3))
        self.assertEqual("sender", self.plugin.get_nick_of_user(0, 1, 3))
        self.assertEqual("sender", self.plugin.get_nick_of_user(0, 2, 3))
        self.assertEqual(None, self.plugin.get_nick_of_user(0, 3, 3))