pep/slixmpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

xep_0454: Ensure format_url returns a str

Browse source 
Signed-off-by: Maxime “pep” Buquet <pep@bouah.net>
This commit is contained in:
Maxime “pep” Buquet 2022-03-18 16:11:07 +01:00
parent 14a6c7801d
commit 7222ade0dd
2 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
slixmpp/plugins/xep_0454/__init__.py
View file

@ -117,7 +117,7 @@ class XEP_0454(BasePlugin):
"""Helper to format a HTTPS URL to an AESGCM URI""" """Helper to format a HTTPS URL to an AESGCM URI"""
if not url.startswith('https://') or url.find('#') != -1: if not url.startswith('https://') or url.find('#') != -1:
raise InvalidURL raise InvalidURL
url = 'aesgcm://' + url.removeprefix('https://') + '#' + fragment return 'aesgcm://' + url.removeprefix('https://') + '#' + fragment
async def upload_file( async def upload_file(
self, self,

6
tests/test_xep_0454.py
View file

@ -32,4 +32,10 @@ class TestMediaSharing(SlixTest):
self.assertEqual(plain, result) self.assertEqual(plain, result)
def testFormatURL(self):
url = 'https://foo.bar'
fragment = 'a' * 88
result = XEP_0454.format_url(url, fragment)
self.assertEqual('aesgcm://foo.bar#' + 'a' * 88, result)
suite = unittest.TestLoader().loadTestsFromTestCase(TestMediaSharing) suite = unittest.TestLoader().loadTestsFromTestCase(TestMediaSharing)