libervia-backend: tests/unit/test_plugin_xep

annotate tests/unit/test_plugin_xep_0420.py @ 4351:6a0a081485b8

plugin autocrypt: Autocrypt protocol implementation: Implementation of autocrypt: `autocrypt` header is checked, and if present and no public key is known for the peer, the key is imported. `autocrypt` header is also added to outgoing message (only if an email gateway is detected). For the moment, the JID is use as identifier, but the real email used by gateway should be used in the future. rel 456

author	Goffi <goffi@goffi.org>
date	Fri, 28 Feb 2025 09:23:35 +0100
parents	f1d0cde61af7
children

rev	line source
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	1 #!/usr/bin/env python3
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	2
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	3 # Tests for Libervia's Stanza Content Encryption plugin
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	4 # Copyright (C) 2022-2022 Tim Henkes (me@syndace.dev)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	5
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	6 # This program is free software: you can redistribute it and/or modify
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	7 # it under the terms of the GNU Affero General Public License as published by
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	8 # the Free Software Foundation, either version 3 of the License, or
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	9 # (at your option) any later version.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	10
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	11 # This program is distributed in the hope that it will be useful,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	14 # GNU Affero General Public License for more details.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	15
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	16 # You should have received a copy of the GNU Affero General Public License
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	18
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	19 from datetime import datetime, timezone
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3877 diff changeset	20 from typing import Callable, cast
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	21
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	22 import pytest
4071 4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 3933 diff changeset	23 from libervia.backend.core import exceptions
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	24
4071 4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 3933 diff changeset	25 from libervia.backend.plugins.plugin_xep_0334 import NS_HINTS
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 3933 diff changeset	26 from libervia.backend.plugins.plugin_xep_0420 import (
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	27 NS_SCE,
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	28 XEP_0420,
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	29 AffixVerificationFailed,
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	30 ProfileRequirementsNotMet,
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	31 SCEAffixPolicy,
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	32 SCECustomAffix,
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	33 SCEProfile,
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	34 )
4071 4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 3933 diff changeset	35 from libervia.backend.tools.xml_tools import ElementParser
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	36 from twisted.words.xish import domish
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	37
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	38
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	39 __all__ = [ # pylint: disable=unused-variable
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	40 "test_unpack_matches_original",
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	41 "test_affixes_included",
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	42 "test_all_affixes_verified",
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	43 "test_incomplete_affixes",
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	44 "test_rpad_affix",
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	45 "test_time_affix",
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	46 "test_to_affix",
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	47 "test_from_affix",
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	48 "test_custom_affixes",
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	49 "test_namespace_conversion",
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	50 "test_non_encryptable_elements",
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	51 "test_schema_validation",
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	52 ]
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	53
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	54
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	55 string_to_domish = cast(Callable[[str], domish.Element], ElementParser())
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	56
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	57
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	58 class CustomAffixImpl(SCECustomAffix):
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	59 """
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	60 A simple custom affix implementation for testing purposes. Verifies the full JIDs of
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	61 both sender and recipient.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	62
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	63 @warning: This is just an example, an affix element like this might not make sense due
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	64 to potentially allowed modifications of recipient/sender full JIDs (I don't know
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	65 enough about XMPP routing to know whether full JIDs are always left untouched by
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	66 the server).
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	67 """
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	68
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	69 @property
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	70 def element_name(self) -> str:
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	71 return "full-jids"
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	72
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	73 @property
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	74 def element_schema(self) -> str:
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	75 return """<xs:element name="full-jids">
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	76 <xs:complexType>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	77 <xs:attribute name="recipient" type="xs:string"/>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	78 <xs:attribute name="sender" type="xs:string"/>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	79 </xs:complexType>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	80 </xs:element>"""
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	81
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	82 def create(self, stanza: domish.Element) -> domish.Element:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3877 diff changeset	83 recipient = stanza.getAttribute("to", None)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3877 diff changeset	84 sender = stanza.getAttribute("from", None)
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	85
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	86 if recipient is None or sender is None:
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	87 raise ValueError(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	88 "Stanza doesn't have ``to`` and ``from`` attributes required by the"
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	89 " full-jids custom affix."
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	90 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	91
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	92 element = domish.Element((NS_SCE, "full-jids"))
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	93 element["recipient"] = recipient
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	94 element["sender"] = sender
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	95 return element
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	96
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	97 def verify(self, stanza: domish.Element, element: domish.Element) -> None:
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	98 recipient_target = element["recipient"]
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	99 recipient_actual = stanza.getAttribute("to")
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	100
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	101 sender_target = element["sender"]
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	102 sender_actual = stanza.getAttribute("from")
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	103
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	104 if recipient_actual != recipient_target or sender_actual != sender_target:
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	105 raise AffixVerificationFailed(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	106 f"Full JIDs differ. Recipient: actual={recipient_actual} vs."
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	107 f" target={recipient_target}; Sender: actual={sender_actual} vs."
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	108 f" target={sender_target}"
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	109 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	110
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	111
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	112 def test_unpack_matches_original() -> None: # pylint: disable=missing-function-docstring
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	113 profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	114 SCEAffixPolicy.REQUIRED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	115 SCEAffixPolicy.OPTIONAL,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	116 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	117 SCEAffixPolicy.OPTIONAL,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	118 custom_policies={CustomAffixImpl(): SCEAffixPolicy.NOT_NEEDED},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	119 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	120
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	121 stanza_string = (
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	122 '<message from="foo@example.com" to="bar@example.com"><body>Test with both a body'
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	123 ' and some other custom element.</body><custom xmlns="urn:xmpp:example:0"'
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	124 ' test="matches-original">some more content</custom></message>'
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	125 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	126
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	127 stanza = string_to_domish(stanza_string)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	128
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	129 envelope_serialized = XEP_0420.pack_stanza(profile, stanza)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	130
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	131 # The stanza should not have child elements any more
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	132 assert len(list(stanza.elements())) == 0
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	133
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	134 XEP_0420.unpack_stanza(profile, stanza, envelope_serialized)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	135
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	136 # domish.Element doesn't override __eq__, thus we compare the .toXml() strings here in
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	137 # the hope that serialization for an example as small as this is unique enough to be
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	138 # compared that way.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	139 assert stanza.toXml() == string_to_domish(stanza_string).toXml()
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	140
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	141
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	142 def test_affixes_included() -> None: # pylint: disable=missing-function-docstring
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	143 custom_affix = CustomAffixImpl()
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	144
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	145 profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	146 SCEAffixPolicy.REQUIRED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	147 SCEAffixPolicy.OPTIONAL,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	148 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	149 SCEAffixPolicy.OPTIONAL,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	150 custom_policies={custom_affix: SCEAffixPolicy.OPTIONAL},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	151 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	152
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	153 stanza = string_to_domish(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	154 """<message from="foo@example.com" to="bar@example.com">
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	155 <body>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	156 Make sure that both the REQUIRED and the OPTIONAL affixes are included.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	157 </body>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	158 </message>"""
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	159 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	160
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	161 affix_values = XEP_0420.unpack_stanza(
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	162 profile, stanza, XEP_0420.pack_stanza(profile, stanza)
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	163 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	164
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	165 assert affix_values.rpad is not None
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	166 assert affix_values.timestamp is not None
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	167 assert affix_values.recipient is None
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	168 assert affix_values.sender is not None
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	169 assert custom_affix in affix_values.custom
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	170
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	171
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	172 def test_all_affixes_verified() -> None: # pylint: disable=missing-function-docstring
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	173 packing_profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	174 SCEAffixPolicy.REQUIRED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	175 SCEAffixPolicy.REQUIRED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	176 SCEAffixPolicy.REQUIRED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	177 SCEAffixPolicy.REQUIRED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	178 custom_policies={},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	179 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	180
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	181 unpacking_profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	182 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	183 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	184 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	185 SCEAffixPolicy.NOT_NEEDED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	186 custom_policies={},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	187 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	188
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	189 stanza = string_to_domish(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	190 """<message from="foo@example.com" to="bar@example.com">
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	191 <body>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	192 When unpacking, all affixes are loaded, even those marked as NOT_NEEDED.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	193 </body>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	194 </message>"""
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	195 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	196
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	197 envelope_serialized = XEP_0420.pack_stanza(packing_profile, stanza)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	198
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	199 affix_values = XEP_0420.unpack_stanza(unpacking_profile, stanza, envelope_serialized)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	200
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	201 assert affix_values.rpad is not None
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	202 assert affix_values.timestamp is not None
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	203 assert affix_values.recipient is not None
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	204 assert affix_values.sender is not None
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	205
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	206 # When unpacking, all affixes are verified, even if they are NOT_NEEDED by the profile
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	207 stanza = string_to_domish(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	208 """<message from="fooo@example.com" to="baz@example.com"></message>"""
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	209 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	210
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	211 with pytest.raises(AffixVerificationFailed):
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	212 XEP_0420.unpack_stanza(unpacking_profile, stanza, envelope_serialized)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	213
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	214
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	215 def test_incomplete_affixes() -> None: # pylint: disable=missing-function-docstring
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	216 packing_profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	217 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	218 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	219 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	220 SCEAffixPolicy.NOT_NEEDED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	221 custom_policies={},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	222 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	223
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	224 unpacking_profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	225 SCEAffixPolicy.REQUIRED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	226 SCEAffixPolicy.REQUIRED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	227 SCEAffixPolicy.REQUIRED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	228 SCEAffixPolicy.REQUIRED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	229 custom_policies={},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	230 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	231
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	232 stanza = string_to_domish(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	233 """<message from="foo@example.com" to="bar@example.com">
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	234 <body>Check that all affixes REQUIRED by the profile are present.</body>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	235 </message>"""
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	236 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	237
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	238 with pytest.raises(ProfileRequirementsNotMet):
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	239 XEP_0420.unpack_stanza(
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	240 unpacking_profile, stanza, XEP_0420.pack_stanza(packing_profile, stanza)
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	241 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	242
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	243 # Do the same but with a custom affix missing
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	244 custom_affix = CustomAffixImpl()
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	245
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	246 packing_profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	247 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	248 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	249 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	250 SCEAffixPolicy.NOT_NEEDED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	251 custom_policies={custom_affix: SCEAffixPolicy.NOT_NEEDED},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	252 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	253
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	254 unpacking_profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	255 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	256 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	257 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	258 SCEAffixPolicy.NOT_NEEDED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	259 custom_policies={custom_affix: SCEAffixPolicy.REQUIRED},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	260 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	261
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	262 stanza = string_to_domish(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	263 """<message from="foo@example.com" to="bar@example.com">
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	264 <body>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	265 Check that all affixes REQUIRED by the profile are present, including custom
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	266 affixes.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	267 </body>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	268 </message>"""
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	269 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	270
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	271 with pytest.raises(ProfileRequirementsNotMet):
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	272 XEP_0420.unpack_stanza(
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	273 unpacking_profile, stanza, XEP_0420.pack_stanza(packing_profile, stanza)
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	274 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	275
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	276
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	277 def test_rpad_affix() -> None: # pylint: disable=missing-function-docstring
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	278 profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	279 SCEAffixPolicy.REQUIRED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	280 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	281 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	282 SCEAffixPolicy.NOT_NEEDED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	283 custom_policies={},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	284 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	285
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	286 for _ in range(100):
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	287 stanza = string_to_domish(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	288 """<message from="foo@example.com" to="bar@example.com">
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	289 <body>OK</body>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	290 </message>"""
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	291 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	292
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	293 affix_values = XEP_0420.unpack_stanza(
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	294 profile, stanza, XEP_0420.pack_stanza(profile, stanza)
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	295 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	296
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	297 # Test that the rpad exists and that the content elements are always padded to at
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	298 # least 53 characters
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	299 assert affix_values.rpad is not None
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	300 assert len(affix_values.rpad) >= 53 - len("<body xmlns='jabber:client'>OK</body>")
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	301
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	302
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	303 def test_time_affix() -> None: # pylint: disable=missing-function-docstring
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	304 profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	305 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	306 SCEAffixPolicy.REQUIRED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	307 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	308 SCEAffixPolicy.NOT_NEEDED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	309 custom_policies={},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	310 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	311
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	312 stanza = string_to_domish(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	313 """<message from="foo@example.com" to="bar@example.com"></message>"""
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	314 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	315
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	316 envelope_serialized = f"""<envelope xmlns="{NS_SCE}">
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	317 <content>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	318 <body xmlns="jabber:client">
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	319 The time affix is only parsed and not otherwise verified. Not much to test
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	320 here.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	321 </body>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	322 </content>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	323 <time stamp="1969-07-21T02:56:15Z"/>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	324 </envelope>""".encode(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	325 "utf-8"
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	326 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	327
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	328 affix_values = XEP_0420.unpack_stanza(profile, stanza, envelope_serialized)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	329 assert affix_values.timestamp == datetime(1969, 7, 21, 2, 56, 15, tzinfo=timezone.utc)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	330
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	331
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	332 def test_to_affix() -> None: # pylint: disable=missing-function-docstring
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	333 profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	334 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	335 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	336 SCEAffixPolicy.REQUIRED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	337 SCEAffixPolicy.NOT_NEEDED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	338 custom_policies={},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	339 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	340
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	341 stanza = string_to_domish(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	342 """<message from="foo@example.com" to="bar@example.com">
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	343 <body>Check that the ``to`` affix is correctly added.</body>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	344 </message>"""
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	345 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	346
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	347 envelope_serialized = XEP_0420.pack_stanza(profile, stanza)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	348 affix_values = XEP_0420.unpack_stanza(profile, stanza, envelope_serialized)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	349 assert affix_values.recipient is not None
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	350 assert affix_values.recipient.userhost() == "bar@example.com"
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	351
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	352 # Check that a mismatch in recipient bare JID causes an exception to be raised
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	353 stanza = string_to_domish(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	354 """<message from="foo@example.com" to="baz@example.com"></message>"""
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	355 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	356
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	357 with pytest.raises(AffixVerificationFailed):
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	358 XEP_0420.unpack_stanza(profile, stanza, envelope_serialized)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	359
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	360 # Check that only the bare JID matters
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	361 stanza = string_to_domish(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	362 """<message from="foo@example.com" to="bar@example.com/device"></message>"""
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	363 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	364
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	365 affix_values = XEP_0420.unpack_stanza(profile, stanza, envelope_serialized)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	366 assert affix_values.recipient is not None
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	367 assert affix_values.recipient.userhost() == "bar@example.com"
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	368
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	369 stanza = string_to_domish(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	370 """<message from="foo@example.com">
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	371 <body>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	372 Check that a missing "to" attribute on the stanza fails stanza packing.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	373 </body>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	374 </message>"""
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	375 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	376
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	377 with pytest.raises(ValueError):
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	378 XEP_0420.pack_stanza(profile, stanza)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	379
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	380
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	381 def test_from_affix() -> None: # pylint: disable=missing-function-docstring
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	382 profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	383 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	384 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	385 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	386 SCEAffixPolicy.REQUIRED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	387 custom_policies={},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	388 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	389
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	390 stanza = string_to_domish(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	391 """<message from="foo@example.com" to="bar@example.com">
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	392 <body>Check that the ``from`` affix is correctly added.</body>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	393 </message>"""
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	394 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	395
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	396 envelope_serialized = XEP_0420.pack_stanza(profile, stanza)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	397 affix_values = XEP_0420.unpack_stanza(profile, stanza, envelope_serialized)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	398 assert affix_values.sender is not None
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	399 assert affix_values.sender.userhost() == "foo@example.com"
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	400
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	401 # Check that a mismatch in sender bare JID causes an exception to be raised
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	402 stanza = string_to_domish(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	403 """<message from="fooo@example.com" to="bar@example.com"></message>"""
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	404 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	405
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	406 with pytest.raises(AffixVerificationFailed):
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	407 XEP_0420.unpack_stanza(profile, stanza, envelope_serialized)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	408
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	409 # Check that only the bare JID matters
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	410 stanza = string_to_domish(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	411 """<message from="foo@example.com/device" to="bar@example.com"></message>"""
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	412 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	413
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	414 affix_values = XEP_0420.unpack_stanza(profile, stanza, envelope_serialized)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	415 assert affix_values.sender is not None
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	416 assert affix_values.sender.userhost() == "foo@example.com"
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	417
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	418 stanza = string_to_domish(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	419 """<message to="bar@example.com">
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	420 <body>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	421 Check that a missing "from" attribute on the stanza fails stanza packing.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	422 </body>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	423 </message>"""
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	424 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	425
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	426 with pytest.raises(ValueError):
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	427 XEP_0420.pack_stanza(profile, stanza)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	428
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	429
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	430 def test_custom_affixes() -> None: # pylint: disable=missing-function-docstring
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	431 custom_affix = CustomAffixImpl()
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	432
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	433 packing_profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	434 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	435 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	436 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	437 SCEAffixPolicy.NOT_NEEDED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	438 custom_policies={custom_affix: SCEAffixPolicy.REQUIRED},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	439 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	440
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	441 unpacking_profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	442 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	443 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	444 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	445 SCEAffixPolicy.NOT_NEEDED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	446 custom_policies={},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	447 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	448
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	449 stanza = string_to_domish(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	450 """<message from="foo@example.com" to="bar@example.com">
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	451 <body>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	452 If a custom affix is included in the envelope, but not excpected by the
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	453 recipient, the schema validation should fail.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	454 </body>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	455 </message>"""
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	456 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	457
3933 cecf45416403 plugin XEP-0373 and XEP-0374: Implementation of OX and OXIM: Syndace <me@syndace.dev> parents: 3911 diff changeset	458 with pytest.raises(exceptions.ParsingError):
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	459 XEP_0420.unpack_stanza(
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	460 unpacking_profile, stanza, XEP_0420.pack_stanza(packing_profile, stanza)
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	461 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	462
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	463 profile = packing_profile
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	464
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	465 stanza = string_to_domish(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	466 """<message from="foo@example.com/device0"
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	467 to="bar@example.com/Libervia.123">
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	468 <body>The affix element should be returned as part of the affix values.</body>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	469 </message>"""
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	470 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	471
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	472 affix_values = XEP_0420.unpack_stanza(
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	473 profile, stanza, XEP_0420.pack_stanza(profile, stanza)
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	474 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	475
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	476 assert custom_affix in affix_values.custom
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	477 assert (
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	478 affix_values.custom[custom_affix].getAttribute("recipient")
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	479 == "bar@example.com/Libervia.123"
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	480 )
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	481 assert (
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	482 affix_values.custom[custom_affix].getAttribute("sender")
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	483 == "foo@example.com/device0"
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	484 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	485
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	486
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	487 def test_namespace_conversion() -> None: # pylint: disable=missing-function-docstring
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	488 profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	489 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	490 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	491 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	492 SCEAffixPolicy.NOT_NEEDED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	493 custom_policies={},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	494 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	495
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	496 stanza = domish.Element((None, "message"))
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	497 stanza["from"] = "foo@example.com"
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	498 stanza["to"] = "bar@example.com"
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	499 stanza.addElement(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	500 "body",
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	501 content=(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	502 "This body element has namespace ``None``, which has to be replaced with"
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	503 " jabber:client."
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	504 ),
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	505 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	506
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	507 envelope_serialized = XEP_0420.pack_stanza(profile, stanza)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	508 envelope = string_to_domish(envelope_serialized.decode("utf-8"))
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3877 diff changeset	509 content = next(envelope.elements(NS_SCE, "content"))
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	510
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	511 # The body should have been assigned ``jabber:client`` as its namespace
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3877 diff changeset	512 assert next(content.elements("jabber:client", "body"), None) is not None
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	513
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	514 XEP_0420.unpack_stanza(profile, stanza, envelope_serialized)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	515
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	516 # The body should still have ``jabber:client`` after unpacking
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3877 diff changeset	517 assert next(stanza.elements("jabber:client", "body"), None) is not None
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	518
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	519
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	520 def test_non_encryptable_elements() -> None: # pylint: disable=missing-function-docstring
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	521 profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	522 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	523 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	524 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	525 SCEAffixPolicy.NOT_NEEDED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	526 custom_policies={},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	527 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	528
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	529 stanza = string_to_domish(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	530 """<message from="foo@example.com" to="bar@example.com">
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	531 <body>This stanza includes a store hint which must not be encrypted.</body>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	532 <store xmlns="urn:xmpp:hints"/>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	533 </message>"""
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	534 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	535
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	536 envelope_serialized = XEP_0420.pack_stanza(profile, stanza)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	537 envelope = string_to_domish(envelope_serialized.decode("utf-8"))
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3877 diff changeset	538 content = next(envelope.elements(NS_SCE, "content"))
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	539
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	540 # The store hint must not have been moved to the content element
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3877 diff changeset	541 assert next(stanza.elements(NS_HINTS, "store"), None) is not None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3877 diff changeset	542 assert next(content.elements(NS_HINTS, "store"), None) is None
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	543
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	544 stanza = string_to_domish(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	545 """<message from="foo@example.com" to="bar@example.com"></message>"""
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	546 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	547
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	548 envelope_serialized = f"""<envelope xmlns="{NS_SCE}">
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	549 <content>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	550 <body xmlns="jabber:client">
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	551 The store hint must not be moved to the stanza.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	552 </body>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	553 <store xmlns="urn:xmpp:hints"/>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	554 </content>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	555 </envelope>""".encode(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	556 "utf-8"
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	557 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	558
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	559 XEP_0420.unpack_stanza(profile, stanza, envelope_serialized)
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	560
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3877 diff changeset	561 assert next(stanza.elements(NS_HINTS, "store"), None) is None
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	562
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	563
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	564 def test_schema_validation() -> None: # pylint: disable=missing-function-docstring
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	565 profile = SCEProfile(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	566 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	567 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	568 SCEAffixPolicy.NOT_NEEDED,
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	569 SCEAffixPolicy.NOT_NEEDED,
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	570 custom_policies={},
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	571 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	572
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	573 stanza = string_to_domish(
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	574 """<message from="foo@example.com" to="bar@example.com"></message>"""
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	575 )
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	576
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	577 envelope_serialized = f"""<envelope xmlns="{NS_SCE}">
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	578 <content>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	579 <body xmlns="jabber:client">
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	580 An unknwon affix should cause a schema validation error.
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	581 </body>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	582 <store xmlns="urn:xmpp:hints"/>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	583 </content>
00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	584 <unknown-affix unknown-attr="unknown"/>
4285 f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	585 </envelope>""".encode(
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	586 "utf-8"
f1d0cde61af7 tests (unit): fix tests + black reformatting. Goffi <goffi@goffi.org> parents: 4071 diff changeset	587 )
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	588
3933 cecf45416403 plugin XEP-0373 and XEP-0374: Implementation of OX and OXIM: Syndace <me@syndace.dev> parents: 3911 diff changeset	589 with pytest.raises(exceptions.ParsingError):
3877 00212260f659 plugin XEP-0420: Implementation of Stanza Content Encryption: Syndace <me@syndace.dev> parents: diff changeset	590 XEP_0420.unpack_stanza(profile, stanza, envelope_serialized)

Mercurial > libervia-backend

annotate tests/unit/test_plugin_xep_0420.py @ 4351:6a0a081485b8