Mercurial > libervia-backend
view libervia/backend/test/test_plugin_xep_0297.py @ 4212:5f2d496c633f
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/>. """ Plugin XEP-0297 """ from .constants import Const as C from libervia.backend.test import helpers from libervia.backend.plugins.plugin_xep_0203 import XEP_0203 from libervia.backend.plugins.plugin_xep_0297 import XEP_0297 from twisted.words.protocols.jabber.jid import JID from dateutil.tz import tzutc import datetime from wokkel.generic import parseXml NS_PUBSUB = "http://jabber.org/protocol/pubsub" class XEP_0297Test(helpers.SatTestCase): def setUp(self): self.host = helpers.FakeSAT() self.plugin = XEP_0297(self.host) self.host.plugins["XEP-0203"] = XEP_0203(self.host) def test_delay(self): stanza = parseXml( """ <message from='juliet@capulet.lit/orchard' id='0202197' to='romeo@montague.lit' type='chat'> <body>Yet I should kill thee with much cherishing.</body> <mood xmlns='http://jabber.org/protocol/mood'> <amorous/> </mood> </message> """.encode( "utf-8" ) ) output = """ <message to='mercutio@verona.lit' type='chat'> <body>A most courteous exposition!</body> <forwarded xmlns='urn:xmpp:forward:0'> <delay xmlns='urn:xmpp:delay' stamp='2010-07-10T23:08:25Z'/> <message from='juliet@capulet.lit/orchard' id='0202197' to='romeo@montague.lit' type='chat' xmlns='jabber:client'> <body>Yet I should kill thee with much cherishing.</body> <mood xmlns='http://jabber.org/protocol/mood'> <amorous/> </mood> </message> </forwarded> </message> """ stamp = datetime.datetime(2010, 7, 10, 23, 8, 25, tzinfo=tzutc()) d = self.plugin.forward( stanza, JID("mercutio@verona.lit"), stamp, body="A most courteous exposition!", profile_key=C.PROFILE[0], ) d.addCallback( lambda __: self.assert_equal_xml( self.host.get_sent_message_xml(0), output, True ) ) return d